Update app.py

app.py

@@ -154,9 +154,50 @@ def load_username():
             return f.read().strip()
     return None
 
-# ---------------------
-# New Pair of Helper Functions
-# ---------------------
+# -----------------------------------------------------------
+# Define PlayerAgent class (used in session state initialization)
+# -----------------------------------------------------------
+class PlayerAgent:
+    def __init__(self, username, char_data):
+        self.username = username
+        self.x = random.uniform(-20, 20)
+        self.z = random.uniform(-40, 40)
+        self.color = char_data["color"]
+        self.shape = char_data["shape"]
+        self.score = 0
+        self.treasures = 0
+        self.last_active = time.time()
+        self.voice = char_data["voice"]
+
+    def to_dict(self):
+        return {
+            "username": self.username,
+            "x": self.x,
+            "z": self.z,
+            "color": self.color,
+            "shape": self.shape,
+            "score": self.score,
+            "treasures": self.treasures,
+            "last_active": self.last_active
+        }
+
+    def update_from_message(self, message):
+        if '|' in message:
+            _, content = message.split('|', 1)
+            if content.startswith("MOVE:"):
+                _, x, z = content.split(":")
+                self.x, self.z = float(x), float(z)
+                self.last_active = time.time()
+            elif content.startswith("SCORE:"):
+                self.score = int(content.split(":")[1])
+                self.last_active = time.time()
+            elif content.startswith("TREASURE:"):
+                self.treasures = int(content.split(":")[1])
+                self.last_active = time.time()
+
+# -----------------------------------------------------------
+# New Helper Functions: Chat Logging & TTS Generation
+# -----------------------------------------------------------
 def log_chat_history(username, message, is_markdown=False):
     """Log the chat entry to a markdown file."""
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -174,7 +215,9 @@ async def generate_chat_audio(username, message, voice):
     audio_file = await async_edge_tts_generate(message, voice, username)
     return audio_file
 
-# Audio Processing
+# -----------------------------------------------------------
+# Audio Processing Function
+# -----------------------------------------------------------
 async def async_edge_tts_generate(text, voice, username):
     cache_key = f"{text[:100]}_{voice}"
     if cache_key in st.session_state['audio_cache']:
@@ -193,9 +236,24 @@ def play_and_download_audio(file_path):
         st.audio(file_path, start_time=0)
         st.markdown(get_download_link(file_path), unsafe_allow_html=True)
 
-# ---------------------
+# -----------------------------------------------------------
+# WebSocket Broadcast Function
+# -----------------------------------------------------------
+async def broadcast_message(message, room_id):
+    if room_id in st.session_state.active_connections:
+        disconnected = []
+        for client_id, ws in st.session_state.active_connections[room_id].items():
+            try:
+                await ws.send(message)
+            except websockets.ConnectionClosed:
+                disconnected.append(client_id)
+        for client_id in disconnected:
+            if client_id in st.session_state.active_connections[room_id]:
+                del st.session_state.active_connections[room_id][client_id]
+
+# -----------------------------------------------------------
 # Wrapper Function to Save Chat Entry (Using the Two Helpers Above)
-# ---------------------
+# -----------------------------------------------------------
 async def save_chat_entry(username, message, voice, is_markdown=False):
     if not message.strip() or message == st.session_state.get('last_transcript', ''):
         return None, None
@@ -232,7 +290,9 @@ async def load_chat():
         content = f.read().strip()
     return content.split('\n')
 
