update

pages/AcademicTranslate.py

@@ -14,6 +14,12 @@ st.set_page_config(
 
 st.title("学术中英互译")
 
+
+def clean_screen():
+    st.session_state.ac_translate_history = []
+    st.rerun()
+
+
 with st.sidebar:
     side_bar_links()
 
@@ -23,9 +29,14 @@ with st.sidebar:
     else:
         st.caption('zh → en')
 
+    st.button("清屏", on_click=clean_screen)
+
 if 'ac_translate_history' not in st.session_state:
     st.session_state.ac_translate_history = []
 
+if 'ac_code_text' not in st.session_state:
+    st.session_state.ac_code_text = ''
+
 chat_container = st.container(height=610, border=False)
 
 with chat_container:
@@ -34,6 +45,9 @@ with chat_container:
         with st.chat_message(message['role']):
             st.markdown(message['content'])
 
+with st.container(height=150,border=False):
+    st.code(st.session_state.ac_code_text, 'markdown')
+
 
 def ac_translate(original_text: str, is_en2zh: bool):
     if is_en2zh:
@@ -76,3 +90,6 @@ if prompt := st.chat_input():
     response = ac_translate(prompt, st.session_state.get('is_en2zh'))
     translate_result = chat_container.chat_message("ai").write_stream(response)
     st.session_state.ac_translate_history.append({'role': 'assistant', 'content': translate_result})
+    st.session_state.ac_code_text = translate_result
+
+    st.rerun()