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')