Spaces:
Sleeping
Sleeping
File size: 1,301 Bytes
73d5bd0 1c72454 73d5bd0 c336be0 73d5bd0 1c72454 73d5bd0 edec87d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
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')
|