Update app.py

app.py

@@ -1,42 +1,34 @@
 import gradio as gr
 from gradio_client import Client, handle_file
-import os
-
-# آدرس اسپیس مقصد که در عکس فرستادی
-SOURCE_SPACE = "ibuildproducts/wan22-i2v-v4-demo"
 
 def generate_video(image_path, prompt):
     try:
-        # اتصال به API اسپیس مقصد
-        client = Client(SOURCE_SPACE)
+        # اتصال به اسپیس مقصد
+        client = Client("ibuildproducts/wan22-i2v-v4-demo")
         
-        # ارسال درخواست به متد predict
-        # نکته: پارامترها باید دقیقا مطابق با API مقصد باشد
+        # در آپدیت جدید، بهتر است از اسم پیش‌فرض استفاده نکنی
+        # و اجازه دهی خود کلاینت جابجایی فایل را مدیریت کند
         result = client.predict(
             input_image=handle_file(image_path),
             prompt=prompt,
-            api_name="/predict"
+            api_name="/predict" 
         )
         
-        # result معمولاً مسیر فایل ویدیویی ذخیره شده است
+        # نتیجه در ZeroGPU معمولاً یک مسیر فایل محلی است
         return result
     except Exception as e:
-        return f"Error: {str(e)}"
+        return None
 
-# طراحی رابط کاربری ساده
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown(f"### AI Video Generator (Bridge to {SOURCE_SPACE})")
-    
+with gr.Blocks() as demo:
+    gr.Markdown("### Wan 2.1 Video Generator")
     with gr.Row():
         with gr.Column():
-            input_img = gr.Image(type="filepath", label="عکس منظره")
-            input_prompt = gr.Textbox(value="A beautiful cinematic landscape, 1 second animation", label="پرامپت")
-            btn = gr.Button("ساخت ویدیو (۱ ثانیه)", variant="primary")
-        
+            img = gr.Image(type="filepath")
+            txt = gr.Textbox(value="moving cinematic eyes, 1 second", label="Prompt")
+            btn = gr.Button("ساخت ویدیو")
         with gr.Column():
-            output_video = gr.Video(label="ویدیو خروجی")
+            out = gr.Video()
 
-    btn.click(fn=generate_video, inputs=[input_img, input_prompt], outputs=output_video)
+    btn.click(generate_video, [img, txt], out)
 
-if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch()