Update app.py

app.py

@@ -270,6 +270,74 @@ async def load_chat():
     numbered_content = "\n".join(f"{i+1}. {line}" for i, line in enumerate(lines) if line.strip())
     return numbered_content
 
+# Claude Search Function
+async def perform_claude_search(query, username):
+    client = anthropic.Anthropic(api_key=anthropic_key)
+    response = client.messages.create(
+        model="claude-3-sonnet-20240229",
+        max_tokens=1000,
+        messages=[{"role": "user", "content": query}]
+    )
+    result = response.content[0].text
+    st.markdown(f"### Claude's Reply 🧠\n{result}")
+    
+    # Save to chat history with audio
+    audio_file, _ = await async_edge_tts_generate(result, st.session_state['tts_voice'], username)
+    if audio_file:
+        central = pytz.timezone('US/Central')
+        timestamp = datetime.now(central).strftime("%Y-%m-%d %H:%M:%S")
+        entry = f"[{timestamp}] {username}:\n```markdown\nClaude Search: {query}\nResponse: {result}\n```"
+        with open(CHAT_FILE, 'a') as f:
+            f.write(f"{entry}\n")
+        with open(HISTORY_FILE, 'a') as f:
+            f.write(f"[{timestamp}] {username}: Audio - {audio_file}\n")
+        st.session_state['mp3_files'][os.path.basename(audio_file)] = audio_file
+        st.session_state.chat_history.append(entry)
+        play_and_download_audio(audio_file)
+    return result
+
+# ArXiv Search Function
+async def perform_arxiv_search(query, username):
+    # Step 1: Claude Search
+    client = anthropic.Anthropic(api_key=anthropic_key)
+    claude_response = client.messages.create(
+        model="claude-3-sonnet-20240229",
+        max_tokens=1000,
+        messages=[{"role": "user", "content": query}]
+    )
+    claude_result = claude_response.content[0].text
+    st.markdown(f"### Claude's Reply 🧠\n{claude_result}")
+
+    # Step 2: Feed Claude result into ArXiv
+    enhanced_query = f"{query}\n\n{claude_result}"
+    gradio_client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
+    refs = gradio_client.predict(
+        enhanced_query, 10, "Semantic Search", "mistralai/Mixtral-8x7B-Instruct-v0.1", api_name="/update_with_rag_md"
+    )[0]
+    result = f"🔎 {enhanced_query}\n\n{refs}"
+    st.markdown(f"### ArXiv Results 🔍\n{result}")
+
+    # Parse papers and generate long audio
+    papers = parse_arxiv_refs(refs)
+    if papers:
+        long_audio_text = f"Claude Search: {query}\nResponse: {claude_result}\n\nArXiv Results:\n"
+        for i, paper in enumerate(papers, 1):
+            long_audio_text += f"Paper {i}: {paper['title']} by {paper['authors']}. Summary: {paper['summary']}\n"
+        audio_file, _ = await async_edge_tts_generate(long_audio_text, st.session_state['tts_voice'], username)
+        if audio_file:
+            central = pytz.timezone('US/Central')
+            timestamp = datetime.now(central).strftime("%Y-%m-%d %H:%M:%S")
+            entry = f"[{timestamp}] {username}:\n```markdown\nArXiv Search: {query}\nClaude Response: {claude_result}\nArXiv Results: {refs}\n```"
+            with open(CHAT_FILE, 'a') as f:
+                f.write(f"{entry}\n")
+            with open(HISTORY_FILE, 'a') as f:
+                f.write(f"[{timestamp}] {username}: Audio - {audio_file}\n")
+            st.session_state['mp3_files'][os.path.basename(audio_file)] = audio_file
+            st.session_state.chat_history.append(entry)
+            st.subheader("📝 Full ArXiv Audio")
+            play_and_download_audio(audio_file)
+    return result, papers
+
 async def perform_ai_lookup(q, vocal_summary=True, extended_refs=False, titles_summary=True, full_audio=False, useArxiv=True, useArxivAudio=False):
     start = time.time()
     client = anthropic.Anthropic(api_key=anthropic_key)
@@ -553,7 +621,6 @@ def main():
     tab_main = st.radio("Action:", ["🎤 Chat & Voice", "📸 Media", "🔍 ArXiv", "📚 PDF to Audio"], horizontal=True, key="tab_main")
     useArxiv = st.checkbox("Search ArXiv", True, key="use_arxiv")
     useArxivAudio = st.checkbox("ArXiv Audio", False, key="use_arxiv_audio")
-    # Use st.checkbox without reassigning to st.session_state
     st.checkbox("Autosend Chat", value=True, key="autosend")
     st.checkbox("Autosearch ArXiv", value=True, key="autosearch")
 
@@ -592,17 +659,37 @@ def main():
                 filename = asyncio.run(save_pasted_image(paste_result.image_data, st.session_state.username))
                 if filename:
                     st.session_state.pasted_image_data = filename
+        
         if (message and message != st.session_state.last_message) or st.session_state.pasted_image_data:
             st.session_state.last_message = message
-            if st.session_state.autosend or st.button("Send 🚀", key="send_button"):  # Use st.session_state.autosend directly
-                if message.strip():
-                    asyncio.run(save_chat_entry(st.session_state.username, message, True))
-                if st.session_state.pasted_image_data:
-                    asyncio.run(save_chat_entry(st.session_state.username, f"Pasted image: {st.session_state.pasted_image_data}"))
-                    st.session_state.pasted_image_data = None
-                st.session_state.timer_start = time.time()
-                save_username(st.session_state.username)
-                st.rerun()
+            col_send, col_claude, col_arxiv = st.columns([1, 1, 1])
+            
+            with col_send:
+                if st.session_state.autosend or st.button("Send 🚀", key="send_button"):  # Existing Send button
+                    if message.strip():
+                        asyncio.run(save_chat_entry(st.session_state.username, message, True))
+                    if st.session_state.pasted_image_data:
+                        asyncio.run(save_chat_entry(st.session_state.username, f"Pasted image: {st.session_state.pasted_image_data}"))
+                        st.session_state.pasted_image_data = None
+                    st.session_state.timer_start = time.time()
+                    save_username(st.session_state.username)
+                    st.rerun()
+
+            with col_claude:
+                if st.button("🧠 Claude", key="claude_button"):  # New Claude button
+                    if message.strip():
+                        await perform_claude_search(message, st.session_state.username)
+                        st.session_state.timer_start = time.time()
+                        save_username(st.session_state.username)
+                        st.rerun()
+
+            with col_arxiv:
+                if st.button("🔍 ArXiv", key="arxiv_button"):  # New ArXiv button
+                    if message.strip():
+                        await perform_arxiv_search(message, st.session_state.username)
+                        st.session_state.timer_start = time.time()
+                        save_username(st.session_state.username)
+                        st.rerun()
 
     # 📸 Media
     elif tab_main == "📸 Media":