import streamlit as st
from freeGPT import Client

st.title("Dude AI - Your stupid assistant")
st.header("Type any message and get stupid response", divider="rainbow")
messages = st.container(height=600)
prompt = st.chat_input(placeholder="Eg. How are you today?")

if prompt:
    resp = Client.create_completion("gpt3", prompt + " . Give answer as very stupid dude guy, as stupid and funny as it can be. I should laugh when i'll read answer")
    user_msg = messages.message("user")
    user_msg.write(prompt)
    resp_msg = messages.message("assistant", avatar="avatar.jpg")
    resp_msg.write(resp)