添加复制块

pages/AcademicTranslate.py

@@ -26,7 +26,11 @@ with st.sidebar:
 if 'ac_translate_history' not in st.session_state:
     st.session_state.ac_translate_history = []
 
-chat_container = st.container(height=610, border=False)
+if 'ac_markdown_text' not in st.session_state:
+    st.session_state.ac_markdown_text = ''
+
+chat_container = st.container(height=580, border=False)
+st.code(st.session_state.ac_markdown_text, language='markdown')
 
 with chat_container:
     for message in st.session_state.ac_translate_history:
@@ -78,3 +82,4 @@ 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_markdown_text = translate_result