Update app.py

app.py

@@ -341,7 +341,7 @@ def process_video_with_gpt(video_input, user_prompt):
     return response.choices[0].message.content
 
 # ArXiv Search Functions
-def search_arxiv(query):
+def search_arxiv_old(query):
     """Search ArXiv papers using Hugging Face client."""
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
     response = client.predict(
@@ -352,6 +352,92 @@ def search_arxiv(query):
     )
     return response
 
+def search_arxiv(query):
+    
+    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
+
+
+# Function to generate a filename based on prompt and time (because names matter 🕒)
+def generate_filename(prompt, file_type):
+    central = pytz.timezone('US/Central')
+    safe_date_time = datetime.now(central).strftime("%m%d_%H%M")
+    safe_prompt = re.sub(r'\W+', '_', prompt)[:90]
+    return f"{safe_date_time}_{safe_prompt}.{file_type}"
+
+# Function to create and save a file (and avoid the black hole of lost data 🕳)
+def create_file(filename, prompt, response):
+    with open(filename, 'w', encoding='utf-8') as file:
+        file.write(prompt + "\n\n" + response)
+
+    
+def perform_ai_lookup(query):
+    start_time = time.strftime("%Y-%m-%d %H:%M:%S")
+    client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
+    response1 = client.predict(
+        query,
+        20,
+        "Semantic Search",
+        "mistralai/Mixtral-8x7B-Instruct-v0.1",
+        api_name="/update_with_rag_md"
+    )
+    Question = '### 🔎 ' + query + '\r\n'  # Format for markdown display with links
+    References =  response1[0]  
+    ReferenceLinks = extract_urls(References)
+
+    RunSecondQuery = True
+    results=''
+    if RunSecondQuery:
+        # Search 2 - Retrieve the Summary with Papers Context and Original Query
+        response2 = client.predict(
+            query,
+            "mistralai/Mixtral-8x7B-Instruct-v0.1",
+            True,
+            api_name="/ask_llm"
+        )
+        if len(response2) > 10:
+            Answer = response2
+            SpeechSynthesis(Answer)
+            # Restructure results to follow format of Question, Answer, References, ReferenceLinks
+            results = Question + '\r\n' + Answer + '\r\n' + References + '\r\n' + ReferenceLinks
+            st.markdown(results)
+
+    st.write('🔍Run of Multi-Agent System Paper Summary Spec is Complete')
+    end_time = time.strftime("%Y-%m-%d %H:%M:%S")
+    start_timestamp = time.mktime(time.strptime(start_time, "%Y-%m-%d %H:%M:%S"))
+    end_timestamp = time.mktime(time.strptime(end_time, "%Y-%m-%d %H:%M:%S"))
+    elapsed_seconds = end_timestamp - start_timestamp
+    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)
+    return results
+
 # Chat Processing Functions
 def process_with_gpt(text_input):
     """Process text with GPT-4o."""
@@ -549,7 +635,7 @@ def main():
         # Model Selection
         model_choice = st.sidebar.radio(
             "Choose AI Model:",
-            ["GPT-4o", "Claude-3", "Both"]
+            ["GPT-4o", "Claude-3", "GPT+Claude+Arxiv"]
         )
         
         # Chat Interface
@@ -578,7 +664,9 @@ def main():
                     with col3:
                         st.subheader("Arxiv and Mistral Research:")
                         with st.spinner("Searching ArXiv..."):
-                            results = search_arxiv(user_input)
+                            #results = search_arxiv(user_input)
+                            results = perform_ai_lookup(user_input)
+                            
                             st.markdown(results)
                             
         # Display Chat History
@@ -587,9 +675,9 @@ def main():
         
         with tab1:
             for chat in st.session_state.chat_history:
-                st.text_area("You:", chat["user"], height=100, disabled=True)
-                st.text_area("Claude:", chat["claude"], height=200, disabled=True)
-                st.markdown("---")
+                st.text_area("You:", chat["user"], height=100)
+                st.text_area("Claude:", chat["claude"], height=200)
+                st.markdown(chat["claude"])
         
         with tab2:
             for message in st.session_state.messages: