Update app.py

app.py

@@ -147,11 +147,31 @@ def search_arxiv(query):
 def perform_ai_lookup(q):
     start = time.time()
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
+    # Perform a RAG-based search
     r = client.predict(q,20,"Semantic Search","mistralai/Mixtral-8x7B-Instruct-v0.1",api_name="/update_with_rag_md")
     refs = r[0]
+    # Ask model for answer
     r2 = client.predict(q,"mistralai/Mixtral-8x7B-Instruct-v0.1",True,api_name="/ask_llm")
     result = f"### 🔎 {q}\n\n{r2}\n\n{refs}"
+    # Speak results
     speech_synthesis_html(r2)
+
+    # Attempt to speak summaries and titles from refs
+    # Assuming refs contain a set of references in Markdown with possible titles.
+    # We'll just re-speak refs as "summaries".
+    summaries_text = "Here are the summaries from the references: " + refs.replace('"','')
+    speech_synthesis_html(summaries_text)
+
+    # Extract titles from refs (looking for markdown links [Title](URL))
+    titles = []
+    for line in refs.split('\n'):
+        m = re.search(r"\[([^\]]+)\]", line)
+        if m:
+            titles.append(m.group(1))
+    if titles:
+        titles_text = "Here are the titles of the papers: " + ", ".join(titles)
+        speech_synthesis_html(titles_text)
+
     st.markdown(result)
     elapsed = time.time()-start
     st.write(f"Elapsed: {elapsed:.2f} s")
@@ -266,7 +286,9 @@ def display_file_manager():
 def main():
     st.sidebar.markdown("### 🚲BikeAI🏆 Multi-Agent Research AI")
     tab_main = st.radio("Action:",["🎤 Voice Input","📸 Media Gallery","🔍 Search ArXiv","📝 File Editor"],horizontal=True)
-    model_choice = st.sidebar.radio("AI Model:",["GPT+Claude+Arxiv","GPT-4o","Claude-3"])
+    
+    # Changed model order and default:
+    model_choice = st.sidebar.radio("AI Model:", ["Arxiv","GPT-4o","Claude-3","GPT+Claude+Arxiv"], index=0)
 
     # Speech-to-Text component placeholder (example)
     mycomponent = components.declare_component("mycomponent", path="mycomponent")
@@ -277,7 +299,12 @@ def main():
             process_with_gpt(user_input)
         elif model_choice == "Claude-3":
             process_with_claude(user_input)
+        elif model_choice == "Arxiv":
+            # Just Arxiv on its own, full column, speak results
+            st.subheader("Arxiv Only Results:")
+            perform_ai_lookup(user_input)
         else:
+            # GPT+Claude+Arxiv
             col1,col2,col3=st.columns(3)
             with col1:
                 st.subheader("GPT-4o Omni:")
@@ -304,7 +331,11 @@ def main():
                     process_with_gpt(user_text)
                 elif model_choice == "Claude-3":
                     process_with_claude(user_text)
+                elif model_choice == "Arxiv":
+                    st.subheader("Arxiv Only Results:")
+                    perform_ai_lookup(user_text)
                 else:
+                    # GPT+Claude+Arxiv
                     col1,col2,col3=st.columns(3)
                     with col1:
                         st.subheader("GPT-4o Omni:")