Update app.py

app.py

@@ -89,7 +89,7 @@ UNICODE_FONTS = [
     ("Sans Serif Bold Italic", lambda x: "".join(chr(ord(c) + 0x1D63C - 0x41) if 'A' <= c <= 'Z' else chr(ord(c) + 0x1D656 - 0x61) if 'a' <= c <= 'z' else c for c in x)),
     ("Monospace", lambda x: "".join(chr(ord(c) + 0x1D670 - 0x41) if 'A' <= c <= 'Z' else chr(ord(c) + 0x1D68A - 0x61) if 'a' <= c <= 'z' else c for c in x)),
     ("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) - 0x1F130 - 0x41) 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)),
     ("Regional Indicator", lambda x: "".join(chr(ord(c) - 0x41 + 0x1F1E6) if 'A' <= c <= 'Z' else c for c in x)),
@@ -173,9 +173,9 @@ def log_action(username, action):
 def clean_text_for_tts(text):
     cleaned = re.sub(r'[#*!\[\]]+', '', text)
     cleaned = ' '.join(cleaned.split())
-    return cleaned[:200] if cleaned else "No text to speak"
+    return cleaned if cleaned else "No text to speak"  # Default if empty
 
-# Audio Processor Class from your code
+# Audio Processor Class from your code, adapted
 class AudioProcessor:
     def __init__(self):
         self.cache_dir = "audio_cache"
@@ -196,7 +196,7 @@ class AudioProcessor:
         cache_path = filename or os.path.join(self.cache_dir, f"{cache_key}.mp3")
 
         if cache_key in self.metadata and os.path.exists(cache_path):
-            return open(cache_path, 'rb').read()
+            return cache_path
 
         # Clean text for speech
         text = text.replace("\n", " ").replace("</s>", " ").strip()
@@ -204,8 +204,14 @@ class AudioProcessor:
             return None
 
         # Generate audio with edge_tts
-        communicate = edge_tts.Communicate(text, voice)
-        await communicate.save(cache_path)
+        try:
+            communicate = edge_tts.Communicate(text, voice)
+            await communicate.save(cache_path)
+            if not os.path.exists(cache_path):
+                raise edge_tts.exceptions.NoAudioReceived("No audio file created")
+        except edge_tts.exceptions.NoAudioReceived as e:
+            log_action("System 🌟", f"TTS failed for text '{text}' with voice '{voice}': {str(e)}")
+            return None
 
         # Update metadata
         self.metadata[cache_key] = {
@@ -215,7 +221,7 @@ class AudioProcessor:
         }
         self._save_metadata()
 
-        return open(cache_path, 'rb').read()
+        return cache_path
 
 # Chat saver - words locked tight! 💬🔒
 async def save_chat_entry(username, message, is_markdown=False, quote_line=None):
@@ -245,8 +251,9 @@ async def save_chat_entry(username, message, is_markdown=False, quote_line=None)
     cleaned_message = clean_text_for_tts(message)
     audio_processor = AudioProcessor()
     audio_filename = f"{format_timestamp_prefix(username)}-{hashlib.md5(cleaned_message.encode()).hexdigest()[:8]}.mp3"
-    audio_data = await audio_processor.create_audio(cleaned_message, voice, audio_filename)
-    if audio_data:
+    log_action(username, f"Attempting TTS with text: '{cleaned_message}' and voice: '{voice}'")
+    audio_file = await audio_processor.create_audio(cleaned_message, voice, audio_filename)
+    if audio_file:
         with open(HISTORY_FILE, 'a') as f:
             f.write(f"[{timestamp}] {username} ({voice}): Audio generated - {audio_filename}\n")
         with open(user_history_file, 'a') as f:
@@ -259,6 +266,8 @@ async def save_chat_entry(username, message, is_markdown=False, quote_line=None)
                 file_path = message.split(": ")[1]
                 if os.path.exists(file_path) and file_path.endswith(('.png', '.jpg', '.mp4', '.pdf')):
                     f.write(f"{indent}[{timestamp}] Media: ![Media]({file_path})\n")
+    else:
+        log_action(username, f"Failed to generate audio for message: '{message}'")
 
     await broadcast_message(f"{username}|{message}", "chat")
     st.session_state.last_chat_update = time.time()
@@ -488,16 +497,6 @@ async def run_websocket_server():
         st.session_state.server_running = True
         await server.wait_closed()
 
-# Dummy AI lookup function (replace with actual implementation)
-async def perform_ai_lookup(query, vocal_summary=True, extended_refs=False, titles_summary=True, full_audio=False, useArxiv=True, useArxivAudio=False):
-    username = st.session_state.get('username', 'System 🌟')
-    result = f"AI Lookup Result for '{query}' (Arxiv: {useArxiv}, Audio: {useArxivAudio})"
-    await save_chat_entry(username, result)
-    if useArxivAudio:
-        audio_file = await async_edge_tts_generate(result, st.session_state.voice)
-        if audio_file:
-            st.audio(audio_file)
-
 # Delete all user files function
 def delete_user_files():
     protected_files = {'app.py', 'requirements.txt', 'README.md', CHAT_FILE, QUOTE_VOTES_FILE, MEDIA_VOTES_FILE, HISTORY_FILE, STATE_FILE}
@@ -597,11 +596,6 @@ def main():
                             audio_ref = line.split("Audio: ")[-1].strip()
                             if os.path.exists(audio_ref):
                                 st.session_state.audio_cache[cache_key] = audio_ref
-                            else:
-                                cleaned_text = clean_text_for_tts(line.split(': ', 1)[1].split("Audio:")[0])
-                                voice = st.session_state.voice if username == st.session_state.username else FUN_USERNAMES.get(username, "en-US-AriaNeural")
-                                audio_file = await async_edge_tts_generate(cleaned_text, voice)
-                                st.session_state.audio_cache[cache_key] = audio_file
                         audio_file = st.session_state.audio_cache.get(cache_key)
                         if audio_file:
                             play_and_download_audio(audio_file)
@@ -689,8 +683,9 @@ def main():
                     pdf_filename, texts, audio_files = await save_pdf_and_generate_audio(uploaded_file, username)
                     await save_chat_entry(username, f"Uploaded PDF: {pdf_filename}")
                     for i, (text, audio_file) in enumerate(zip(texts, audio_files)):
-                        with open(CHAT_FILE, 'a') as f:
-                            f.write(f"  [{timestamp}] Page {i+1} Audio: {audio_file}\n")
+                        if audio_file:
+                            with open(CHAT_FILE, 'a') as f:
+                                f.write(f"  [{timestamp}] Page {i+1} Audio: {audio_file}\n")
                 else:
                     await save_chat_entry(username, f"Uploaded media: {file_path}")
                 await save_chat_history_with_image(username, file_path)