Update app.py

app.py

@@ -133,7 +133,7 @@ if 'user_hash' not in st.session_state:
 def format_timestamp_prefix(username):
     central = pytz.timezone('US/Central')
     now = datetime.now(central)
-    return f"{now.strftime('%I-%M-%p-ct-%m-%d-%Y')}-by-{username}"
+    return f"{now.strftime('%I-%M-%p-ct-%m-%d-%Y')}-by-{username}-{st.session_state.user_id}"
 
 # Compute image hash from binary data
 def compute_image_hash(image_data):
@@ -385,7 +385,7 @@ async def async_edge_tts_generate(text, voice, rate=0, pitch=0, file_format="mp3
         return None
 
 # Audio player - tunes blast off! 🔊🚀
-def play_and_download_audio(file_path, timestamp):
+def play_and_download_audio(file_path):
     if file_path and os.path.exists(file_path):
         with open(file_path, "rb") as f:
             audio_data = f.read()
@@ -394,9 +394,10 @@ def play_and_download_audio(file_path, timestamp):
         <audio controls style="display:inline; vertical-align:middle;">
             <source src="data:audio/mpeg;base64,{b64}" type="audio/mpeg">
         </audio>
-        <a href="data:audio/mpeg;base64,{b64}" download="{os.path.basename(file_path)}" style="vertical-align:middle;">🎵 Download</a>
+        <a href="data:audio/mpeg;base64,{b64}" download="{os.path.basename(file_path)}" style="vertical-align:middle;">🎵</a>
         '''
-        st.markdown(f"{timestamp} {audio_html}", unsafe_allow_html=True)
+        return audio_html
+    return ""
 
 # Image saver - pics preserved with naming! 📸💾
 async def save_pasted_image(image, username):
@@ -535,6 +536,40 @@ def check_query_params():
         st.session_state.user_id = q_value  # Use as user_id if not a valid username
     return q_value
 
+# Mermaid graph generator
+def generate_mermaid_graph(chat_lines):
+    mermaid_code = "graph TD\n"
+    nodes = {}
+    edges = []
+    for i, line in enumerate(chat_lines):
+        if line.strip():
+            timestamp = line.split('] ')[0][1:] if '] ' in line else "Unknown"
+            minute_key = timestamp[:16]  # Up to minute
+            content = line.split(': ', 1)[1] if ': ' in line else line
+            user = content.split(' ')[0] if ' ' in content else "System"
+            node_id = f"{user}_{i}"
+            nodes[node_id] = f"{user} @ {timestamp}"
+            
+            if "Audio:" in line:
+                audio_file = line.split("Audio: ")[-1].strip()
+                nodes[f"audio_{i}"] = f"🎵 {os.path.basename(audio_file)}"
+                edges.append(f"{node_id} --> audio_{i}")
+            elif "Media:" in line:
+                media_file = line.split("Media: ")[-1].strip('![]()')
+                media_type = "🖼" if media_file.endswith(('.png', '.jpg')) else "🎥" if media_file.endswith('.mp4') else "📜"
+                nodes[f"media_{i}"] = f"{media_type} {os.path.basename(media_file)}"
+                edges.append(f"{node_id} --> media_{i}")
+            elif "> " in line:  # Quote
+                parent_line = chat_lines[i-1] if i > 0 else None
+                if parent_line:
+                    parent_id = f"{parent_line.split(': ')[1].split(' ')[0]}_{i-1}"
+                    edges.append(f"{parent_id} --> {node_id}")
+
+    for node_id, label in nodes.items():
+        mermaid_code += f"    {node_id}[\"{label}\"]\n"
+    mermaid_code += "\n".join(f"    {edge}" for edge in edges)
+    return f"```mermaid\n{mermaid_code}\n```"
+
 # Main execution - let’s roll! 🎲🚀
 def main():
     NODE_NAME, port = get_node_name()
@@ -571,35 +606,45 @@ def main():
         if not st.session_state.server_task:
             st.session_state.server_task = loop.create_task(run_websocket_server())
 
-        # Unified Chat History at Top
+        # Unified Chat History at Top with Markdown Emoji Output
         st.subheader(f"{START_ROOM} Chat History 💬")
         chat_content = await load_chat()
         chat_lines = [line for line in chat_content.split('\n') if line.strip() and not line.startswith('#')]
-        current_entry = None
         if chat_lines:
-            col1, col2 = st.columns([2, 1])
-            with col1:
-                st.write("### Text & Audio Chat")
-                for line in reversed(chat_lines):
-                    if not line.startswith('  '):  # Main message line
-                        current_entry = line
-                        st.markdown(line)
-                    elif "Audio:" in line and current_entry:
-                        audio_ref = line.split("Audio: ")[-1].strip()
-                        if os.path.exists(audio_ref):
-                            timestamp = current_entry.split('] ')[0] + ']'
-                            play_and_download_audio(audio_ref, timestamp)
-                    elif "Media:" in line and current_entry:
-                        file_path = line.split("Media: ")[-1].strip('![]()')
-                        if os.path.exists(file_path):
-                            timestamp = current_entry.split('] ')[0] + ']'
-                            st.markdown(f"{timestamp}")
-                            if file_path.endswith(('.png', '.jpg')):
-                                st.image(file_path, use_container_width=True)
-                            elif file_path.endswith('.mp4'):
-                                st.markdown(get_video_html(file_path), unsafe_allow_html=True)
-                            elif file_path.endswith('.pdf'):
-                                st.write(f"PDF: {os.path.basename(file_path)}")
+            chat_by_minute = {}
+            for line in reversed(chat_lines):
+                timestamp = line.split('] ')[0][1:] if '] ' in line else "Unknown"
+                minute_key = timestamp[:16]  # Up to minute
+                if minute_key not in chat_by_minute:
+                    chat_by_minute[minute_key] = []
+                chat_by_minute[minute_key].append(line)
+
+            markdown_output = ""
+            for minute, lines in chat_by_minute.items():
+                minute_output = f"### {minute}\n"
+                for line in lines:
+                    if ': ' in line and not line.startswith('  '):
+                        user_message = line.split(': ', 1)[1]
+                        minute_output += f"- 💬 **{user_message.split(' ')[0]}**: {user_message.split(' ', 1)[1] if ' ' in user_message else ''}\n"
+                    elif "Audio:" in line:
+                        audio_file = line.split("Audio: ")[-1].strip()
+                        audio_html = play_and_download_audio(audio_file)
+                        minute_output += f"  - 🎵 {audio_html}\n"
+                    elif "Media:" in line:
+                        media_file = line.split("Media: ")[-1].strip('![]()')
+                        if media_file.endswith(('.png', '.jpg')):
+                            minute_output += f"  - 🖼 <img src='file://{media_file}' width='100'>\n"
+                        elif media_file.endswith('.mp4'):
+                            minute_output += f"  - 🎥 <video src='file://{media_file}' width='100' controls autoplay muted loop></video>\n"
+                        elif media_file.endswith('.pdf'):
+                            minute_output += f"  - 📜 {os.path.basename(media_file)}\n"
+                markdown_output += minute_output
+            st.markdown(markdown_output, unsafe_allow_html=True)
+
+            # Mermaid Graph Visualization
+            st.subheader("Chat Relationship Tree 🌳")
+            mermaid_graph = generate_mermaid_graph(chat_lines)
+            st.markdown(mermaid_graph)
 
         if st.session_state.quote_line:
             st.markdown(f"### Quoting: {st.session_state.quote_line}")
@@ -614,7 +659,7 @@ def main():
                     filename = await save_pasted_image(paste_result_quote.image_data, st.session_state.username)
                     if filename:
                         st.session_state.pasted_image_data = filename
-                        await save_chat_entry(st.session_state.username, f"Pasted image: {filename}", quote_line=st.session_state.quote_line)
+                        await save_chat_entry(st.session_state.username, f"Pasted image: {filename}", quote_line=st.session_state.quote_line, media_file=filename)
             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:
@@ -669,7 +714,7 @@ def main():
         useArxivAudio = st.checkbox("Generate Audio File for Research Paper Answers", value=False)
 
         st.subheader("Upload Media 🎨🎶📜🎥")
-        uploaded_file = st.file_uploader("Upload Media", type=['png', 'jpg', 'mp4', 'mp3', 'wav', 'pdf'])
+        uploaded_file = st.file_uploader("Upload Media", type=['png', 'jpg', 'mp4', 'mp3', 'wav', 'pdf', 'txt', 'md', 'py'])
         if uploaded_file:
             timestamp = format_timestamp_prefix(st.session_state.username)
             username = st.session_state.username
@@ -715,36 +760,67 @@ def main():
             time.sleep(1)
         st.rerun()
 
-        # Consistent Three-Gallery Set
+        # Separate Galleries for Own and Shared Files
+        all_files = glob.glob("*.md") + glob.glob("*.pdf") + glob.glob("*.txt") + glob.glob("*.py") + glob.glob("*.png") + glob.glob("*.jpg") + glob.glob("*.mp3") + glob.glob("*.mp4")
+        own_files = [f for f in all_files if st.session_state.user_id in f]
+        shared_files = [f for f in all_files if f not in own_files and not f in [CHAT_FILE, QUOTE_VOTES_FILE, MEDIA_VOTES_FILE, HISTORY_FILE, STATE_FILE]]
+
+        st.subheader("Your Files 📂")
+        st.subheader("Image Gallery 🖼")
+        own_image_files = [f for f in own_files if f.endswith(('.png', '.jpg'))]
+        image_cols = st.slider("Image Gallery Columns 🖼 (Own)", min_value=1, max_value=15, value=5)
+        cols = st.columns(image_cols)
+        for idx, image_file in enumerate(own_image_files):
+            with cols[idx % image_cols]:
+                st.image(image_file, use_container_width=True)
+
+        st.subheader("Video Gallery 🎥")
+        own_video_files = [f for f in own_files if f.endswith('.mp4')]
+        video_cols = st.slider("Video Gallery Columns 🎬 (Own)", min_value=1, max_value=5, value=3)
+        cols = st.columns(video_cols)
+        for idx, video_file in enumerate(own_video_files):
+            with cols[idx % video_cols]:
+                st.markdown(get_video_html(video_file, width="100%"), unsafe_allow_html=True)
+
+        st.subheader("Audio Gallery 🎧")
+        own_audio_files = [f for f in own_files if f.endswith(('.mp3', '.wav'))]
+        audio_cols = st.slider("Audio Gallery Columns 🎶 (Own)", min_value=1, max_value=15, value=5)
+        cols = st.columns(audio_cols)
+        for idx, audio_file in enumerate(own_audio_files):
+            with cols[idx % audio_cols]:
+                st.markdown(await get_audio_html(audio_file, width="100%"), unsafe_allow_html=True)
+
+        st.subheader("Shared Files 📤")
         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)
+        shared_image_files = [f for f in shared_files if f.endswith(('.png', '.jpg'))]
+        image_cols = st.slider("Image Gallery Columns 🖼 (Shared)", min_value=1, max_value=15, value=5)
         cols = st.columns(image_cols)
-        for idx, image_file in enumerate(image_files):
+        for idx, image_file in enumerate(shared_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)
+        shared_video_files = [f for f in shared_files if f.endswith('.mp4')]
+        video_cols = st.slider("Video Gallery Columns 🎬 (Shared)", min_value=1, max_value=5, value=3)
         cols = st.columns(video_cols)
-        for idx, video_file in enumerate(video_files):
+        for idx, video_file in enumerate(shared_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)
+        shared_audio_files = [f for f in shared_files if f.endswith(('.mp3', '.wav'))]
+        audio_cols = st.slider("Audio Gallery Columns 🎶 (Shared)", min_value=1, max_value=15, value=5)
         cols = st.columns(audio_cols)
-        for idx, audio_file in enumerate(audio_files):
+        for idx, audio_file in enumerate(shared_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
+        # Full Log at End with Download
         st.subheader("Full Chat Log 📜")
-        with open(HISTORY_FILE, 'r') as f:
+        with open(CHAT_FILE, 'r') as f:
             history_content = f.read()
         st.markdown(history_content)
+        st.download_button("Download Chat Log as .md", history_content, file_name=f"chat_{st.session_state.user_id}.md", mime="text/markdown")
 
     loop.run_until_complete(async_interface())