Update app.py

app.py

@@ -1,10 +1,19 @@
 import sys
 import os
 sys.path.append(os.path.abspath("."))
+
 import gradio as gr
 from pathlib import Path
 import argparse
 
+# تحميل النماذج تلقائيًا إذا ما كانت موجودة
+from download_all_models import maybe_download_model
+
+maybe_download_model("Wan-AI/Wan2.1-I2V-14B-720P", "./models/Wan2.1-I2V-14B-720P")
+maybe_download_model("facebook/wav2vec2-base-960h", "./models/wav2vec2-base-960h")
+maybe_download_model("acvlab/FantasyTalking", "./models/fantasytalking_model")
+
+# استيراد بقية الأشياء
 from STT.sst import speech_to_text
 from LLM.llm import generate_reply
 from TTS_X.tts import generate_voice
@@ -43,7 +52,6 @@ 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. تحويل الصوت إلى نص
     user_text = speech_to_text(user_audio)
@@ -64,7 +72,6 @@ def full_pipeline(user_audio, user_image):
 
     return user_text, reply, reply_audio_path, video_path
 
-
 # واجهة Gradio
 with gr.Blocks(title="🧠 صوتك يحرك صورة!") as demo:
     gr.Markdown("## 🎤➡️💬➡️🔊➡️📽️ من صوتك إلى فيديو متكلم!")