Update app.py

app.py

@@ -409,6 +409,11 @@ def create_streamlit_interface():
             col1, col2, col3, col4 = st.columns([3, 1, 1, 2])
             with col1:
                 st.markdown(line)
+                # Display image if the line contains an image reference
+                if "Pasted image:" in line:
+                    image_path = line.split("Pasted image: ")[1].strip()
+                    if os.path.exists(image_path):
+                        st.image(image_path, use_container_width=True)
             with col2:
                 vote_count = chat_votes.get(line.split('. ')[1] if '. ' in line else line, 0)
                 if st.button(f"👍 {vote_count}", key=f"chat_vote_{i}"):
@@ -446,6 +451,7 @@ def create_streamlit_interface():
                     filename = await save_pasted_image(st.session_state.pasted_image_data, st.session_state.username)
                     if filename:
                         markdown_response += f"\n- **Image**: ![Pasted Image]({filename})"
+                        await save_chat_entry(st.session_state.username, f"Pasted image: {filename}")
                         st.session_state.pasted_image_data = None
                 await save_chat_entry(st.session_state.username, markdown_response)
                 del st.session_state.quote_line
@@ -464,14 +470,15 @@ def create_streamlit_interface():
         if st.button("Send 🚀", key="send_button") and message.strip():
             loop.run_until_complete(save_chat_entry(st.session_state.username, message))
             if st.session_state.pasted_image_data:
-                filename = loop.run_until_complete(save_pasted_image(st.session_state.pasted_image_data, st.session_state.username))
+                filename = await save_pasted_image(st.session_state.pasted_image_data, st.session_state.username)
                 if filename:
                     loop.run_until_complete(save_chat_entry(st.session_state.username, f"Pasted image: {filename}"))
                     st.session_state.pasted_image_data = None
             st.session_state.message_text = ''
             st.rerun()
 
-        components.html(
+        # Image paste component
+        pasted_image = components.html(
             """
             <div id="paste-target">Paste an image here (Ctrl+V)</div>
             <script>
@@ -498,6 +505,13 @@ def create_streamlit_interface():
             """,
             height=100
         )
+        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()
 
         st.subheader("Media Gallery 🎨🎶🎥")
         uploaded_file = st.file_uploader("Upload Media", type=['png', 'jpg', 'mp3', 'mp4'])