Update app.py

app.py

@@ -477,8 +477,8 @@ def create_streamlit_interface():
             st.session_state.message_text = ''
             st.rerun()
 
-        # Image paste component
-        pasted_image = components.html(
+        # Image paste component with a key to capture value
+        components.html(
             """
             <div id="paste-target">Paste an image here (Ctrl+V)</div>
             <script>
@@ -503,15 +503,21 @@ def create_streamlit_interface():
             });
             </script>
             """,
-            height=100
+            height=100,
+            key="image_paste_component"
         )
-        if pasted_image and st.session_state.pasted_image_data != pasted_image:
-            st.session_state.pasted_image_data = pasted_image
-            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
-            st.rerun()
+
+        # 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:
+                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.rerun()
 
         st.subheader("Media Gallery 🎨🎶🎥")
         uploaded_file = st.file_uploader("Upload Media", type=['png', 'jpg', 'mp3', 'mp4'])