Update app.py

app.py

@@ -2319,7 +2319,7 @@ def apply_example(idx: str):
 
 
 ####################################################################################################
-### PART 20: Gradio UI Layout & Launch (FINAL & PICKLE-SAFE VERSION)
+### PART 20: Gradio UI Layout & Launch (FINAL, CLIENT-SIDE, DUAL-EVENT)
 ####################################################################################################
 import requests
 import time
@@ -2343,25 +2343,47 @@ async (video) => {
 }
 """
 
-def enhance_prompt_api(image_path: str, prompt: str, mode: str) -> str:
-    """این تابع از بک اند پایتون، API بهینه ساز را فراخوانی می کند."""
-    if mode not in ['تبدیل تصویر به ویدیو', 'تکمیل فریم‌های میانی'] or not image_path:
-        return prompt
+# این تابع جاوااسکریپت در مرورگر کاربر اجرا می شود
+# و نتیجه را برای فعال کردن رویداد بعدی برمی گرداند
+js_enhancer_api_call = """
+async (firstFrame, promptText, mode) => {
+    // اگر بهینه سازی لازم نیست، پرامپت اصلی را برگردان
+    if (mode !== 'تبدیل تصویر به ویدیو' && mode !== 'تکمیل فریم‌های میانی' || !firstFrame) {
+        if (!firstFrame) console.warn("Enhancement skipped: First frame image is required.");
+        return promptText;
+    }
 
-    url = 'https://ezmarynoori-vidtolani.hf.space/api/enhance-animation-prompt'
-    try:
-        with open(image_path, 'rb') as f:
-            files = {'image': f}
-            data = {'prompt': prompt, 'is_extension': 'false'}
-            response = requests.post(url, files=files, data=data, timeout=45)
-            response.raise_for_status()
-            result = response.json()
-            enhanced = result.get("animation_prompt", prompt)
-            print(f"Prompt enhanced successfully: {enhanced}")
-            return enhanced
-    except Exception as e:
-        print(f"Could not enhance prompt, falling back to original. Error: {e}")
-        return prompt
+    try {
+        const response = await fetch(firstFrame.url);
+        const imageBlob = await response.blob();
+        
+        const formData = new FormData();
+        const imageFile = new File([imageBlob], "image.jpg", { type: "image/jpeg" });
+        formData.append('image', imageFile);
+        formData.append('prompt', promptText);
+        formData.append('is_extension', 'false');
+
+        const apiResponse = await fetch('https://ezmarynoori-vidtolani.hf.space/api/enhance-animation-prompt', {
+            method: 'POST',
+            body: formData
+        });
+
+        if (!apiResponse.ok) {
+            const errorData = await apiResponse.json();
+            throw new Error(errorData.error || 'خطای ناشناخته از سرور بهینه ساز.');
+        }
+
+        const data = await apiResponse.json();
+        console.log("Enhanced Prompt received:", data.animation_prompt);
+        return data.animation_prompt;
+
+    } catch (error) {
+        console.error("Enhancement API Error:", error);
+        // در صورت خطا، پرامپت اصلی را برگردان تا فرآیند متوقف نشود
+        return promptText;
+    }
+}
+"""
 
 def apply_example(idx: str):
     idx = int(idx)
@@ -2375,48 +2397,35 @@ def apply_example(idx: str):
 
 # --- توابع کنترل کننده در سطح گلوبال تعریف شده اند ---
 
-@spaces.GPU(duration=60)
-def enhancement_step(first_frame, prompt, mode):
-    """رویداد اول: UI را قفل کرده، بهینه سازی را انجام داده و محرک را به روز می کند."""
-    # به‌روزرسانی اول: شروع فرآیند
-    yield (
-        gr.Button(value="⏳ در حال بهینه سازی پرامپت...", interactive=False),
-        None, # output_video
-        gr.Button(visible=False), # download_btn
-        gr.update(), # prompt_ui
-        gr.update() # enhanced_prompt_trigger
-    )
-    
-    enhanced_prompt = enhance_prompt_api(first_frame, prompt, mode)
-    
-    # به‌روزرسانی دوم: پایان بهینه سازی و آماده سازی برای ساخت
-    yield (
-        gr.Button(value="🚀 آماده سازی برای ساخت...", interactive=False),
-        gr.update(),
-        gr.update(),
-        gr.update(value=enhanced_prompt),
-        enhanced_prompt # مقدار دهی محرک برای شروع رویداد بعدی
-    )
+def start_enhancement_ui():
+    """فقط UI را برای شروع بهینه سازی آماده می کند."""
+    return gr.Button(value="⏳ در حال بهینه سازی پرامپت...", interactive=False)
 
 @spaces.GPU(duration=get_duration)
-def generation_step(
+def run_generation_from_trigger(
     first_frame, end_frame, enhanced_prompt, duration, input_video,
     radioanimated_mode, enhance_prompt_flag, seed, randomize_seed,
     height, width, camera_lora, audio_input, progress=gr.Progress(track_tqdm=True)
 ):
-    """رویداد دوم: ویدیو را می سازد."""
+    """رویداد دوم: ویدیو را می سازد و UI را در هر مرحله آپدیت می کند."""
+    # به‌روزرسانی اول: شروع ساخت ویدیو
+    yield {
+        generate_btn: gr.Button(value="🚀 در حال ساخت ویدیو...", interactive=False),
+        prompt_ui: gr.update(value=enhanced_prompt)
+    }
+    
     video_path = generate_video(
         first_frame, end_frame, enhanced_prompt, duration, input_video,
         radioanimated_mode, enhance_prompt_flag, seed, randomize_seed,
         height, width, camera_lora, audio_input, progress
     )
-    # پس از اتمام، UI را نهایی می کند
-    return (
-        gr.Button(value="🤩 ساخت ویدیو", interactive=True),
-        video_path,
-        gr.Button(visible=True)
-    )
-
+    
+    # به‌روزرسانی نهایی: پایان کار
+    yield {
+        generate_btn: gr.Button(value="🤩 ساخت ویدیو", interactive=True),
+        output_video: video_path,
+        download_btn: gr.Button(visible=True)
+    }
 
 with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
     gr.HTML(
@@ -2477,22 +2486,28 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
     audio_ui.change(fn=lambda v: None if v == "__CLEAR__" or not v else gr.update(), inputs=audio_ui, outputs=audio_input, api_visibility="private")
     download_btn.click(fn=None, inputs=[output_video], js=js_download_video)
     
-    # رویداد اول: با کلیک کاربر شروع می شود
+    # رویداد اول: با کلیک کاربر، UI بلافاصله آپدیت شده و سپس JS فراخوانی می شود
+    # این رویداد دکوراتور @spaces.GPU ندارد و بسیار سریع است.
     generate_btn.click(
-        fn=enhancement_step,
+        fn=start_enhancement_ui,
+        outputs=[generate_btn]
+    ).then(
+        fn=None, # هیچ تابع پایتونی در این مرحله اجرا نمی شود
         inputs=[first_frame, prompt, radioanimated_mode],
-        outputs=[generate_btn, output_video, download_btn, prompt_ui, enhanced_prompt_trigger]
+        outputs=[enhanced_prompt_trigger], # نتیجه JS در این محرک ذخیره می شود
+        js=js_enhancer_api_call
     )
 
-    # رویداد دوم: با تغییر مقدار محرک پنهان شروع می شود
+    # رویداد دوم: با تغییر مقدار محرک پنهان (بعد از اتمام کار JS) شروع می شود
+    # این رویداد دکوراتور @spaces.GPU دارد و بودجه زمانی کامل خود را دریافت می کند.
     enhanced_prompt_trigger.change(
-        fn=generation_step,
+        fn=run_generation_from_trigger,
         inputs=[
             first_frame, end_frame, enhanced_prompt_trigger, duration, input_video,
             radioanimated_mode, enhance_prompt_flag_input, seed, randomize_seed,
             height, width, camera_lora, audio_input
         ],
-        outputs=[generate_btn, output_video, download_btn]
+        outputs=[generate_btn, prompt_ui, output_video, download_btn]
     )
 
     # منطق نمونه ها