Update app.py

app.py

@@ -258,6 +258,8 @@ async def save_chat_entry(username, message, is_markdown=False, quote_line=None)
             f.write(f"[{timestamp}] {username} ({voice}): Audio generated - {audio_filename}\n")
         with open(user_history_file, 'a') as f:
             f.write(f"{indent}[{timestamp}] Audio: {audio_filename}\n")
+        with open(CHAT_FILE, 'a') as f:
+            f.write(f"{indent}[{timestamp}] Audio: {audio_filename}\n")
         
         # Embed audio and image in global chat if applicable
         if message.startswith("Pasted image:") or message.startswith("Uploaded media:") or message.startswith("Uploaded PDF:"):
@@ -449,7 +451,12 @@ async def get_audio_html(audio_path, width="100%"):
     username = st.session_state.get('username', 'System 🌟')
     await asyncio.to_thread(log_action, username, "🎶✈️ - Audio renderer - sounds soar!")
     audio_url = f"data:audio/mpeg;base64,{base64.b64encode(await asyncio.to_thread(open, audio_path, 'rb').read()).decode()}"
-    return f'<audio controls style="width: {width};"><source src="{audio_url}" type="audio/mpeg">Your browser does not support the audio element.</audio>'
+    return f'''
+    <audio controls style="width: {width};">
+        <source src="{audio_url}" type="audio/mpeg">
+        Your browser does not support the audio element.
+    </audio>
+    '''
 
 # Websocket handler - chat links up! 🌐🔗
 async def websocket_handler(websocket, path):
@@ -590,15 +597,13 @@ def main():
                                 elif file_path.endswith('.pdf'):
                                     st.write(f"PDF: {os.path.basename(file_path)}")
                     with col_audio:
-                        username = line.split(': ')[1].split(' ')[0] if ': ' in line else "Unknown"
-                        cache_key = f"{line}_{FUN_USERNAMES.get(username, 'en-US-AriaNeural')}"
-                        if cache_key not in st.session_state.audio_cache and "Audio:" in line:
+                        if "Audio:" in line:
                             audio_ref = line.split("Audio: ")[-1].strip()
                             if os.path.exists(audio_ref):
+                                play_and_download_audio(audio_ref)
+                                username = line.split(': ')[1].split(' ')[0] if ': ' in line else "Unknown"
+                                cache_key = f"{line}_{FUN_USERNAMES.get(username, 'en-US-AriaNeural')}"
                                 st.session_state.audio_cache[cache_key] = audio_ref
-                        audio_file = st.session_state.audio_cache.get(cache_key)
-                        if audio_file:
-                            play_and_download_audio(audio_file)
 
         if st.session_state.quote_line:
             st.markdown(f"### Quoting: {st.session_state.quote_line}")
@@ -714,33 +719,30 @@ def main():
             time.sleep(1)
         st.rerun()
 
-        # Gallery with Adjustable Tiles
-        st.subheader("Media Gallery 🎨🎶📜🎥")
-        gallery_columns = st.slider("Number of Gallery Tiles", 1, 20, st.session_state.gallery_columns)
-        st.session_state.gallery_columns = gallery_columns
-        media_files = glob.glob("*.png") + glob.glob("*.jpg") + glob.glob("*.mp4") + glob.glob("*.pdf")
-        if media_files:
-            media_votes = await load_votes(MEDIA_VOTES_FILE)
-            seen_files = set()
-            cols = st.columns(gallery_columns)
-            col_idx = 0
-            for media_file in sorted(media_files, key=os.path.getmtime, reverse=True):
-                if media_file not in seen_files:
-                    seen_files.add(media_file)
-                    with cols[col_idx]:
-                        filename = os.path.basename(media_file)
-                        vote_count = media_votes.get(media_file, 0)
-                        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(get_video_html(media_file), unsafe_allow_html=True)
-                        elif media_file.endswith('.pdf'):
-                            st.write(f"PDF: {filename}")
-                        if st.button(f"👍 {vote_count}", key=f"media_vote_{media_file}"):
-                            await save_vote(MEDIA_VOTES_FILE, media_file, await generate_user_hash(), st.session_state.username)
-                            st.rerun()
-                    col_idx = (col_idx + 1) % gallery_columns
+        # Gallery with Adjustable Tiles - Consistent Three-Set from Provided Code
+        st.subheader("Image Gallery 🖼")
+        image_files = glob.glob("*.png") + glob.glob("*.jpg") + glob.glob("*.jpeg")
+        image_cols = st.slider("Image Gallery Columns 🖼", min_value=1, max_value=15, value=5)
+        cols = st.columns(image_cols)
+        for idx, image_file in enumerate(image_files):
+            with cols[idx % image_cols]:
+                st.image(image_file, use_container_width=True)
+
+        st.subheader("Video Gallery 🎥")
+        video_files = glob.glob("*.mp4")
+        video_cols = st.slider("Video Gallery Columns 🎬", min_value=1, max_value=5, value=3)
+        cols = st.columns(video_cols)
+        for idx, video_file in enumerate(video_files):
+            with cols[idx % video_cols]:
+                st.markdown(get_video_html(video_file, width="100%"), unsafe_allow_html=True)
+
+        st.subheader("Audio Gallery 🎧")
+        audio_files = glob.glob("*.mp3") + glob.glob("*.wav")
+        audio_cols = st.slider("Audio Gallery Columns 🎶", min_value=1, max_value=15, value=5)
+        cols = st.columns(audio_cols)
+        for idx, audio_file in enumerate(audio_files):
+            with cols[idx % audio_cols]:
+                st.markdown(await get_audio_html(audio_file, width="100%"), unsafe_allow_html=True)
 
         # Full Log at End
         st.subheader("Full Chat Log 📜")