Update app.py

app.py

@@ -18,6 +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
 
 # Patch for nested async - sneaky fix! 🐍✨
 nest_asyncio.apply()
@@ -418,6 +419,8 @@ def main():
                             st.image(file_path, use_container_width=True)
                         elif file_path.endswith('.mp4'):
                             st.markdown(await get_video_html(file_path), unsafe_allow_html=True)
+                        elif file_path.endswith('.mp3'):
+                            st.markdown(await get_audio_html(file_path), unsafe_allow_html=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}"):
@@ -470,34 +473,33 @@ def main():
         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)
-            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()
 
-        # Custom component for ASR and Image Pasting
-        mycomponent = components.declare_component("mycomponent", path="./frontend")
-        val = mycomponent(my_input_value="Hello from MyComponent")
+        # 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)
+            if filename:
+                await save_chat_entry(st.session_state.username, f"Pasted image: {filename}")
+                st.session_state.pasted_image_data = None
 
         # 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)
-                elif val.get('type') == 'image' and 'image' in val:
-                    filename = await save_pasted_image(val['image'], 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
 
             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)"])
@@ -519,8 +521,8 @@ def main():
                     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)
 
-        st.subheader("Media Gallery 🎨🎶🎥")
-        uploaded_file = st.file_uploader("Upload Media", type=['png', 'jpg', 'mp4'])
+        st.subheader("Upload Media 🎨🎶🎥")
+        uploaded_file = st.file_uploader("Upload Media", type=['png', 'jpg', 'mp4', 'mp3'])
         if uploaded_file:
             timestamp = format_timestamp_prefix()
             username = st.session_state.username
@@ -533,37 +535,29 @@ def main():
             if file_path.endswith('.mp4'):
                 st.session_state.media_notifications.append(file_path)
 
-        # Enhanced Media Gallery
-        media_files = glob.glob(f"{MEDIA_DIR}/*.png") + glob.glob(f"{MEDIA_DIR}/*.jpg") + glob.glob(f"{MEDIA_DIR}/*.mp4")
+        # 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:
             media_votes = await load_votes(MEDIA_VOTES_FILE)
             st.write("### All Media Uploads")
             for media_file in sorted(media_files, key=os.path.getmtime, reverse=True):  # Sort by modification time, newest first
                 filename = os.path.basename(media_file)
-                timestamp_str, uploader = filename.split('_', 1)
-                uploader = uploader.rsplit('.', 1)[0]  # Remove extension
-                timestamp = datetime.strptime(timestamp_str, "%Y%m%d%H%M%S").strftime("%Y-%m-%d %H:%M:%S")
                 vote_count = media_votes.get(media_file, 0)
                 
-                col1, col2, col3, col4 = st.columns([3, 1, 1, 2])
+                col1, col2 = st.columns([3, 1])
                 with col1:
-                    st.markdown(f"**{filename}** - Uploaded by {uploader} at {timestamp}")
+                    st.markdown(f"**{filename}**")
                     if media_file.endswith(('.png', '.jpg')):
                         st.image(media_file, use_container_width=True)
                     elif media_file.endswith('.mp4'):
                         st.markdown(await get_video_html(media_file), unsafe_allow_html=True)
+                    elif media_file.endswith('.mp3'):
+                        st.markdown(await get_audio_html(media_file), unsafe_allow_html=True)
                 with col2:
                     if st.button(f"👍 {vote_count}", key=f"media_vote_{media_file}"):
-                        comment = st.session_state.message_text
-                        await save_vote(MEDIA_VOTES_FILE, media_file, await generate_user_hash(), st.session_state.username, comment)
-                        st.session_state.message_text = ''
-                        st.rerun()
-                with col3:
-                    if st.button("🗑️", key=f"media_delete_{media_file}"):
-                        await asyncio.to_thread(os.remove, media_file)
+                        await save_vote(MEDIA_VOTES_FILE, media_file, await generate_user_hash(), st.session_state.username)
                         st.rerun()
-                with col4:
-                    pass  # Placeholder for potential future use
 
         st.subheader("Refresh ⏳")
         refresh_rate = st.slider("Refresh Rate", 1, 300, st.session_state.refresh_rate)