Update app.py

app.py

@@ -621,8 +621,14 @@ def search_arxiv(query):
     st.write(f"Start time: {start_time}")
     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)
+
+    filename = generate_filename(text_input, "md")
+    create_file(filename, text_input, results, should_save)
+    st.session_state.messages.append({"role": "assistant", "content": results})
+
+    #filename = generate_filename(query, "md")
+    #create_file(filename, query, results, should_save)
+    
     return results
 
 def download_pdfs_and_generate_html(urls):