import streamlit as st
from freeGPT import Client

@st.cache_data(ttl=3600)
def get_response(user_input):
    try:
        response = client.create_completion("gpt3", user_input)
        st.chat_message("You", user_input, avatar=None)
        st.chat_message("KviGPT", response, avatar=None)
    except Exception as e:
        return "Error: {}".format(str(e))

st.sidebar.title("KviGPT")
st.sidebar.subheader("Chat based on Gpt-3")

user_input = st.chat_input("Say something")

if user_input:
    get_response(user_input)