Update pages/Translate.py

pages/Translate.py

@@ -29,6 +29,8 @@ with st.sidebar:
         st.caption('markdown')
     else:
         st.caption('latex')
+    
+    st.toggle('双语输出', key='trans_en_output')
 
 
 def get_translate_and_conclude(question: str, step: int):
@@ -114,6 +116,11 @@ if prompt := st.chat_input():
     translate_result = chat_container.chat_message("ai").write_stream(response)
     st.session_state.translate_messages.append({'role': 'assistant', 'content': translate_result})
 
+    if st.session_state.get('trans_en_output'):
+        markdown_text = f"""{prompt}\t\n\n"""
+    else:
+        markdown_text = ""
+
     if st.session_state.get('trans_conclusion'):
         query = "接下来，请用两到四句话总结一下这段文本的内容"
         chat_container.chat_message("human").write(query)
@@ -125,13 +132,13 @@ if prompt := st.chat_input():
         st.session_state.translate_messages.append({'role': 'assistant', 'content': conclusion_result})
 
         if st.session_state.get('trans_text_mode'):
-            markdown_text = f"""{prompt}\t\r\n{translate_result}\t\r\n> {conclusion_result}"""
+            markdown_text += f"""{translate_result}\t\r\n> {conclusion_result}"""
         else:
-            markdown_text = f"""{prompt}\n\n{translate_result}\n\n\\tbox{{ {conclusion_result} }}"""
+            markdown_text += f"""{translate_result}\n\n\\tbox{{ {conclusion_result} }}"""
             markdown_text = markdown_text.replace('%', r'\%')
         st.session_state.markdown_text = markdown_text
     else:
-        markdown_text = f"""{prompt}\t\r\n{translate_result}"""
+        markdown_text += f"""{translate_result}"""
         st.session_state.markdown_text = markdown_text
 
     st.rerun()