Update app.py

app.py

@@ -472,46 +472,16 @@ def perform_ai_lookup(query):
     )
     st.write('🔍Run of Multi-Agent System Paper Summary Spec is Complete')
     st.markdown(response2)
-
-    # ArXiv searcher ~-<>-~ Paper References - Update with RAG
+    
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
-    response1 = client.predict(
-    		query,	
-    		10,	
-    		#"Semantic Search - up to 10 Mar 2024",	# Literal['Semantic Search - up to 10 Mar 2024', 'Arxiv Search - Latest - (EXPERIMENTAL)']  in 'Search Source' Dropdown component
-    		
-            "Arxiv Search - Latest - (EXPERIMENTAL)",
-            "mistralai/Mixtral-8x7B-Instruct-v0.1",	# Literal['mistralai/Mixtral-8x7B-Instruct-v0.1', 'mistralai/Mistral-7B-Instruct-v0.2', 'google/gemma-7b-it', 'None']  in 'LLM Model' Dropdown component
+    result = client.predict(
+    		message=response2,
+    		llm_results_use=5,
+    		database_choice="Semantic Search",
+    		llm_model_picked="mistralai/Mistral-7B-Instruct-v0.2",
     		api_name="/update_with_rag_md"
     )
-    st.write('🔍Run of Multi-Agent System Paper References is Complete')
-    responseall = response2 + response1[0] + response1[1]
-    st.markdown(responseall)
-
-                      
-    """Perform AI lookup using Gradio client."""
-    st.write("Performing AI Lookup...")
-    client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
-    result1 = client.predict(
-        prompt=query,
-        llm_model_picked="mistralai/Mixtral-8x7B-Instruct-v0.1",
-        stream_outputs=True,
-        api_name="/ask_llm"
-    )
-    st.markdown("### Mixtral-8x7B-Instruct-v0.1 Result")
-    st.markdown(result1)
-    result2 = client.predict(
-        prompt=query,
-        llm_model_picked="mistralai/Mistral-7B-Instruct-v0.2",
-        stream_outputs=True,
-        api_name="/ask_llm"
-    )
-    st.markdown("### Mistral-7B-Instruct-v0.2 Result")
-    st.markdown(result2)
-    combined_result = f"{result1}\n\n{result2}"
-    #return combined_result
-
-    return responseall
+    return result     
 
 def display_file_content(file_path):
     """Display file content with editing capabilities."""