Update app.py

app.py

@@ -98,8 +98,8 @@ UNICODE_FONTS = [
     ("Circled", lambda x: "".join(chr(ord(c) - 0x41 + 0x24B6) if 'A' <= c <= 'Z' else chr(ord(c) - 0x61 + 0x24D0) if 'a' <= c <= 'z' else c for c in x)),
     ("Squared", lambda x: "".join(chr(ord(c) - 0x41 + 0x1F130) if 'A' <= c <= 'Z' else c for c in x)),
     ("Negative Circled", lambda x: "".join(chr(ord(c) - 0x41 + 0x1F150) if 'A' <= c <= 'Z' else c for c in x)),
-    ("Negative Squared", lambda x: "".join(chr(ord(c) - 0x41 + 0x1F170) if 'A' <= c <= 'Z' else c for c in x)),  # Fixed to add offset
-    ("Regional Indicator", lambda x: "".join(chr(ord(c) - 0x41 + 0x1F1E6) if 'A' <= c <= 'Z' else c for c in x)),  # Fixed to add offset
+    ("Negative Squared", lambda x: "".join(chr(ord(c) - 0x41 + 0x1F170) if 'A' <= c <= 'Z' else c for c in x)),
+    ("Regional Indicator", lambda x: "".join(chr(ord(c) - 0x41 + 0x1F1E6) if 'A' <= c <= 'Z' else c for c in x)),
 ]
 
 # Global state - keeping tabs! 🌍📋
@@ -411,10 +411,13 @@ def main():
             col1, col2, col3, col4 = st.columns([3, 1, 1, 2])
             with col1:
                 st.markdown(line)
-                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)
+                if "Pasted image:" in line or "Uploaded media:" in line:
+                    file_path = line.split(': ')[-1].strip()
+                    if os.path.exists(file_path):
+                        if file_path.endswith(('.png', '.jpg')):
+                            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)
             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}"):
@@ -494,6 +497,7 @@ def main():
                     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)"])
@@ -516,7 +520,7 @@ def main():
                     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', 'mp3', 'mp4'])
+        uploaded_file = st.file_uploader("Upload Media", type=['png', 'jpg', 'mp4'])
         if uploaded_file:
             timestamp = format_timestamp_prefix()
             username = st.session_state.username
@@ -529,38 +533,37 @@ def main():
             if file_path.endswith('.mp4'):
                 st.session_state.media_notifications.append(file_path)
 
-        media_files = glob.glob(f"{MEDIA_DIR}/*.png") + glob.glob(f"{MEDIA_DIR}/*.jpg") + glob.glob(f"{MEDIA_DIR}/*.mp3") + glob.glob(f"{MEDIA_DIR}/*.mp4")
+        # Enhanced Media Gallery
+        media_files = glob.glob(f"{MEDIA_DIR}/*.png") + glob.glob(f"{MEDIA_DIR}/*.jpg") + glob.glob(f"{MEDIA_DIR}/*.mp4")
         if media_files:
             media_votes = await load_votes(MEDIA_VOTES_FILE)
-            users = sorted(set(f.split('_')[-1].split('.')[0] for f in media_files))
-            for user in users:
-                with st.expander(f"{user}'s Media"):
-                    user_files = [f for f in media_files if user in f]
-                    for media_file in user_files:
-                        vote_count = media_votes.get(media_file, 0)
-                        if vote_count > 0:
-                            col1, col2, col3 = st.columns([3, 1, 1])
-                            with col1:
-                                if media_file.endswith(('.png', '.jpg')):
-                                    st.image(media_file, use_container_width=True, caption=os.path.basename(media_file))
-                                elif media_file.endswith('.mp3'):
-                                    st.markdown(await get_audio_html(media_file), unsafe_allow_html=True)
-                                    st.caption(os.path.basename(media_file))
-                                elif media_file.endswith('.mp4'):
-                                    st.markdown(await get_video_html(media_file), unsafe_allow_html=True)
-                                    st.caption(os.path.basename(media_file))
-                                    if media_file in st.session_state.media_notifications:
-                                        st.session_state.media_notifications.remove(media_file)
-                            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)
-                                    st.rerun()
+            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])
+                with col1:
+                    st.markdown(f"**{filename}** - Uploaded by {uploader} at {timestamp}")
+                    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)
+                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)
+                        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)
@@ -570,7 +573,7 @@ def main():
             font_name, font_func = random.choice(UNICODE_FONTS)
             countdown_str = "".join(UNICODE_DIGITS[int(d)] for d in str(i)) if i < 10 else font_func(str(i))
             timer_placeholder.markdown(f"<p class='timer'>⏳ {font_func('Refresh in:')} {countdown_str}</p>", unsafe_allow_html=True)
-            time.sleep(1)  # Use synchronous sleep instead of asyncio.sleep
+            time.sleep(1)  # Use synchronous sleep
         st.rerun()
 
         st.sidebar.subheader("Chat History 📜")