Update app.py

app.py

@@ -1,14 +1,14 @@
-
-chmod +x setup.sh && ./setup.sh
-
-import sys
 import os
-sys.path.append(os.path.abspath("."))
-
-import gradio as gr
+import sys
 from pathlib import Path
 import argparse
+import gradio as gr
 
+# تشغيل سكربت التحميل إذا كنتِ تبغي تنفذيه هنا
+os.system("chmod +x setup.sh && ./setup.sh")
+
+# إعداد المسارات
+sys.path.append(os.path.abspath("."))
 
 # استيراد بقية الأشياء
 from STT.sst import speech_to_text
@@ -18,9 +18,9 @@ from FantasyTalking.infer import load_models, main
 
 # ثابتات تحميل النموذج
 args_template = argparse.Namespace(
-    wan_model_dir="./models/Wan2.1-I2V-14B-720P",
-    fantasytalking_model_path="./models/fantasytalking_model.ckpt",
-    wav2vec_model_dir="./models/wav2vec2-base-960h",
+    wan_model_dir="./wan",  # افترضي إنك حملتيه هناك
+    fantasytalking_model_path="./fantasy/fantasytalking_model.ckpt",
+    wav2vec_model_dir="./wav2vec",
     image_path="",
     audio_path="",
     prompt="",
@@ -37,7 +37,9 @@ args_template = argparse.Namespace(
 )
 
 # تحميل النماذج مرة وحدة فقط
+print("🚀 جاري تحميل النماذج...")
 pipe, fantasytalking, wav2vec_processor, wav2vec = load_models(args_template)
+print("✅ تم تحميل النماذج بنجاح!")
 
 def generate_video(image_path, audio_path, prompt, output_dir="./output"):
     args = argparse.Namespace(
@@ -50,16 +52,16 @@ def generate_video(image_path, audio_path, prompt, output_dir="./output"):
     return main(args, pipe, fantasytalking, wav2vec_processor, wav2vec)
 
 def full_pipeline(user_audio, user_image):
-    # 1. تحويل الصوت إلى نص
+    print("🎤 تحويل الصوت إلى نص...")
     user_text = speech_to_text(user_audio)
 
-    # 2. توليد الرد من LLM
+    print("💬 توليد الرد...")
     reply = generate_reply(user_text)
 
-    # 3. تحويل الرد إلى صوت
+    print("🔊 تحويل الرد إلى صوت...")
     reply_audio_path = generate_voice(reply)
 
-    # 4. توليد فيديو من الصورة والصوت
+    print("📽️ توليد الفيديو...")
     Path("./output").mkdir(parents=True, exist_ok=True)
     video_path = generate_video(
         image_path=user_image,