Update app.py

app.py

@@ -335,6 +335,7 @@ def search_glossary(query):
         api_name="/ask_llm"
     )
     st.markdown(result)
+    
     st.code(result, language="python", line_numbers=True)
     # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /ask_llm
     result2 = client.predict(
@@ -344,6 +345,7 @@ def search_glossary(query):
         api_name="/ask_llm"
     )
     st.markdown(result2)
+    
     st.code(result2, language="python", line_numbers=True)
     # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /ask_llm
     result3 = client.predict(
@@ -353,6 +355,7 @@ def search_glossary(query):
         api_name="/ask_llm"
     )
     st.markdown(result3)
+    
     st.code(result3, language="python", line_numbers=True)
     # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /update_with_rag_md
     response2 = client.predict(
@@ -363,9 +366,31 @@ def search_glossary(query):
         api_name="/update_with_rag_md"
     )
     st.markdown(response2[0])
-    st.code(response2[0], language="python", line_numbers=True, wrap_lines=True)  
+    st.code(response2[0], language="python", line_numbers=True, wrap_lines=True) 
+    
     st.markdown(response2[1])
     st.code(response2[1], language="python", line_numbers=True, wrap_lines=True)
+
+
+    # Persist AI Results to Markdown Files
+    filename = generate_filename(result, "md")
+    create_file(filename, query, result)
+    st.session_state.chat_history.append({"assistant": query, "ArXiV": result})
+    
+    filename = generate_filename(result2, "md")
+    create_file(filename, query, result2)
+    st.session_state.chat_history.append({"assistant": query, "ArXiV": result2})
+    
+    filename = generate_filename(result3, "md")
+    create_file(filename, query, result3)
+    st.session_state.chat_history.append({"assistant": query, "ArXiV": result3})
+    
+    filename = generate_filename(response2, "md")
+    create_file(filename, query, response2)
+    st.session_state.chat_history.append({"assistant": query, "ArXiV": response2})
+    
+
+    
     return result, result2, result3, response2