Update app.py

app.py

@@ -115,6 +115,8 @@ if 'last_chat_update' not in st.session_state:
     st.session_state.last_chat_update = 0
 if 'displayed_chat_lines' not in st.session_state:
     st.session_state.displayed_chat_lines = []
+if 'pasted_image_data_temp' not in st.session_state:
+    st.session_state.pasted_image_data_temp = None
 
 # Timestamp wizardry - clock ticks with flair! ⏰🎩
 def format_timestamp_prefix():
@@ -477,7 +479,7 @@ def create_streamlit_interface():
             st.session_state.message_text = ''
             st.rerun()
 
-        # Image paste component with a key to capture value
+        # Image paste component without key
         components.html(
             """
             <div id="paste-target">Paste an image here (Ctrl+V)</div>
@@ -495,6 +497,11 @@ def create_streamlit_interface():
                                 value: e.target.result
                             }, '*');
                             pasteTarget.innerHTML = '<p>Image pasted! Processing...</p>';
+                            // Trigger a custom event to notify Python
+                            window.parent.postMessage({
+                                type: 'streamlit:imagePasted',
+                                value: e.target.result
+                            }, '*');
                         };
                         reader.readAsDataURL(blob);
                     }
@@ -503,20 +510,19 @@ def create_streamlit_interface():
             });
             </script>
             """,
-            height=100,
-            key="image_paste_component"
+            height=100
         )
 
-        # Check for pasted image data in session state
-        if "image_paste_component" in st.session_state and st.session_state.image_paste_component:
-            pasted_image_data = st.session_state.image_paste_component
-            if isinstance(pasted_image_data, str) and st.session_state.pasted_image_data != pasted_image_data:
+        # Custom JavaScript listener for pasted image (simulated via session state)
+        if 'pasted_image_data_temp' in st.session_state and st.session_state.pasted_image_data_temp and st.session_state.pasted_image_data != st.session_state.pasted_image_data_temp:
+            pasted_image_data = st.session_state.pasted_image_data_temp
+            if isinstance(pasted_image_data, str):
                 st.session_state.pasted_image_data = pasted_image_data
                 filename = await save_pasted_image(st.session_state.pasted_image_data, st.session_state.username)
                 if filename:
                     await save_chat_entry(st.session_state.username, f"Pasted image: {filename}")
                     st.session_state.pasted_image_data = None
-                    del st.session_state.image_paste_component  # Clear the component value after processing
+                    st.session_state.pasted_image_data_temp = None  # Clear after processing
                 st.rerun()
 
         st.subheader("Media Gallery 🎨🎶🎥")