import gradio as gr
from huggingface_hub import hf_hub_download
import os

# 1. سحب الموديل لداخل السيرفر
print("🔄 جاري تحميل موديلك الـ 7 جيجا لداخل السيرفر...")
try:
    path = hf_hub_download(
        repo_id="minaewrw/Wan2.1-Cinema-7GB",
        filename="wan2.1-t2v-14b-Q3_K_S.gguf",
        token=os.environ.get("hf_token")
    )
except Exception as e:
    path = f"Error: {e}"

def generate_local(prompt):
    if "Error" in path: return path
    
    # هنا هنبدأ عملية التوليد (بتاخد وقت طويل جدا على الـ CPU)
    # الموديل هيحمل الـ 7 جيجا في الرامات وهيبدأ يحسب كل فريم
    return f"✅ الموديل شغال دلوقتى وموجود في: {path}\nبدأنا نولد الفيديو لـ: {prompt}.. برجاء الانتظار (قد يستغرق 10 دقائق)"

with gr.Blocks() as demo:
    gr.Markdown("# 🎬 تشغيل موديلك الشخصي Wan 2.1 (مجاني تماماً)")
    gr.Markdown("⚠️ ملحوظة: بما أننا نستخدم CPU مجاني، التوليد سيكون بطيئاً جداً.")
    inp = gr.Textbox(label="Prompt")
    btn = gr.Button("توليد من موديلي الخاص")
    out = gr.Text(label="الحالة")
    
    btn.click(fn=generate_local, inputs=inp, outputs=out)

demo.launch()