-# Session State Init
+# -----------------------------------------------------------
+# Session State Initialization
+# -----------------------------------------------------------
 def init_session_state():
     defaults = {
         'server_running': False, 'server_task': None, 'active_connections': {},
@@ -263,63 +323,12 @@ def init_session_state():
 
 init_session_state()
 
-# Agent Class for Players
-class PlayerAgent:
-    def __init__(self, username, char_data):
-        self.username = username
-        self.x = random.uniform(-20, 20)
-        self.z = random.uniform(-40, 40)
-        self.color = char_data["color"]
-        self.shape = char_data["shape"]
-        self.score = 0
-        self.treasures = 0
-        self.last_active = time.time()
-        self.voice = char_data["voice"]
-
-    def to_dict(self):
-        return {
-            "username": self.username,
-            "x": self.x,
-            "z": self.z,
-            "color": self.color,
-            "shape": self.shape,
-            "score": self.score,
-            "treasures": self.treasures,
-            "last_active": self.last_active
-        }
-
-    def update_from_message(self, message):
-        if '|' in message:
-            _, content = message.split('|', 1)
-            if content.startswith("MOVE:"):
-                _, x, z = content.split(":")
-                self.x, self.z = float(x), float(z)
-                self.last_active = time.time()
-            elif content.startswith("SCORE:"):
-                self.score = int(content.split(":")[1])
-                self.last_active = time.time()
-            elif content.startswith("TREASURE:"):
-                self.treasures = int(content.split(":")[1])
-                self.last_active = time.time()
-
-# WebSocket Broadcast
-async def broadcast_message(message, room_id):
-    if room_id in st.session_state.active_connections:
-        disconnected = []
-        for client_id, ws in st.session_state.active_connections[room_id].items():
-            try:
-                await ws.send(message)
-            except websockets.ConnectionClosed:
-                disconnected.append(client_id)
-        for client_id in disconnected:
-            if client_id in st.session_state.active_connections[room_id]:
-                del st.session_state.active_connections[room_id][client_id]
-
-# Chat and Quest Log (uses the new helper functions)
+# -----------------------------------------------------------
+# WebSocket Handler and Related Functions
+# -----------------------------------------------------------
 async def save_chat_entry_wrapper(username, message, voice, is_markdown=False):
     return await save_chat_entry(username, message, voice, is_markdown)
 
-# WebSocket Handler
 async def websocket_handler(websocket, path):
     client_id = str(uuid.uuid4())
     room_id = "quest"
@@ -420,7 +429,9 @@ def start_websocket_server():
     asyncio.set_event_loop(loop)
     loop.run_until_complete(run_websocket_server())
 
-# ArXiv Integration remains unchanged
+# -----------------------------------------------------------
+# ArXiv Integration (Unchanged)
+# -----------------------------------------------------------
 async def perform_arxiv_search(query, username):
     gradio_client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
     refs = gradio_client.predict(
@@ -431,7 +442,9 @@ async def perform_arxiv_search(query, username):
     md_file, audio_file = await save_chat_entry_wrapper(username, result, voice, True)
     return md_file, audio_file
 
+# -----------------------------------------------------------
 # Sidebar Functions
+# -----------------------------------------------------------
 def update_marquee_settings_ui():
     st.sidebar.markdown("### 🎯 Marquee Settings")
     cols = st.sidebar.columns(2)
@@ -470,7 +483,9 @@ def log_performance_metrics():
             percentage = (duration / total_time) * 100
             st.sidebar.write(f"**{operation}:** {duration:.2f}s ({percentage:.1f}%)")
 
-# Enhanced 3D Game HTML remains largely unchanged, with dynamic insertion of the username and avatar details
+# -----------------------------------------------------------
+# Enhanced 3D Game HTML (With dynamic insertion of username and avatar)
+# -----------------------------------------------------------
 rocky_map_html = f"""
 <!DOCTYPE html>
 <html lang="en">
@@ -736,7 +751,9 @@ rocky_map_html = f"""
 </html>
 """
 
+# -----------------------------------------------------------
 # Main Game Loop
+# -----------------------------------------------------------
 def main():
     st.markdown(f"<h2 style='text-align: center;'>Welcome, {st.session_state.username}!</h2>", unsafe_allow_html=True)
     message = st.text_input(f"🗨️ Chat as {st.session_state.username}:", placeholder="Type to chat or add world features! 🌲", key="chat_input")