Update app.py

app.py

@@ -64,7 +64,7 @@ VOTE_DIR = "vote_logs"
 STATE_FILE = "user_state.txt"
 AUDIO_DIR = "audio_logs"
 HISTORY_DIR = "history_logs"
-MEDIA_DIR = "media_files"  # New directory for uploaded media
+MEDIA_DIR = "media_files"
 os.makedirs(CHAT_DIR, exist_ok=True)
 os.makedirs(VOTE_DIR, exist_ok=True)
 os.makedirs(AUDIO_DIR, exist_ok=True)
@@ -111,6 +111,10 @@ if 'active_connections' not in st.session_state:
     st.session_state.active_connections = {}
 if 'media_notifications' not in st.session_state:
     st.session_state.media_notifications = []
+if 'last_chat_update' not in st.session_state:
+    st.session_state.last_chat_update = 0
+if 'displayed_chat_lines' not in st.session_state:
+    st.session_state.displayed_chat_lines = []
 
 # Timestamp wizardry - clock ticks with flair! ⏰🎩
 def format_timestamp_prefix():
@@ -158,6 +162,7 @@ async def save_chat_entry(username, message):
         with open(HISTORY_FILE, 'a') as f:
             f.write(f"[{timestamp}] {username}: Audio generated - {audio_file}\n")
     await broadcast_message(f"{username}|{message}", "chat")  # Notify all clients
+    st.session_state.last_chat_update = time.time()  # Update timestamp for new chat
 
 # Chat loader - history unleashed! 📜🚀
 async def load_chat():
@@ -366,8 +371,6 @@ def create_streamlit_interface():
             st.session_state.message_text = ""
         if 'audio_cache' not in st.session_state:
             st.session_state.audio_cache = {}
-        if 'chat_history' not in st.session_state:
-            st.session_state.chat_history = []
 
         st.markdown("""
             <style>
@@ -389,43 +392,52 @@ def create_streamlit_interface():
             await process_voice_input(audio_bytes)
             st.rerun()
 
+        # Load chat only if there’s an update
         st.subheader(f"{START_ROOM} Chat 💬")
         chat_content = await load_chat()
         chat_lines = chat_content.split('\n')
         chat_votes = await load_votes(QUOTE_VOTES_FILE)
-        for i, line in enumerate(chat_lines):
-            if line.strip() and ': ' in line:
-                col1, col2, col3, col4 = st.columns([3, 1, 1, 2])
-                with col1:
-                    st.markdown(line)
-                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}"):
-                        comment = st.session_state.message_text
-                        await save_vote(QUOTE_VOTES_FILE, line.split('. ')[1] if '. ' in line else line, await generate_user_hash(), st.session_state.username, comment)
-                        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()
-                with col3:
-                    if st.button("📢 Quote", key=f"quote_{i}"):
-                        st.session_state.quote_line = line
-                        st.rerun()
-                with col4:
-                    username = line.split(': ')[1].split(' ')[0]
-                    audio_file = None
-                    cache_key = f"{line}_{FUN_USERNAMES.get(username, 'en-US-AriaNeural')}"
-                    if cache_key in st.session_state.audio_cache:
-                        audio_file = st.session_state.audio_cache[cache_key]
-                    else:
-                        cleaned_text = clean_text_for_tts(line.split(': ', 1)[1])
-                        audio_file = await async_edge_tts_generate(cleaned_text, FUN_USERNAMES.get(username, "en-US-AriaNeural"))
-                        st.session_state.audio_cache[cache_key] = audio_file
-                    if audio_file:
-                        play_and_download_audio(audio_file)
+
+        # Only add new lines that haven’t been displayed yet
+        current_time = time.time()
+        if current_time - st.session_state.last_chat_update > 1 or not st.session_state.displayed_chat_lines:
+            new_lines = [line for line in chat_lines if line.strip() and ': ' in line and line not in st.session_state.displayed_chat_lines]
+            st.session_state.displayed_chat_lines.extend(new_lines)
+            st.session_state.last_chat_update = current_time
+
+        # Display chat lines from session state
+        for i, line in enumerate(st.session_state.displayed_chat_lines):
+            col1, col2, col3, col4 = st.columns([3, 1, 1, 2])
+            with col1:
+                st.markdown(line)
+            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}"):
+                    comment = st.session_state.message_text
+                    await save_vote(QUOTE_VOTES_FILE, line.split('. ')[1] if '. ' in line else line, await generate_user_hash(), st.session_state.username, comment)
+                    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()
+            with col3:
+                if st.button("📢 Quote", key=f"quote_{i}"):
+                    st.session_state.quote_line = line
+                    st.rerun()
+            with col4:
+                username = line.split(': ')[1].split(' ')[0]
+                audio_file = None
+                cache_key = f"{line}_{FUN_USERNAMES.get(username, 'en-US-AriaNeural')}"
+                if cache_key in st.session_state.audio_cache:
+                    audio_file = st.session_state.audio_cache[cache_key]
+                else:
+                    cleaned_text = clean_text_for_tts(line.split(': ', 1)[1])
+                    audio_file = await async_edge_tts_generate(cleaned_text, FUN_USERNAMES.get(username, "en-US-AriaNeural"))
+                    st.session_state.audio_cache[cache_key] = audio_file
+                if audio_file:
+                    play_and_download_audio(audio_file)
 
         if 'quote_line' in st.session_state:
             st.markdown(f"### Quoting: {st.session_state.quote_line}")