import streamlit as st
from freeGPT import Client

st.title("ChatGPT4 UI")
cont = st.container(height=600)
prompt = st.chat_input(placeholder="Eg. How are you?")

if prompt:
    output = Client.create_completion("gpt3", prompt)
    user_pr = cont.chat_message("user")
    user_pr.write(prompt)
    output_msg = cont.chat_message("assistant")
    output_msg.write(output)