Update app.py

app.py

@@ -127,7 +127,7 @@ def init_session_state():
         '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,
-        'last_refresh': time.time(), 'paper_metadata': {}, 'paste_image_base64': None
+        'last_refresh': time.time(), 'paper_metadata': {}, 'paste_image_base64': ""
     }
     for k, v in defaults.items():
         if k not in st.session_state:
@@ -282,7 +282,6 @@ async def perform_claude_search(query, username, image=None):
     client = anthropic.Anthropic(api_key=anthropic_key)
     message_content = [{"type": "text", "text": query}]
     if image:
-        # Convert PIL Image to base64
         buffered = io.BytesIO()
         image.save(buffered, format="PNG")
         img_base64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
@@ -565,49 +564,43 @@ def create_zip_of_files(md_files, mp3_files, png_files, mp4_files, query):
 
 # Custom Paste Image Component
 def paste_image_component():
-    # JavaScript to capture clipboard image and send it to Streamlit
-    paste_script = """
-    <script>
-    function pasteImage() {
-        navigator.clipboard.read().then(clipboardItems => {
-            for (const item of clipboardItems) {
-                if (item.types.includes('image/png') || item.types.includes('image/jpeg')) {
-                    item.getType('image/png').then(blob => {
-                        const reader = new FileReader();
-                        reader.onload = function(event) {
-                            document.getElementById('paste_input').value = event.target.result.split(',')[1];
-                            document.getElementById('paste_form').submit();
-                        };
-                        reader.readAsDataURL(blob);
-                    });
-                    break;
-                }
-            }
-        }).catch(err => {
-            console.error('Failed to read clipboard: ', err);
-            document.getElementById('paste_input').value = 'ERROR: ' + err.message;
-            document.getElementById('paste_form').submit();
-        });
-    }
-    </script>
-    <form id="paste_form" action="/" method="POST" style="display: none;">
-        <input type="text" id="paste_input" name="paste_data" />
-    </form>
-    <button onclick="pasteImage()">Paste Image 📋</button>
-    """
-    paste_data = components.html(paste_script, height=50)
-    if paste_data and 'paste_data' in paste_data:
-        base64_str = paste_data['paste_data']
-        if base64_str.startswith('ERROR:'):
-            st.warning(f"Paste failed: {base64_str}")
-            return None
-        try:
-            img_bytes = base64.b64decode(base64_str)
-            img = Image.open(io.BytesIO(img_bytes))
-            return img
-        except Exception as e:
-            st.warning(f"Error decoding pasted image: {e}")
-            return None
+    with st.form(key="paste_form"):
+        # Hidden text input to receive clipboard data
+        paste_input = st.text_input("Paste Base64 Image Here (hidden)", value="", key="paste_input", label_visibility="collapsed")
+        # JavaScript to paste clipboard content into the input
+        st.markdown("""
+        <script>
+        function pasteClipboard() {
+            navigator.clipboard.readText().then(text => {
+                document.getElementById('paste_input').value = text;
+                document.getElementById('paste_form').requestSubmit();
+            }).catch(err => {
+                console.error('Failed to read clipboard: ', err);
+                document.getElementById('paste_input').value = 'ERROR: ' + err.message;
+                document.getElementById('paste_form').requestSubmit();
+            });
+        }
+        </script>
+        """, unsafe_allow_html=True)
+        # Button to trigger paste
+        paste_button = st.form_submit_button("Paste Image 📋", on_click=lambda: st.markdown("<script>pasteClipboard();</script>", unsafe_allow_html=True))
+        
+        if paste_button and paste_input:
+            if paste_input.startswith('ERROR:'):
+                st.warning(f"Paste failed: {paste_input}")
+                return None
+            if paste_input.startswith('data:image'):
+                try:
+                    base64_str = paste_input.split(',')[1]
+                    img_bytes = base64.b64decode(base64_str)
+                    img = Image.open(io.BytesIO(img_bytes))
+                    return img
+                except Exception as e:
+                    st.warning(f"Error decoding pasted image: {e}")
+                    return None
+            else:
+                st.warning("Clipboard does not contain a valid image (expected base64 data:image)")
+                return None
     return None
 
 # 🎮 Main Interface
@@ -641,10 +634,10 @@ def main():
             st.rerun()
 
     tab_main = st.radio("Action:", ["🎤 Chat & Voice", "🔍 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")
-    st.checkbox("Autosend Chat", value=True, key="autosend")
-    st.checkbox("Autosearch ArXiv", value=True, key="autosearch")
+    st.checkbox("Search ArXiv", value=st.session_state['use_arxiv'], key="use_arxiv")
+    st.checkbox("ArXiv Audio", value=st.session_state['use_arxiv_audio'], key="use_arxiv_audio")
+    st.checkbox("Autosend Chat", value=st.session_state['autosend'], key="autosend")
+    st.checkbox("Autosearch ArXiv", value=st.session_state['autosearch'], key="autosearch")
 
     # 🎤 Chat & Voice
     if tab_main == "🎤 Chat & Voice":
@@ -659,8 +652,8 @@ def main():
         
         # Custom Paste Image Button
         pasted_image = paste_image_component()
-        if pasted_image is not None and st.session_state['paste_image_base64'] != pasted_image.tobytes():
-            st.session_state['paste_image_base64'] = pasted_image.tobytes()
+        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")
             filename = asyncio.run(save_pasted_image(pasted_image, st.session_state.username))
@@ -679,7 +672,7 @@ def main():
                     if audio_file_arxiv:
                         play_and_download_audio(audio_file_arxiv)
                     st.session_state.pasted_image_data = None
-                    st.session_state['paste_image_base64'] = None
+                    st.session_state['paste_image_base64'] = ""
                     st.session_state.timer_start = time.time()
                     save_username(st.session_state.username)
                     st.rerun()
@@ -731,7 +724,7 @@ def main():
         if q and q != st.session_state.last_query:
             st.session_state.last_query = q
             if st.session_state.autosearch or st.button("🔍 Run", key="arxiv_run"):
-                result, papers = asyncio.run(perform_ai_lookup(q, useArxiv=useArxiv, useArxivAudio=useArxivAudio))
+                result, papers = asyncio.run(perform_ai_lookup(q, useArxiv=st.session_state['use_arxiv'], useArxivAudio=st.session_state['use_arxiv_audio']))
                 st.markdown(f"### Query: {q}")
                 for i, p in enumerate(papers, 1):
                     expander_label = f"{p['title']} | [arXiv Link]({p['url']})"
@@ -874,7 +867,7 @@ def main():
     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
+    # Start WebSocket server in a 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)
         st.session_state.server_task.start()