Update app.py

app.py

@@ -1,29 +1,63 @@
-import openai
 import streamlit as st
+from streamlit_chat import message
+
+import openai
 
-openai.api_key ="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" # Replace this with your API key: https://beta.openai.com/docs/quickstart/add-your-api-key
+openai.api_key = "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"
+# openAI code
+def openai_create(prompt):
 
-def openai_chat(prompt):
-    completions = openai.Completion.create(
-        engine="text-davinci-003",
-        prompt="请给出这道题的答案与解释:"+prompt,
-        max_tokens=1024,
-        n=1,
-        temperature=0.5,
+    response = openai.Completion.create(
+    model="text-davinci-003",
+    prompt=prompt,
+    temperature=0.9,
+    max_tokens=150,
+    top_p=1,
+    frequency_penalty=0,
+    presence_penalty=0.6,
+    stop=[" Human:", " AI:"]
     )
 
-    message = completions.choices[0].text
-    return message.strip()
-
-def chatbot(history = []):
-    
-    input_text = st.text_input("Enter your question:")
-    if st.button('Submit'):
-        output = openai_chat(input_text)
-        history.append((input_text, output))
-        st.write("Robot: ",output)
-        st.write("History:",history)
-      
-        
-if __name__ == "__main__":
-    chatbot()
+    return response.choices[0].text
+
+
+def chatgpt_clone(input):
+    output = openai_create(input)
+    return output
+
+# Streamlit App
+st.set_page_config(
+    page_title="Streamlit Chat - Demo",
+    page_icon=":robot:"
+)
+
+st.header("🏅和小译学长聊聊天")
+
+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("🏅请告诉小译学长你想说的吧: ", key="input")
+    if st.button("发送"):
+        return input_text
+    return None
+
+
+
+user_input = get_text()
+
+if user_input:
+     
+    output = chatgpt_clone(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')