Update app.py

app.py

@@ -261,7 +261,6 @@ async def save_chat_entry(username, message, voice, is_markdown=False):
         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)] = (len(st.session_state['chat_history']) + 1, audio_file)
-        # Clear the text input if it matches the message and this was from speech
         if st.session_state.get('speech_processed', False) and st.session_state.get('message_input', '') == message:
             st.session_state['message_input'] = ""
             st.session_state['speech_processed'] = False
@@ -299,13 +298,17 @@ async def perform_claude_search(query, username, image=None):
                 "data": img_base64
             }
         })
-    response = client.messages.create(
-        model="claude-3-sonnet-20240229",
-        max_tokens=1000,
-        messages=[{"role": "user", "content": message_content}]
-    )
-    result = response.content[0].text
-    st.markdown(f"### Claude's Reply 🧠\n{result}")
+    try:
+        response = client.messages.create(
+            model="claude-3-sonnet-20240229",
+            max_tokens=1000,
+            messages=[{"role": "user", "content": message_content}]
+        )
+        result = response.content[0].text
+        st.markdown(f"### Claude's Reply 🧠\n{result}")
+    except Exception as e:
+        st.error(f"Claude processing failed: {e}")
+        return None, None, None
     
     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)
@@ -554,10 +557,14 @@ async def save_pasted_image(image, username, prompt=""):
     timestamp = format_timestamp_prefix(username)
     filename = f"{timestamp}-{clean_text_for_filename(context)}-{img_hash}.png"
     filepath = filename
-    image.save(filepath, "PNG")
-    st.session_state.image_hashes.add(img_hash)
-    await save_chat_entry(username, f"Pasted image: {filepath}", FUN_USERNAMES.get(username, "en-US-AriaNeural"))
-    return filepath
+    try:
+        image.save(filepath, "PNG")
+        st.session_state.image_hashes.add(img_hash)
+        await save_chat_entry(username, f"Pasted image saved: {filepath}", FUN_USERNAMES.get(username, "en-US-AriaNeural"))
+        return filepath
+    except Exception as e:
+        st.error(f"Failed to save image: {e}")
+        return None
 
 # 📦 Zip Files
 def create_zip_of_files(files, prefix="All", query="latest"):
@@ -598,9 +605,10 @@ def paste_image_component():
                     base64_str = paste_input.split(',')[1]
                     img_bytes = base64.b64decode(base64_str)
                     img = Image.open(io.BytesIO(img_bytes))
+                    st.image(img, caption="Image Pasted Successfully", use_column_width=True)  # Immediate feedback
                     return img
                 except Exception as e:
-                    st.warning(f"Error decoding pasted image: {e}")
+                    st.error(f"Error decoding pasted image: {e}")
                     return None
             else:
                 st.warning("Clipboard does not contain a valid image (expected base64 data:image)")
@@ -654,25 +662,30 @@ def main():
         message = st.text_input(f"Message as {st.session_state.username}", key="message_input")
         
         pasted_image = paste_image_component()
-        if pasted_image is not None and st.session_state['paste_image_base64'] != base64.b64encode(pasted_image.tobytes()).decode('utf-8'):
-            st.session_state['paste_image_base64'] = base64.b64encode(pasted_image.tobytes()).decode('utf-8')
-            voice = FUN_USERNAMES.get(st.session_state.username, "en-US-AriaNeural")
-            st.image(pasted_image, caption="Pasted Image")
-            image_prompt = st.text_input("Add a prompt for Claude (e.g., 'OCR this image')", key="image_prompt")
-            filename = asyncio.run(save_pasted_image(pasted_image, st.session_state.username, image_prompt))
-            if filename:
-                st.session_state.pasted_image_data = filename
-                if image_prompt:
-                    md_file_claude, audio_file_claude, claude_result = asyncio.run(
-                        perform_claude_search(image_prompt, st.session_state.username, paved_image)
-                    )
-                    if audio_file_claude:
-                        play_and_download_audio(audio_file_claude)
-                    md_file_arxiv, audio_file_arxiv = asyncio.run(
-                        perform_arxiv_search(image_prompt, st.session_state.username, claude_result)
-                    )
-                    if audio_file_arxiv:
-                        play_and_download_audio(audio_file_arxiv)
+        if pasted_image is not None:
+            if st.session_state['paste_image_base64'] != base64.b64encode(pasted_image.tobytes()).decode('utf-8'):
+                st.session_state['paste_image_base64'] = base64.b64encode(pasted_image.tobytes()).decode('utf-8')
+                voice = FUN_USERNAMES.get(st.session_state.username, "en-US-AriaNeural")
+                image_prompt = st.text_input("Add a prompt for Claude (e.g., 'OCR this image')", key="image_prompt", value="")
+                with st.spinner("Saving image..."):
+                    filename = asyncio.run(save_pasted_image(pasted_image, st.session_state.username, image_prompt))
+                if filename:
+                    st.session_state.pasted_image_data = filename
+                    st.success(f"Image saved as: {filename}")
+                    if image_prompt:
+                        with st.spinner("Processing with Claude..."):
+                            md_file_claude, audio_file_claude, claude_result = asyncio.run(
+                                perform_claude_search(image_prompt, st.session_state.username, pasted_image)
+                            )
+                            if audio_file_claude:
+                                play_and_download_audio(audio_file_claude)
+                            if claude_result:
+                                with st.spinner("Searching ArXiv..."):
+                                    md_file_arxiv, audio_file_arxiv = asyncio.run(
+                                        perform_arxiv_search(image_prompt, st.session_state.username, claude_result)
+                                    )
+                                    if audio_file_arxiv:
+                                        play_and_download_audio(audio_file_arxiv)
                     st.session_state.pasted_image_data = None
                     st.session_state['paste_image_base64'] = ""
                     st.session_state.timer_start = time.time()