import openai
import streamlit as st
from streamlit_option_menu import option_menu
from streamlit_chat import message

st.markdown("<h1 style='text-align: center';>Arjun_AI 💩</h1>", unsafe_allow_html=True)
st.markdown("---")        

openai.api_key = st.secrets["API"]

def generate_response(prompt):


    completion = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",

    messages=[
            {"role": "user", "content": f"Imitate as Arjun Dutta, and try to give answer in english sometimes: {prompt}"}
                ]
    )


    message = completion.choices[0].message.get("content")
    return message

if 'generated' not in st.session_state:
    st.session_state['generated'] = []

if 'past' not in st.session_state:
    st.session_state['past'] = []


def get_text():
    input_text = st.text_input("You: ","", key="input")
    return input_text 


user_input = get_text()

if user_input:
    output = generate_response(user_input)
    st.session_state.past.append(user_input)
    st.session_state.generated.append(output)

if st.session_state['generated']:

    for i in range(len(st.session_state['generated'])-1, -1, -1):
        message(st.session_state["generated"][i], key=str(i))
        message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')