Create app.py

app.py

@@ -0,0 +1,128 @@
+import gradio as gr
+import uuid
+import os
+import gzip
+import subprocess
+import datetime
+import time
+import soundfile as sf
+
+from app_utils import tts_interface, models
+
+# === Only keep the 'برنا' model ===
+borna_model = next(model for model in models if "برنا" in model[0] or "برنا" in model[2])
+voice_id = borna_model[2]
+voice_url = borna_model[3]
+
+# === Paths ===
+STATIC_DIR = "static"
+os.makedirs(STATIC_DIR, exist_ok=True)
+PHONETIC_FILE = os.path.join(STATIC_DIR, "fa_extra.txt.gz")
+
+# === TTS Function ===
+def tts(text):
+    # Save input text for phonemizer
+    input_file = os.path.join(STATIC_DIR, "input.txt")
+    with open(input_file, "w", encoding="utf-8") as f:
+        f.write(text.strip())
+
+    # Run espeak-ng to get phonemes
+    phonemes_result = subprocess.run(
+        ["espeak-ng", "-v", "fa", "-x", "-q", "-f", input_file],
+        capture_output=True,
+        text=True
+    )
+    phonemes = phonemes_result.stdout.strip()
+
+    # Run TTS
+    (sample_rate, audio_data), _ = tts_interface(voice_id, text.strip(), '')
+    out_path = os.path.join(STATIC_DIR, f"{uuid.uuid4().hex}.wav")
+    sf.write(out_path, audio_data, samplerate=sample_rate, subtype="PCM_16")
+
+    status = f"مدل: {voice_url}\nآوانگاشت: {phonemes}"
+    return out_path, status
+
+# === Phonemizer + Save ===
+def phonemize(word, phonetic, task):
+    if task == "phonemize" or not phonetic.strip():
+        phoneme_target = word
+    else:
+        phoneme_target = f"[[{phonetic.strip()}]]"
+
+    input_file = os.path.join(STATIC_DIR, "input.txt")
+    with open(input_file, "w", encoding="utf-8") as f:
+        f.write(phoneme_target)
+
+    result = subprocess.run(
+        ["espeak-ng", "-v", "fa", "-x", "-q", "-f", input_file],
+        capture_output=True,
+        text=True
+    )
+    phonemes = result.stdout.strip()
+
+    out_path = os.path.join(STATIC_DIR, f"{uuid.uuid4().hex}.wav")
+    subprocess.run(f'espeak-ng -v fa -w "{out_path}" -f "{input_file}"', shell=True)
+
+    if task == "send" and phonetic.strip():
+        with gzip.open(PHONETIC_FILE, "at", encoding="utf-8") as f:
+            f.write(f"{word.strip()}\t{phonetic.strip()}\n")
+
+    status = f"متن: {word}\nآوانگاشت: {phonemes}"
+    return out_path, phonemes, status
+
+# === List saved words ===
+def list_words():
+    entries = []
+    with gzip.open(PHONETIC_FILE, "rt", encoding="utf-8") as f:
+        for line in f:
+            if not line.startswith("//"):
+                parts = line.strip().split("\t")
+                if len(parts) == 2:
+                    entries.append((parts[0], parts[1]))
+    return entries
+
+# === Iran time without pytz ===
+def get_iran_time():
+    offset = 3.5 * 3600
+    utc = time.gmtime()
+    iran_time = time.localtime(time.mktime(utc) + offset)
+    return datetime.datetime(*iran_time[:6]).strftime("%Y-%m-%d %H:%M:%S")
+
+# Log start
+with gzip.open(PHONETIC_FILE, "at", encoding="utf-8") as f:
+    f.write(f"// started at : {get_iran_time()}\n")
+
+# === UI ===
+
+with gr.Blocks(title="TTS for Persian with Bornaa") as demo:
+    gr.Markdown("## متن به گفتار 🗣️")
+
+    with gr.Tab("تبدیل متن به گفتار"):
+        text_input = gr.Textbox(label="متن فارسی", lines=2)
+        out_audio = gr.Audio(label="خروجی صوتی")
+        out_status = gr.Textbox(label="وضعیت مدل", interactive=False)
+
+        btn_tts = gr.Button("تبدیل کن")
+        btn_tts.click(tts, inputs=text_input, outputs=[out_audio, out_status])
+
+    with gr.Tab("ویرایش تلفظ"):
+        word_input = gr.Textbox(label="کلمه", placeholder="مثلاً: سلام")
+        phonetic_input = gr.Textbox(label="آوانویسی دلخواه (اختیاری)")
+        task_choice = gr.Radio(label="عملیات", choices=["phonemize", "send"], value="phonemize")
+
+        ph_audio = gr.Audio(label="صدای تولیدشده")
+        ph_output = gr.Textbox(label="آوانگاشت")
+        ph_status = gr.Textbox(label="وضعیت", interactive=False)
+
+        btn_phonemize = gr.Button("انجام")
+        btn_phonemize.click(phonemize,
+                            inputs=[word_input, phonetic_input, task_choice],
+                            outputs=[ph_audio, ph_output, ph_status])
+
+    with gr.Tab("واژه‌های اخیر"):
+        gr.Markdown("### واژه‌های ذخیره‌شده")
+        word_table = gr.Dataframe(headers=["کلمه", "آوانویسی"], datatype=["str", "str"])
+        btn_load_words = gr.Button("بارگذاری")
+        btn_load_words.click(fn=list_words, inputs=[], outputs=[word_table])
+
+demo.launch()