Update app.py

app.py

@@ -2,33 +2,40 @@ import gradio as gr
 from gradio_client import Client, handle_file
 
 def generate_video(image_path, prompt):
+    if image_path is None:
+        return None
+    
     try:
-        # اتصال به اسپیس مقصد
+        # ۱. اتصال به کلاینت (این کار خودش Config و API Schema رو می‌خونه)
         client = Client("ibuildproducts/wan22-i2v-v4-demo")
         
-        # در آپدیت جدید، بهتر است از اسم پیش‌فرض استفاده نکنی
-        # و اجازه دهی خود کلاینت جابجایی فایل را مدیریت کند
-        result = client.predict(
+        # ۲. ارسال به صف (Join the Queue)
+        # این متد خودش فایل رو آپلود می‌کنه و پارامترها رو می‌فرسته
+        job = client.submit(
             input_image=handle_file(image_path),
             prompt=prompt,
-            api_name="/predict" 
+            api_name="/predict"
         )
         
-        # نتیجه در ZeroGPU معمولاً یک مسیر فایل محلی است
+        # ۳. منتظر ماندن برای نتیجه (Stream results رو مدیریت می‌کنه)
+        result = job.result() 
+        
+        # ۴. بازگرداندن مسیر ویدیو
         return result
     except Exception as e:
+        print(f"Error: {e}")
         return None
 
+# رابط کاربری Gradio
 with gr.Blocks() as demo:
-    gr.Markdown("### Wan 2.1 Video Generator")
     with gr.Row():
         with gr.Column():
-            img = gr.Image(type="filepath")
-            txt = gr.Textbox(value="moving cinematic eyes, 1 second", label="Prompt")
-            btn = gr.Button("ساخت ویدیو")
+            img_input = gr.Image(type="filepath")
+            prompt_input = gr.Textbox(label="Prompt", value="cinematic motion, high quality")
+            submit_btn = gr.Button("ساخت ویدیو")
         with gr.Column():
-            out = gr.Video()
+            video_output = gr.Video()
 
-    btn.click(generate_video, [img, txt], out)
+    submit_btn.click(generate_video, [img_input, prompt_input], video_output)
 
 demo.launch()