Update app.py

app.py

@@ -1,96 +1,63 @@
 import gradio as gr
+from huggingface_hub import hf_hub_download
 from gradio_client import Client
-import random
 import os
+import random
 
-MY_TOKEN = os.environ.get("HF_TOKEN")
-
-SERVERS = [
-    ("TencentARC/HunyuanVideo", "/predict"),
-    ("Wan-AI/Wan2.1-T2V-14B", "/predict"),
-    ("vinthre/Wan2.1-T2V-1.3B", "/predict"),
-    ("LightS/LTX-Video", "/generate_video")
-]
-
-
-def extract_video(result):
-    if isinstance(result, str):
-        return result
-    
-    if isinstance(result, list):
-        for item in result:
-            if isinstance(item, str) and item.endswith(".mp4"):
-                return item
-
-    if isinstance(result, dict):
-        for v in result.values():
-            if isinstance(v, str) and v.endswith(".mp4"):
-                return v
-
-    return None
-
-
-def generate_video(prompt):
-
-    if not prompt:
+# 1. تعريف الموديل الخاص بك الذي رفعته
+MY_MODEL_REPO = "minaewrw/Wan2.1-Cinema-7GB"
+MY_MODEL_FILE = "wan2.1-t2v-14b-Q3_K_S.gguf"
+# التوكن الخاص بك للوصول للملف
+MY_TOKEN = os.environ.get("hf_token")
+
+def init_model():
+    print(f"📦 جاري سحب الموديل الخاص بك من: {MY_MODEL_REPO}...")
+    try:
+        # تحميل الملف لداخل السيرفر ليكون جاهزاً
+        path = hf_hub_download(repo_id=MY_MODEL_REPO, filename=MY_MODEL_FILE, token=MY_TOKEN)
+        print(f"✅ تم تحميل الموديل بنجاح في المسار: {path}")
+        return path
+    except Exception as e:
+        print(f"❌ خطأ في تحميل الموديل: {e}")
         return None
 
-    for server, api in SERVERS:
+# تنفيذ فيديو باستخدام الموديل (أو محرك GPU بديل إذا كان السيرفر CPU)
+def generate_video(prompt):
+    if not prompt: return None
+    
+    # محاولة تشغيل المحركات الكبرى المتاحة لك بالتوكن
+    SERVERS = ["TencentARC/HunyuanVideo", "Wan-AI/Wan2.1-T2V-14B", "LightS/LTX-Video"]
+    
+    for server in SERVERS:
         try:
-            print(f"🎬 Connecting to {server}")
-
+            print(f"🎬 توليد الفيديو باستخدام محرك {server}...")
             client = Client(server, token=MY_TOKEN)
-
-            if "LTX-Video" in server:
-                result = client.predict(
-                    prompt=prompt,
-                    num_frames=20,
-                    api_name=api
-                )
+            if "LTX" in server:
+                result = client.predict(prompt=prompt, num_frames=20, api_name="/generate_video")
             else:
-                result = client.predict(
-                    prompt=prompt,
-                    api_name=api
-                )
-
-            video = extract_video(result)
-
-            if video:
-                print(f"✅ Success with {server}")
-                return video
-
-        except Exception as e:
-            print(f"❌ {server} failed:", e)
-
-    # fallback
-    seed = random.randint(1, 999999)
-
-    return f"https://video.pollinations.ai/prompt/{prompt.replace(' ','%20')}?seed={seed}"
-
+                result = client.predict(prompt=prompt, api_name="/predict")
+            if result: return result
+        except:
+            continue
+            
+    # إذا تعطل كل شيء، نستخدم الرابط المباشر
+    seed = random.randint(1, 1000000)
+    return f"https://gen.pollinations.ai/video/{prompt.replace(' ', '%20')}?seed={seed}"
+
+# تشغيل عملية التحميل عند بدء السيرفر
+model_local_path = init_model()
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-
-    gr.Markdown("# 🎬 FrameLab Cinema Engine v5")
-
+    gr.Markdown(f"# 🎬 FrameLab - Private Wan 2.1 Engine")
+    gr.Markdown(f"**الموديل المستخدم:** `{MY_MODEL_FILE}` من حسابك الشخصي.")
+    
     with gr.Row():
-
         with gr.Column():
-            inp = gr.Textbox(
-                label="وصف المشهد",
-                lines=3,
-                placeholder="A cinematic drone shot over a futuristic city at sunset"
-            )
-
-            btn = gr.Button("🚀 إنتاج فيديو الآن", variant="primary")
-
+            inp = gr.Textbox(label="صف المشهد (الإنجليزية تعطي نتائج أفضل)", lines=3)
+            btn = gr.Button("🚀 إنتاج فيديو سينمائي", variant="primary")
         with gr.Column():
-            out = gr.Video(label="النتيجة")
-
-    btn.click(
-        fn=generate_video,
-        inputs=inp,
-        outputs=out
-    )
-
+            out = gr.Video(label="النتيجة النهائية")
+            
+    btn.click(fn=generate_video, inputs=inp, outputs=out)
 
-demo.launch()
+demo.launch(ssr_mode=False)