Update app.py

app.py

@@ -2360,6 +2360,58 @@ async (video) => {
 }
 """
 
+# JS Function to Optimize Prompt Client-Side via External API
+# This prevents using the Space's GPU for prompt enhancement
+js_enhance_prompt = """
+async (image, prompt, enhance_enabled) => {
+    // اگر بهینه‌سازی غیرفعال باشد یا تصویری وجود نداشته باشد، تغییری نده
+    if (!enhance_enabled || !image) {
+        return [prompt, enhance_enabled];
+    }
+
+    try {
+        // تبدیل آدرس فایل گرادیو به Blob برای ارسال به API
+        let blob = null;
+        if (typeof image === 'string') {
+            const res = await fetch(image);
+            blob = await res.blob();
+        } else if (image instanceof File || image instanceof Blob) {
+            blob = image;
+        }
+
+        if (!blob) return [prompt, enhance_enabled];
+
+        // آماده‌سازی داده‌ها طبق الگوی HTML ارسالی کاربر
+        const formData = new FormData();
+        formData.append('image', blob);
+        formData.append('prompt', prompt);
+        formData.append('is_extension', 'false');
+
+        // ارسال به سرور خارجی جهت بهینه‌سازی (سمت کلاینت)
+        const response = await fetch('https://ezmarynoori-vidtolani.hf.space/api/enhance-animation-prompt', {
+            method: 'POST',
+            body: formData
+        });
+
+        const data = await response.json();
+
+        if (data.animation_prompt) {
+            // 1. بازگرداندن پرامپت بهینه شده
+            // 2. بازگرداندن مقدار False برای چک‌باکس enhance_prompt
+            // این کار باعث می‌شود تابع پایتون تصور کند بهینه‌سازی خاموش است و از GPU استفاده نکند
+            return [data.animation_prompt, false];
+        } else {
+            console.error("External API Error:", data);
+            return [prompt, enhance_enabled];
+        }
+
+    } catch (error) {
+        console.error("Enhance Prompt Failed:", error);
+        return [prompt, enhance_enabled];
+    }
+}
+"""
+
 def apply_example(idx: str):
     idx = int(idx)
 
@@ -2530,10 +2582,7 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
             with gr.Column(elem_id="step-column"):
                 output_video = gr.Video(label="ویدیوی ساخته شده", autoplay=True, height=512)
                 
-                # --- دکمه دانلود (تغییرات ظاهری) ---
-                # scale=0 باعث می‌شود دکمه تمام عرض را نگیرد
-                # variant="secondary" رنگ متفاوت (معمولا خاکستری/سفید)
-                # visible=False به صورت پیش‌فرض مخفی است
+                # --- دکمه دانلود ---
                 with gr.Row():
                     download_btn = gr.Button(
                         "📥 دانلود ویدیو", 
@@ -2641,8 +2690,15 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
         api_visibility="private"
     )
 
-    # --- منطق تولید ویدیو و نمایش دکمه دانلود ---     
+    # --- منطق اصلی (بهینه سازی سمت کلاینت + اجرا) ---     
+    # مرحله 1: اجرای جاوااسکریپت برای بهینه‌سازی پرامپت (بدون درگیری GPU سرور)
     generate_btn.click(
+        fn=None, # تابع پایتونی ندارد
+        inputs=[first_frame, prompt, enhance_prompt],
+        outputs=[prompt, enhance_prompt], # خروجی: پرامپت جدید و خاموش کردن تیک بهینه‌سازی
+        js=js_enhance_prompt,
+    ).then(
+        # مرحله 2: اجرای تابع ساخت ویدیو (با پرامپت بهینه شده و enhance=False)
         fn=generate_video,
         inputs=[
             first_frame,
@@ -2651,7 +2707,7 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
             duration,
             input_video,
             radioanimated_mode,
-            enhance_prompt,
+            enhance_prompt, # مقدار این الان False شده است (توسط مرحله 1)
             seed,
             randomize_seed,
             height,
@@ -2661,7 +2717,7 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
         ],
         outputs=[output_video]
     ).then(
-        fn=lambda: gr.update(visible=True), # فقط بعد از اتمام ساخت، دکمه ظاهر می‌شود
+        fn=lambda: gr.update(visible=True), # نمایش دکمه دانلود
         outputs=[download_btn]
     )