Update app.py

app.py

@@ -126,7 +126,8 @@ def init_session_state():
         'enable_audio': True, 'download_link_cache': {}, 'username': None,
         'autosend': True, 'autosearch': True, 'last_message': "", 'last_query': "",
         'mp3_files': {}, 'timer_start': time.time(), 'quote_index': 0,
-        'quote_source': "famous", 'last_sent_transcript': "", 'old_val': None
+        'quote_source': "famous", 'last_sent_transcript': "", 'old_val': None,
+        'last_refresh': time.time()
     }
     for k, v in defaults.items():
         if k not in st.session_state:
@@ -258,7 +259,7 @@ async def save_chat_entry(username, message, voice, is_markdown=False):
     await broadcast_message(f"{username}|{message}", "chat")
     st.session_state.last_chat_update = time.time()
     st.session_state.chat_history.append(entry)
-    st.session_state.last_transcript = message  # Update last processed transcript
+    st.session_state.last_transcript = message
     return md_file, audio_file
 
 async def load_chat():
@@ -268,7 +269,6 @@ async def load_chat():
     with open(CHAT_FILE, 'r') as f:
         content = f.read().strip()
     lines = content.split('\n')
-    # Remove duplicates and empty lines
     unique_lines = list(dict.fromkeys(line for line in lines if line.strip()))
     numbered_content = "\n".join(f"{i+1}. {line}" for i, line in enumerate(unique_lines))
     return numbered_content
@@ -276,7 +276,7 @@ async def load_chat():
 # Claude Search Function
 async def perform_claude_search(query, username):
     if not query.strip() or query == st.session_state.last_transcript:
