Update app.py

app.py

@@ -18,7 +18,7 @@ import edge_tts
 from audio_recorder_streamlit import audio_recorder
 import nest_asyncio
 import re
-from streamlit_paste_button import paste_image_button  # Import the new library
+from streamlit_paste_button import paste_image_button  # Using this for image pasting
 
 # Patch for nested async - sneaky fix! 🐍✨
 nest_asyncio.apply()
@@ -280,16 +280,13 @@ def play_and_download_audio(file_path):
         st.markdown(dl_link, unsafe_allow_html=True)
 
 # Image saver - pics preserved with naming! 📸💾
-async def save_pasted_image(image_data, username):
+async def save_pasted_image(image, username):
     await asyncio.to_thread(log_action, username, "📸💾 - Image saver - pics preserved!")
     timestamp = format_timestamp_prefix()
     filename = f"paste_{timestamp}_{username}.png"
     filepath = os.path.join(MEDIA_DIR, filename)
-    if ',' in image_data:
-        image_data = image_data.split(',')[1]
-    img_bytes = base64.b64decode(image_data)
-    img = Image.open(io.BytesIO(img_bytes))
-    await asyncio.to_thread(img.save, filepath, "PNG")
+    # Directly save the PIL Image object without base64 conversion
+    await asyncio.to_thread(image.save, filepath, "PNG")
     return filepath
 
 # Video renderer - movies roll with autoplay! 🎥🎬
@@ -426,11 +423,6 @@ def main():
                 if st.button(f"👍 {vote_count}", key=f"chat_vote_{i}"):
                     comment = st.session_state.message_text
                     await save_vote(QUOTE_VOTES_FILE, line.split('. ')[1] if '. ' in line else line, await generate_user_hash(), st.session_state.username, comment)
-                    if st.session_state.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
                     st.session_state.message_text = ''
                     st.rerun()
             with col3:
@@ -451,14 +443,19 @@ def main():
         if st.session_state.quote_line:
             st.markdown(f"### Quoting: {st.session_state.quote_line}")
             quote_response = st.text_area("Add your response", key="quote_response")
+            # Add paste button within quote response
+            paste_result_quote = paste_image_button("📋 Paste Image with Quote", key="paste_button_quote")
+            if paste_result_quote.image_data is not None:
+                st.image(paste_result_quote.image_data, caption="Received Image for Quote")
+                filename = await save_pasted_image(paste_result_quote.image_data, st.session_state.username)
+                if filename:
+                    st.session_state.pasted_image_data = filename  # Store for use in quote submission
             if st.button("Send Quote 🚀", key="send_quote"):
                 markdown_response = f"### Quote Response\n- **Original**: {st.session_state.quote_line}\n- **{st.session_state.username} Replies**: {quote_response}"
                 if st.session_state.pasted_image_data:
-                    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
+                    markdown_response += f"\n- **Image**: ![Pasted Image]({st.session_state.pasted_image_data})"
+                    await save_chat_entry(st.session_state.username, f"Pasted image: {st.session_state.pasted_image_data}")
+                    st.session_state.pasted_image_data = None
                 await save_chat_entry(st.session_state.username, markdown_response)
                 st.session_state.quote_line = None
                 st.session_state.message_text = ''
@@ -471,56 +468,28 @@ def main():
             st.rerun()
 
         message = st.text_input(f"Message as {st.session_state.username}", key="message_input", value=st.session_state.message_text, on_change=lambda: st.session_state.update(message_text=st.session_state.message_input))
-        if st.button("Send 🚀", key="send_button") and message.strip():
-            await save_chat_entry(st.session_state.username, message)
-            st.session_state.message_text = ''
-            st.rerun()
-
-        # Paste Image Button
-        paste_result = paste_image_button("📋 Paste Image", key="paste_button")
-        if paste_result.image_data is not None:
-            st.image(paste_result.image_data, caption="Received Image")
-            filename = await save_pasted_image(base64.b64encode(paste_result.image_data.tobytes()).decode(), st.session_state.username)
+        # Add paste button next to message input
+        paste_result_msg = paste_image_button("📋 Paste Image with Message", key="paste_button_msg")
+        if paste_result_msg.image_data is not None:
+            st.image(paste_result_msg.image_data, caption="Received Image for Message")
+            filename = await save_pasted_image(paste_result_msg.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 = filename  # Store for use in message submission
+        if st.button("Send 🚀", key="send_button") and (message.strip() or st.session_state.pasted_image_data):
+            if message.strip():
+                await save_chat_entry(st.session_state.username, message)
+            if st.session_state.pasted_image_data:
+                await save_chat_entry(st.session_state.username, f"Pasted image: {st.session_state.pasted_image_data}")
                 st.session_state.pasted_image_data = None
+            st.session_state.message_text = ''
+            st.rerun()
 
         # Main action tabs and model use choices
         tab_main = st.radio("Action:", ["🎤 Voice", "📸 Media", "🔍 ArXiv", "📝 Editor"], horizontal=True)
         useArxiv = st.checkbox("Search Arxiv for Research Paper Answers", value=True)
         useArxivAudio = st.checkbox("Generate Audio File for Research Paper Answers", value=False)
 
-        # Custom component for ASR
-        mycomponent = components.declare_component("mycomponent", path="./frontend")
-        val = mycomponent(my_input_value="Hello from MyComponent")
-
-        if val:
-            if isinstance(val, dict):
-                if val.get('type') == 'text' and 'transcript' in val:
-                    val_stripped = val['transcript'].replace('\n', ' ')
-                    if val_stripped.strip():
-                        await save_chat_entry(st.session_state.username, val_stripped)
-
-            edited_input = st.text_area("✏️ Edit Input:", value=val_stripped if isinstance(val, dict) and val.get('type') == 'text' else "", height=100)
-            run_option = st.selectbox("Model:", ["Arxiv", "Other (demo)"])
-            col1, col2 = st.columns(2)
-            with col1:
-                autorun = st.checkbox("⚙ AutoRun", value=True)
-            with col2:
-                full_audio = st.checkbox("📚FullAudio", value=False)
-
-            input_changed = (val != st.session_state.old_val)
-
-            if autorun and input_changed and edited_input.strip():
-                st.session_state.old_val = val
-                st.session_state.last_query = edited_input
-                await perform_ai_lookup(edited_input, vocal_summary=True, extended_refs=False, titles_summary=True, full_audio=full_audio, useArxiv=useArxiv, useArxivAudio=useArxivAudio)
-            else:
-                if st.button("▶ Run") and edited_input.strip():
-                    st.session_state.old_val = val
-                    st.session_state.last_query = edited_input
-                    await perform_ai_lookup(edited_input, vocal_summary=True, extended_refs=False, titles_summary=True, full_audio=full_audio, useArxiv=useArxiv, useArxivAudio=useArxivAudio)
-
+        # Enhanced Media Gallery with Image, Audio, Video
         st.subheader("Upload Media 🎨🎶🎥")
         uploaded_file = st.file_uploader("Upload Media", type=['png', 'jpg', 'mp4', 'mp3'])
         if uploaded_file:
@@ -535,7 +504,6 @@ def main():
             if file_path.endswith('.mp4'):
                 st.session_state.media_notifications.append(file_path)
 
-        # Enhanced Media Gallery with Image, Audio, Video
         st.subheader("Media Gallery 🎨🎶🎥")
         media_files = glob.glob(f"{MEDIA_DIR}/*.png") + glob.glob(f"{MEDIA_DIR}/*.jpg") + glob.glob(f"{MEDIA_DIR}/*.mp4") + glob.glob(f"{MEDIA_DIR}/*.mp3")
         if media_files: