Update app.py

app.py

@@ -100,7 +100,7 @@ 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)),
+    ("Negative Squared", lambda x: "".join(chr(ord(c) - 0x1F170 - 0x41) 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)),
 ]
 
@@ -139,6 +139,8 @@ if 'last_transcript' not in st.session_state:
     st.session_state.last_transcript = ""
 if 'image_hashes' not in st.session_state:
     st.session_state.image_hashes = set()
+if 'gallery_columns' not in st.session_state:
+    st.session_state.gallery_columns = 1  # Default gallery tiles
 
 # Timestamp wizardry - clock ticks with flair! ⏰🎩
 def format_timestamp_prefix(username):
@@ -201,11 +203,25 @@ async def save_chat_entry(username, message, is_markdown=False):
     audio_file = await async_edge_tts_generate(cleaned_message, voice)
     if audio_file:
         with open(HISTORY_FILE, 'a') as f:
-            f.write(f"[{timestamp}] {username}: Audio generated - {audio_file}\n")
+            f.write(f"[{timestamp}] {username} ({voice}): Audio generated - {audio_file}\n")
     await broadcast_message(f"{username}|{message}", "chat")
     st.session_state.last_chat_update = time.time()
     return audio_file
 
+# Save chat history with image
+async def save_chat_history_with_image(username, image_path):
+    central = pytz.timezone('US/Central')
+    timestamp = datetime.now(central).strftime("%Y-%m-%d_%H-%M-%S")
+    history_filename = f"chat_history_{timestamp}-by-{username}.md"
+    history_filepath = os.path.join(HISTORY_DIR, history_filename)
+    chat_content = await load_chat()
+    voice = FUN_USERNAMES.get(username, "en-US-AriaNeural")
+    with open(history_filepath, 'w') as f:
+        f.write(f"# Chat History at {timestamp} by {username} (Voice: {voice})\n\n")
+        f.write(f"## Image Shared: {os.path.basename(image_path)}\n")
+        f.write(chat_content)
+    return history_filepath
+
 # Chat loader - history unleashed! 📜🚀
 async def load_chat():
     username = st.session_state.get('username', 'System 🌟')
@@ -323,6 +339,7 @@ async def save_pasted_image(image, username):
     filepath = os.path.join(MEDIA_DIR, filename)
     await asyncio.to_thread(image.save, filepath, "PNG")
     st.session_state.image_hashes.add(img_hash)
+    await save_chat_history_with_image(username, filepath)  # Save chat history with image
     return filepath
 
 # Video renderer - movies roll with autoplay! 🎥🎬
@@ -405,6 +422,33 @@ async def perform_ai_lookup(query, vocal_summary=True, extended_refs=False, titl
         if audio_file:
             st.audio(audio_file)
 
+# Delete all user files function
+def delete_user_files():
+    protected_files = {'app.py', 'requirements.txt', 'README.md'}
+    deleted_files = []
+    directories = [MEDIA_DIR, AUDIO_DIR, CHAT_DIR, VOTE_DIR, HISTORY_DIR]
+    for directory in directories:
+        if os.path.exists(directory):
+            for root, _, files in os.walk(directory):
+                for file in files:
+                    file_path = os.path.join(root, file)
+                    if os.path.basename(file_path) not in protected_files:
+                        try:
+                            os.remove(file_path)
+                            deleted_files.append(file_path)
+                        except Exception as e:
+                            st.error(f"Failed to delete {file_path}: {e}")
+            try:
+                shutil.rmtree(directory, ignore_errors=True)
+                os.makedirs(directory, exist_ok=True)
+            except Exception as e:
+                st.error(f"Failed to remove directory {directory}: {e}")
+    st.session_state.image_hashes.clear()
+    st.session_state.audio_cache.clear()
+    st.session_state.base64_cache.clear()
+    st.session_state.displayed_chat_lines.clear()
+    return deleted_files
+
 # ASR Component HTML
 ASR_HTML = """
 <html>
@@ -569,33 +613,6 @@ ASR_HTML = """
 </html>
 """
 
-# Delete all user files function
-def delete_user_files():
-    protected_files = {'app.py', 'requirements.txt', 'README.md'}
-    deleted_files = []
-    directories = [MEDIA_DIR, AUDIO_DIR, CHAT_DIR, VOTE_DIR, HISTORY_DIR]
-    for directory in directories:
-        if os.path.exists(directory):
-            for root, _, files in os.walk(directory):
-                for file in files:
-                    file_path = os.path.join(root, file)
-                    if os.path.basename(file_path) not in protected_files:
-                        try:
-                            os.remove(file_path)
-                            deleted_files.append(file_path)
-                        except Exception as e:
-                            st.error(f"Failed to delete {file_path}: {e}")
-            try:
-                shutil.rmtree(directory, ignore_errors=True)
-                os.makedirs(directory, exist_ok=True)
-            except Exception as e:
-                st.error(f"Failed to remove directory {directory}: {e}")
-    st.session_state.image_hashes.clear()
-    st.session_state.audio_cache.clear()
-    st.session_state.base64_cache.clear()
-    st.session_state.displayed_chat_lines.clear()
-    return deleted_files
-
 # Main execution - let’s roll! 🎲🚀
 def main():
     NODE_NAME, port = get_node_name()
@@ -633,12 +650,11 @@ def main():
                 st.session_state.last_transcript = transcript
                 st.rerun()
 
-        # Chat History Grid
+        # Unified Chat History at Top
         st.subheader(f"{START_ROOM} Chat History 💬")
         chat_content = await load_chat()
         chat_lines = chat_content.split('\n')
         chat_lines = [line for line in chat_lines if line.strip() and ': ' in line and not line.startswith('#')]
-
         if chat_lines:
             col1, col2 = st.columns([2, 1])
             with col1:
@@ -664,25 +680,6 @@ def main():
                         audio_file = st.session_state.audio_cache.get(cache_key)
                         if audio_file:
                             play_and_download_audio(audio_file)
-            with col2:
-                st.write("### Image Media")
-                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)
-                    seen_files = set()
-                    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)
-                            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(await get_video_html(media_file), unsafe_allow_html=True)
-                            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()
 
         if st.session_state.quote_line:
             st.markdown(f"### Quoting: {st.session_state.quote_line}")
@@ -762,6 +759,7 @@ def main():
                 await asyncio.to_thread(lambda: open(file_path, 'wb').write(uploaded_file.getbuffer()))
                 st.success(f"Uploaded {filename}")
                 await save_chat_entry(username, f"Uploaded media: {file_path}")
+                await save_chat_history_with_image(username, file_path)  # Save chat history with upload
                 st.session_state.image_hashes.add(file_hash)
                 if file_path.endswith('.mp4'):
                     st.session_state.media_notifications.append(file_path)
@@ -787,10 +785,37 @@ def main():
             time.sleep(1)
         st.rerun()
 
-        st.sidebar.subheader("Chat History 📜")
+        # 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(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)
+            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(await get_video_html(media_file), unsafe_allow_html=True)
+                        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
+
+        # Full Log at End
+        st.subheader("Full Chat Log 📜")
         with open(HISTORY_FILE, 'r') as f:
             history_content = f.read()
-        st.sidebar.markdown(history_content)
+        st.markdown(history_content)
 
     loop.run_until_complete(async_interface())