-        return None, None
+        return None, None, None
     client = anthropic.Anthropic(api_key=anthropic_key)
     response = client.messages.create(
         model="claude-3-sonnet-20240229",
@@ -286,26 +286,25 @@ async def perform_claude_search(query, username):
     result = response.content[0].text
     st.markdown(f"### Claude's Reply 🧠\n{result}")
     
-    # Save to chat history with audio
     voice = FUN_USERNAMES.get(username, "en-US-AriaNeural")
     md_file, audio_file = await save_chat_entry(username, f"Claude Search: {query}\nResponse: {result}", voice, True)
-    return md_file, audio_file
+    return md_file, audio_file, result
 
 # ArXiv Search Function
-async def perform_arxiv_search(query, username):
+async def perform_arxiv_search(query, username, claude_result=None):
     if not query.strip() or query == st.session_state.last_transcript:
         return None, None
-    # 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
+    # Use Claude result if provided, otherwise perform Claude search
+    if claude_result is None:
+        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}")
+    
     enhanced_query = f"{query}\n\n{claude_result}"
     gradio_client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
     refs = gradio_client.predict(
@@ -314,7 +313,6 @@ async def perform_arxiv_search(query, username):
     result = f"🔎 {enhanced_query}\n\n{refs}"
     st.markdown(f"### ArXiv Results 🔍\n{result}")
 
-    # Save to chat history with audio
     voice = FUN_USERNAMES.get(username, "en-US-AriaNeural")
     md_file, audio_file = await save_chat_entry(username, f"ArXiv Search: {query}\nClaude Response: {claude_result}\nArXiv Results: {refs}", voice, True)
     return md_file, audio_file
@@ -514,7 +512,7 @@ def load_votes(file):
         with open(file, 'r') as f:
             lines = f.read().strip().split('\n')
         votes = {}
-        for line in lines[2:]:  # Skip header
+        for line in lines[2:]:
             if line.strip() and 'voted for' in line:
                 item = line.split('voted for ')[1]
                 votes[item] = votes.get(item, 0) + 1
@@ -599,6 +597,7 @@ def main():
         message = st.text_input(f"Message as {st.session_state.username}", key="message_input")
         paste_result = paste_image_button("📋 Paste Image or Text", key="paste_button_msg")
         if paste_result.image_data is not None:
+            voice = FUN_USERNAMES.get(st.session_state.username, "en-US-AriaNeural")
             if isinstance(paste_result.image_data, str):
                 st.session_state.message_text = paste_result.image_data
                 message = st.text_input(f"Message as {st.session_state.username}", key="message_input_paste", value=st.session_state.message_text)
@@ -607,7 +606,20 @@ 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
-        
+                    # Generate a placeholder speech text for ASR-like behavior
+                    asr_text = f"User {st.session_state.username} requested analysis of an image uploaded at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+                    # Run Claude search first
+                    md_file_claude, audio_file_claude, claude_result = asyncio.run(perform_claude_search(asr_text, st.session_state.username))
+                    if audio_file_claude:
+                        play_and_download_audio(audio_file_claude)
+                    # Feed Claude result to ArXiv search
+                    md_file_arxiv, audio_file_arxiv = asyncio.run(perform_arxiv_search(asr_text, st.session_state.username, claude_result))
+                    if audio_file_arxiv:
+                        play_and_download_audio(audio_file_arxiv)
+                    st.session_state.timer_start = time.time()
+                    save_username(st.session_state.username)
+                    st.rerun()
+
         if (message and message != st.session_state.last_message) or st.session_state.pasted_image_data:
             st.session_state.last_message = message
             col_send, col_claude, col_arxiv = st.columns([1, 1, 1])
@@ -630,7 +642,7 @@ def main():
                 if st.button("🧠 Claude", key="claude_button"):
                     voice = FUN_USERNAMES.get(st.session_state.username, "en-US-AriaNeural")
                     if message.strip():
-                        md_file, audio_file = asyncio.run(perform_claude_search(message, st.session_state.username))
+                        md_file, audio_file, _ = asyncio.run(perform_claude_search(message, st.session_state.username))
                         if audio_file:
                             play_and_download_audio(audio_file)
                         st.session_state.timer_start = time.time()
@@ -742,6 +754,57 @@ def main():
         if zip_name:
             st.sidebar.markdown(get_download_link(zip_name, "zip"), unsafe_allow_html=True)
 
+    # Refresh Timer in Sidebar
+    st.sidebar.subheader("Set Refresh Rate ⏳")
+    st.markdown("""
+        <style>
+        .timer {
+            font-size: 24px;
+            color: #ffcc00;
+            text-align: center;
+            animation: pulse 1s infinite;
+        }
+        @keyframes pulse {
+            0% { transform: scale(1); }
+            50% { transform: scale(1.1); }
+            100% { transform: scale(1); }
+        }
+        </style>
+    """, unsafe_allow_html=True)
+
+    refresh_rate = st.sidebar.slider("Refresh Rate (seconds)", min_value=1, max_value=300, value=st.session_state.refresh_rate, step=1)
+    if refresh_rate != st.session_state.refresh_rate:
+        st.session_state.refresh_rate = refresh_rate
+        st.session_state.timer_start = time.time()
+        save_username(st.session_state.username)
+
+    col1, col2, col3 = st.sidebar.columns(3)
+    with col1:
+        if st.button("🐇 Small (1s)"):
+            st.session_state.refresh_rate = 1
+            st.session_state.timer_start = time.time()
+            save_username(st.session_state.username)
+    with col2:
+        if st.button("🐢 Medium (5s)"):
+            st.session_state.refresh_rate = 5
+            st.session_state.timer_start = time.time()
+            save_username(st.session_state.username)
+    with col3:
+        if st.button("🐘 Large (5m)"):
+            st.session_state.refresh_rate = 300
+            st.session_state.timer_start = time.time()
+            save_username(st.session_state.username)
+
+    timer_placeholder = st.sidebar.empty()
+    start_time = st.session_state.timer_start
+    remaining_time = int(st.session_state.refresh_rate - (time.time() - start_time))
+    if remaining_time <= 0:
+        st.session_state.timer_start = time.time()
+        st.session_state.last_refresh = time.time()
+        st.rerun()
+    else:
+        timer_placeholder.markdown(f"<p class='timer'>⏳ Next refresh in: {remaining_time} seconds</p>", unsafe_allow_html=True)
+
     # Start WebSocket server in a separate thread
     if not st.session_state.server_running and not st.session_state.server_task:
         st.session_state.server_task = threading.Thread(target=start_websocket_server, daemon=True)