Update app.py

app.py

@@ -101,30 +101,31 @@ def load_file(file_name):
 #)
 #st.markdown(result)
 
-def search_arxiv():
+def search_arxiv(query):
     st.title("▶️ Semantic and Episodic Memory System")
 
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
 
-    search_query = st.text_input("Search")
+    search_query = query
     top_n_results = st.slider("Top n results as context", min_value=4, max_value=100, value=100)
     search_source = st.selectbox("Search Source", ["Semantic Search - up to 10 Mar 2024", "Arxiv Search - Latest - (EXPERIMENTAL)"])
     llm_model = st.selectbox("LLM Model", ["mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mistral-7B-Instruct-v0.2", "google/gemma-7b-it", "None"])
 
-    if st.button("🔎 Search"):
-        result = client.predict(
-            search_query,
-            top_n_results,
-            search_source,
-            llm_model,
-            api_name="/update_with_rag_md"
-        )
-        st.markdown(result)
+    st.markdown('### 🔎 ' + query
+    #if st.button("🔎 Search"):
+    result = client.predict(
+        search_query,
+        top_n_results,
+        search_source,
+        llm_model,
+        api_name="/update_with_rag_md"
+    )
+    st.markdown(result)
 
-        file_type = st.radio("Select Memory Flag", ("Semantic", "Episodic"))
-        if st.button("Save"):
-            file_name = save_file(result, file_type)
-            st.success(f"File saved: {file_name}")
+    file_type = st.radio("Select Memory Flag", ("Semantic", "Episodic"))
+    if st.button("Save"):
+        file_name = save_file(result, file_type)
+        st.success(f"File saved: {file_name}")
 
     saved_files = [f for f in os.listdir(".") if f.endswith(".md")]
     selected_file = st.sidebar.selectbox("Saved Files", saved_files)
@@ -145,8 +146,6 @@ def search_arxiv():
             os.remove(selected_file)
             st.warning(f"File deleted: {selected_file}")
 
-
-
 # Set page configuration with a title and favicon
 st.set_page_config(
     page_title="📖🔍Arxiv-Paper-Search-QA-RAG-Streamlit-Gradio-APAI",