Update app.py

app.py

@@ -1,55 +1,139 @@
+import os
+import tempfile
+import logging
+import asyncio
+
 import gradio as gr
 import edge_tts
-import asyncio
-import tempfile
+import edge_tts.communicate as communicate_module
 
-VOICES = [
-    "my-MM-NilarNeural",
-    "my-MM-ThihaNeural",
-    "en-US-AriaNeural",
-    "en-US-GuyNeural"
-]
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger("MyanmarEdgeTTS")
+
+APP_TITLE = "Myanmar Edge TTS"
+
+# New Microsoft Edge Read Aloud endpoint
+NEW_WSS_URL = (
+    "wss://api.msedgeservices.com/"
+    "tts/cognitiveservices/websocket/v1"
+    "?TrustedClientToken=6A5AA1D4EAFF4E9FB37E23D68491D6F4"
+)
+
+# Set endpoint in the module used by Communicate
+communicate_module.WSS_URL = NEW_WSS_URL
+
+VOICES = {
+    "Myanmar Female — Nilar": "my-MM-NilarNeural",
+    "Myanmar Male — Thiha": "my-MM-ThihaNeural",
+}
+
+
+async def create_audio(text, voice):
+    communicate = edge_tts.Communicate(
+        text=text,
+        voice=voice,
+    )
+
+    with tempfile.NamedTemporaryFile(
+        suffix=".mp3",
+        delete=False,
+    ) as temp:
+        output_path = temp.name
 
-async def generate_tts(text, voice, proxy):
-    if not text:
-        return None, "ကျေးဇူးပြု၍ စာသားထည့်ပါ။"
     try:
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
-        temp_file.close()
-        
-        proxy_url = proxy if proxy and proxy.strip() != "" else None
-        
-        # Proxy မှာ Username/Password ပါလာရင် aiohttp/edge-tts က ဒီအတိုင်း လက်ခံနိုင်ပါတယ်
-        communicate = edge_tts.Communicate(text, voice, proxy=proxy_url)
-        await communicate.save(temp_file.name)
-        return temp_file.name, "အောင်မြင်ပါသည်။"
-    except Exception as e:
-        return None, f"Error: {str(e)}"
-
-def tts_interface(text, voice, proxy):
-    return asyncio.run(generate_tts(text, voice, proxy))
-
-with gr.Blocks() as app:
-    gr.Markdown("## Edge TTS with Proxy Support")
-    with gr.Row():
-        with gr.Column():
-            text_input = gr.Textbox(label="စာသားထည့်ပါ", lines=5)
-            voice_dropdown = gr.Dropdown(choices=VOICES, value="my-MM-NilarNeural", label="အသံရွေးချယ်ပါ")
-            proxy_input = gr.Textbox(
-                label="Proxy URL (Optional)", 
-                value="http://mhuekxym:29u66euxw5ww@p.webshare.io:80", 
-                placeholder="http://user:pass@ip:port"
+        await communicate.save(output_path)
+
+        if (
+            not os.path.exists(output_path)
+            or os.path.getsize(output_path) < 100
+        ):
+            raise RuntimeError(
+                "Audio file is empty. Microsoft TTS returned no audio."
             )
-            submit_btn = gr.Button("အသံဖန်တီးမည်", variant="primary")
-        with gr.Column():
-            audio_output = gr.Audio(label="ထွက်ပေါ်လာသောအသံ", type="filepath")
-            status_output = gr.Textbox(label="အခြေအနေ")
-            
-    submit_btn.click(
-        fn=tts_interface,
-        inputs=[text_input, voice_dropdown, proxy_input],
-        outputs=[audio_output, status_output]
+
+        logger.info(
+            "Audio generated: voice=%s size=%d",
+            voice,
+            os.path.getsize(output_path),
+        )
+
+        return output_path
+
+    except Exception:
+        if os.path.exists(output_path):
+            os.remove(output_path)
+
+        logger.exception("TTS generation failed")
+        raise
+
+
+def generate_audio(text, voice_name):
+    text = (text or "").strip()
+
+    if not text:
+        raise gr.Error("မြန်မာစာတစ်ခုခု အရင်ထည့်ပါ။")
+
+    if len(text) > 2000:
+        raise gr.Error("စာလုံးရေ ၂၀၀၀ ထက် မပိုရပါ။")
+
+    if voice_name not in VOICES:
+        raise gr.Error("Voice မမှန်ပါ။")
+
+    voice = VOICES[voice_name]
+
+    try:
+        return asyncio.run(create_audio(text, voice))
+
+    except Exception as exc:
+        logger.exception("Generation failed")
+
+        raise gr.Error(
+            f"TTS Error: {type(exc).__name__}: {str(exc)[:400]}"
+        )
+
+
+with gr.Blocks(title=APP_TITLE) as demo:
+    gr.Markdown(
+        """
+        # 🇲🇲 Myanmar Edge TTS
+
+        Microsoft Edge Read Aloud endpoint အသစ်ဖြင့်
+        မြန်မာအသံထုတ်စမ်းသပ်ရန်။
+        """
     )
 
+    text_input = gr.Textbox(
+        label="မြန်မာစာ",
+        placeholder="မင်္ဂလာပါ။ အသံစမ်းသပ်နေပါတယ်။",
+        lines=4,
+    )
+
+    voice_input = gr.Dropdown(
+        choices=list(VOICES.keys()),
+        value="Myanmar Female — Nilar",
+        label="အသံရွေးပါ",
+    )
+
+    generate_button = gr.Button(
+        "အသံထုတ်မယ်",
+        variant="primary",
+    )
+
+    audio_output = gr.Audio(
+        label="ထုတ်လုပ်ပြီးသောအသံ",
+        type="filepath",
+    )
+
+    generate_button.click(
+        fn=generate_audio,
+        inputs=[text_input, voice_input],
+        outputs=audio_output,
+        api_name="generate",
+    )
+
+
 if __name__ == "__main__":
-    app.launch(server_name="0.0.0.0", server_port=7860)
+    demo.queue(max_size=5).launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+    )