Update app.py

app.py

@@ -596,6 +596,11 @@ def search_arxiv(query):
     References =  response1[0]  
     ReferenceLinks = extract_urls(References)
 
+    
+    filename = generate_filename(query, "md")
+    create_file(filename, query, References + ReferenceLinks, should_save)
+    st.session_state.messages.append({"role": "assistant", "content": References + ReferenceLinks})
+
     RunSecondQuery = True
     results=''
     if RunSecondQuery:
@@ -622,10 +627,6 @@ def search_arxiv(query):
     st.write(f"Finish time: {end_time}")
     st.write(f"Elapsed time: {elapsed_seconds:.2f} seconds")
 
-    filename = generate_filename(query, "md")
-    create_file(filename, query, results, should_save)
-    st.session_state.messages.append({"role": "assistant", "content": results})
-
     #filename = generate_filename(query, "md")
     #create_file(filename, query, results, should_save)