import streamlit as st
from freeGPT import Client

st.title("Freak AI - Your stupid assistant")
st.header("Type any message and get stupid response", divider="rainbow")
messages = st.container(height=400)
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.chat_message("user", avatar="https://cdn.discordapp.com/attachments/1159192117751578705/1207020816332619796/kvikontent_professional_shot_of_clever_man_witha_book_in_hands__73d404c7-c515-4d49-b989-ac84dfe1e4ea.png?ex=65de20bb&is=65cbabbb&hm=1086ecaa2cfd107aa2912981226548f17bef10d54459f69d22176d38bdb1a05a&")
    user_msg.write(prompt)
    resp_msg = messages.chat_message("assistant", avatar="https://cdn.discordapp.com/attachments/1159192117751578705/1207019218466373642/kvikontent_professioanl_shot_of_stupid_man_c81e05bf-4530-497e-a9f3-63acdeea5345.png?ex=65de1f3f&is=65cbaa3f&hm=79516030ac8aa4d53f19c82a05cacd32ec0dceeb9b75d329a253bf4f3b8727a9&")
    resp_msg.write(resp)