Upload 2 files

app.py

@@ -183,6 +183,7 @@ def checklen(text):
     return text
 
 def getanswerByXunFei(t):
+    st.session_state['prompts'].append(t)
     text.clear()
     question = checklen(getText("user", t))
     ans = ""
@@ -191,19 +192,36 @@ def getanswerByXunFei(t):
     return answer
 
 
-st.markdown("我是讯飞聊天机器人,我可以回答您的任何问题")
+if 'prompts' not in st.session_state:
+    st.session_state['prompts'] = []
+    
 if 'generated' not in st.session_state:
     st.session_state['generated'] = []
+
 if 'past' not in st.session_state:
     st.session_state['past'] = []
-user_input=st.text_input("请输入您的问题:",key='input')
-if user_input:
-    output=getanswerByXunFei(user_input)
-    st.session_state['past'].append(user_input)
-    st.session_state['generated'].append(output)
+
+def end_click():
+    st.session_state['prompts'] = []
+    st.session_state['past'] = []
+    st.session_state['generated'] = []
+    st.session_state['user'] = ""
+def chat_click():
+    if st.session_state['user'] != '':
+        chat_input = st.session_state['user']
+        output = getanswerByXunFei(chat_input)
+        st.session_state['past'].append(chat_input)
+        st.session_state['generated'].append(output)
+        st.session_state['prompts'].append(output)
+        st.session_state['user'] = ""
+st.image("./logo.jpg", use_column_width='auto')
+st.title("珠宝贝专属聊天机器人")
+
+user_input = st.text_input("输入:", key="user")
+chat_button = st.button("发送", on_click=chat_click)
+end_button = st.button("新聊天", on_click=end_click)
+
 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')
+    for i in range(0, len(st.session_state['generated']), 1):
+        message(st.session_state['past'][i], is_user=True)
+        message(st.session_state['generated'][i], key=str(i))