add app.py

app.py

@@ -0,0 +1,61 @@
+import streamlit as st
+from streamlit_chat import message
+
+import openai
+from config import open_api_key
+openai.api_key = st.secrets["db_username"]
+
+# openAI code
+def openai_create(prompt):
+ response = openai.Completion.create(
+ model="text-davinci-003",
+ prompt=prompt,
+ temperature=0.9,
+ max_tokens=1000,
+ top_p=1,
+ frequency_penalty=0,
+ presence_penalty=0.6,
+ stop=[" Human:", " AI:"]
+ )
+ return response.choices[0].text
+
+# Streamlit App
+st.set_page_config(
+ page_title="Streamlit Chat - Demo",
+ page_icon=":robot:"
+)
+
+# st.header("GPT 3.5 with Streamlit\nChatGPT API即将到来")
+
+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:
+ prompt = 'The following is a conversation with an AI assistant. The assistant is helpful, creative, clever, and very friendly.'
+ if st.session_state['generated']:
+ for past, generated in zip(st.session_state['past'], st.session_state['generated']):
+ if len(generated)>0:
+ prompt += f'\nHuman:{past}\nAI:{generated}'
+ prompt += f'\nHuman:{user_input}\nAI:'
+ logger.info(prompt)
+ output = openai_create(prompt)
+ logger.info(f'{user_input} -> {output}')
+ 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')