Update app.py

app.py

@@ -1,46 +1,88 @@
-import spaces # يجب أن يكون السطر الأول
+import spaces # السطر الأول دائماً
 import gradio as gr
 import os
+import torch
+import soundfile as sf
 import time
-import subprocess
+from huggingface_hub import hf_hub_download
+from f5_tts.model import DiT
+from f5_tts.infer.utils_infer import load_model, load_vocoder, preprocess_ref_audio_text, infer_process
 
-# تعريف وظيفة التوليد مع ديكوريتور الـ GPU لإرضاء نظام Hugging Face
-@spaces.GPU
-def synthesize_piper(text):
-    if not text or len(text.strip()) < 2:
-        return None, "الرجاء إدخال نص صالح."
+# إعدادات النموذج السعودي (NAMAA Saudi TTS)
+REPO_ID = "NAMAA-Space/NAMAA-Saudi-TTS-V2"
+V1_BASE_CFG = dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4)
+
+# تحميل النموذج مرة واحدة
+print("جاري تحميل المحرك السعودي الفخم...")
+device = "cuda" if torch.cuda.is_available() else "cpu"
+ckpt_path = hf_hub_download(repo_id=REPO_ID, filename="model_last.pt")
+vocab_path = hf_hub_download(repo_id=REPO_ID, filename="vocab.txt")
+
+model = load_model(DiT, V1_BASE_CFG, ckpt_path, vocab_file=vocab_path, device=device)
+model = model.to(torch.float32).eval()
+vocoder = load_vocoder()
+
+@spaces.GPU(duration=120)
+def saudi_tts_predict(text, ref_audio, ref_text):
+    if not text or not ref_audio:
+        return None, "الرجاء إدخال النص ورفع الصوت المرجعي."
+
+    # تنظيف النص ومعالجته
+    clean_text = text.replace("\n", " ").strip()
     
+    output_file = "saudi_output.wav"
     start_time = time.time()
-    output_file = "output.wav"
-    
-    # نموذج "خالد" الأردني - صوت وقور ومسترسل
-    model_name = "ar_JO-khalid-medium" 
     
     try:
-        # تشغيل محرك Piper
-        # ملاحظة: Piper سريع جداً لدرجة أنه قد لا يحتاج للـ GPU فعلياً، لكننا نضعه هنا ليعمل الـ Space
-        command = f'echo "{text}" | piper --model {model_name} --output_file {output_file}'
-        subprocess.run(command, shell=True, check=True)
+        # معالجة الصوت المرجعي
+        # إذا لم يتم إدخال نص المرجع، نستخدم نصاً افتراضياً أو نحاول استخراجه
+        final_ref_text = ref_text if ref_text else "مرحباً بك في نظام القراءة الآلي المطور"
+        ref_audio_data, ref_text_data = preprocess_ref_audio_text(ref_audio, final_ref_text)
         
+        # التوليد باستخدام المحرك السعودي
+        wave, sr, _ = infer_process(
+            ref_audio_data, 
+            ref_text_data, 
+            clean_text, 
+            model, 
+            vocoder, 
+            nfe_step=32, # جودة عالية
+            speed=1.0,
+            device=device
+        )
+        
+        sf.write(output_file, wave, sr)
         duration = time.time() - start_time
-        return output_file, f"✅ تم التوليد بنجاح في {duration:.2f} ثانية."
+        return output_file, f"✅ تم التوليد بنبرة سعودية فخمة في {duration:.2f} ثانية."
     except Exception as e:
         return None, f"❌ حدث خطأ: {str(e)}"
 
-# بناء الواجهة
-with gr.Blocks() as demo:
-    gr.Markdown("# 🚀 قارئ المقالات السريع (Piper Arabic)")
-    gr.Markdown("هذا المحرك مصمم للسرعة القصوى والاستقرار على Hugging Face.")
+# واجهة مستخدم تليق بمشروع سعودي
+custom_css = """
+.gradio-container { direction: rtl; text-align: right; font-family: 'Cairo', sans-serif; }
+#title { text-align: center; color: #006C35; } /* لون العلم السعودي */
+"""
+
+with gr.Blocks(css=custom_css) as demo:
+    gr.Markdown("# 🇸🇦 قارئ المقالات السعودي المطور (NAMAA V2)", elem_id="title")
+    gr.Markdown("استخدم هذا المحرك للحصول على قراءة مسترسلة وفخمة للمقالات الإخبارية.")
     
     with gr.Row():
         with gr.Column():
-            input_text = gr.Textbox(label="نص المقال", lines=10, placeholder="ألصق النص هنا...", rtl=True)
-            btn = gr.Button("🚀 توليد فوري", variant="primary")
+            input_text = gr.Textbox(label="نص المقال الإخباري", lines=10, placeholder="ألصق مقالك هنا...", rtl=True)
+            ref_audio_input = gr.Audio(label="ارفع صوت المذيع الذي تريد تقليده", type="filepath")
+            ref_text_input = gr.Textbox(label="ماذا يقول المذيع في المقطع المرفوع؟", placeholder="اكتب النص لضمان دقة 100%...", rtl=True)
+            generate_btn = gr.Button("🚀 توليد الصوت الفخم", variant="primary")
+        
         with gr.Column():
-            output_audio = gr.Audio(label="الصوت المولد")
-            status = gr.Textbox(label="الحالة")
-
-    btn.click(synthesize_piper, inputs=input_text, outputs=[output_audio, status])
+            output_audio = gr.Audio(label="صوت المذيع السعودي")
+            status_msg = gr.Textbox(label="حالة المعالجة", interactive=False)
+            
+    generate_btn.click(
+        fn=saudi_tts_predict,
+        inputs=[input_text, ref_audio_input, ref_text_input],
+        outputs=[output_audio, status_msg]
+    )
 
-# نقل التنسيق إلى هنا لإصلاح التحذير الأخير
-demo.launch(css=".gradio-container {direction: rtl; text-align: right;}")
+if __name__ == "__main__":
+    demo.launch()