import tempfile
import os
import gradio as gr
import subprocess

MAX_TXT_LEN = 800

def tts(text: str):
    if len(text) > MAX_TXT_LEN:
        text = text[:MAX_TXT_LEN]
        print(f"Input text was cutoff since it went over the {MAX_TXT_LEN} character limit.")
    print(text)
    
    try:
        with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
            # اجرای دستور و دریافت خروجی و خطا
            result = subprocess.run(
                f'mimic3 --voice fa/haaniye_low "{text}" > {fp.name}', 
                shell=True, 
                stdout=subprocess.PIPE, 
                stderr=subprocess.PIPE
            )
            if result.returncode != 0:
                print(f"Error: {result.stderr.decode('utf-8')}")
                raise subprocess.CalledProcessError(result.returncode, result.args)
            return fp.name
    except Exception as e:
        print(f"Exception occurred: {e}")

article= ""
examples=[
    "شیش سیخ جیگر سیخی شیش هزار",
    "سه شیشه شیر ، سه سیر سرشیر",
    "دزدی دزدید ز بز دزدی بزی ، عجب دزدی که دزدید ز بز دزدی بزی",
    "مثنوی یکی از قالب های شعری است ک هر بیت قافیه ی جداگانه دارد",
    "در گلو ماند خس او سالها، چیست آن خس مهر جاه و مالها",
]

iface = gr.Interface(
    fn=tts,
    inputs=[
        gr.Textbox(
            label="Text",
            value="زندگی فقط یک بار است؛ از آن به خوبی استفاده کن",
        )
    ],
    outputs=gr.Audio(label="Output", type='filepath'),
    examples=examples
)

iface.launch(share=False)