Update app.py

app.py

@@ -2461,38 +2461,49 @@ def apply_example(idx: str):
 
 # --- توابع کنترل کننده (بدون وابستگی محلی) ---
 
-def ui_lock_and_enhance(first_frame, prompt, mode):
+def ui_lock_and_enhance(first_frame, prompt, mode, enhance_toggle):
     """
-    مرحله ۱: قفل کردن UI و اجرای بهینه سازی روی CPU
+    مرحله ۱: قفل کردن UI و بررسی اینکه آیا پرامپت نیاز به بهینه‌سازی دارد یا خیر
     """
-    yield (
-        gr.Button(value="⏳ در حال بهینه سازی پرامپت (CPU)...", interactive=False),
-        gr.update(), 
-        gr.update(), 
-        None 
-    )
-    
-    enhanced = enhance_prompt_on_cpu(first_frame, prompt, mode)
-    
-    yield (
-        gr.Button(value="🚀 در حال ساخت ویدیو (GPU)...", interactive=False),
-        gr.update(value=enhanced), 
-        enhanced, 
-        None
-    )
+    if enhance_toggle == "✨ دستور بهینه شود":
+        yield (
+            gr.Button(value="⏳ در حال بهینه سازی پرامپت (CPU)...", interactive=False),
+            gr.update(), 
+            gr.update(), 
+            None 
+        )
+        
+        enhanced = enhance_prompt_on_cpu(first_frame, prompt, mode)
+        
+        yield (
+            gr.Button(value="🚀 در حال ساخت ویدیو (GPU)...", interactive=False),
+            gr.update(value=enhanced), 
+            enhanced, 
+            None
+        )
+    else:
+        # اگر کاربر نخواست پرامپت بهینه شود، مستقیما به مرحله تولید می‌رویم
+        yield (
+            gr.Button(value="🚀 در حال ساخت ویدیو (GPU)...", interactive=False),
+            gr.update(value=prompt), 
+            prompt, 
+            None
+        )
 
 @spaces.GPU(duration=get_duration)
 def ui_generate_video(
     first_frame, end_frame, enhanced_prompt, duration, input_video,
-    radioanimated_mode, enhance_prompt_flag, seed, randomize_seed,
+    radioanimated_mode, enhance_toggle, seed, randomize_seed,
     height, width, camera_lora, audio_input, progress=gr.Progress(track_tqdm=True)
 ):
     """
     مرحله ۲: ساخت ویدیو با استفاده از GPU
     """
+    # پارامتر enhance_prompt (که مقدار آن False داده شده) برای جلوگیری از اعمال بهینه‌ساز داخلی روی GPU است، 
+    # چون ما قبلا در مرحله قبل (CPU) این کار را طبق خواسته کاربر انجام داده‌ایم.
     video_path = generate_video(
         first_frame, end_frame, enhanced_prompt, duration, input_video,
-        radioanimated_mode, enhance_prompt_flag, seed, randomize_seed,
+        radioanimated_mode, False, seed, randomize_seed,
         height, width, camera_lora, audio_input, progress
     )
     return (
@@ -2636,20 +2647,16 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
                 audio_ui = AudioDropUpload(target_audio_elem_id="audio_input_hidden", elem_id="audio_ui")
                 prompt = gr.Textbox(label="Prompt", value="این تصویر را با حرکت سینمایی و انیمیشن روان زنده کن", lines=3, max_lines=3, visible=False)
                 
-                enhance_prompt_flag_input = gr.Checkbox(label="Enhance Prompt", value=False, visible=False) 
-                
                 with gr.Accordion("تنظیمات پیشرفته", open=False, visible=False):
                     seed = gr.Slider(label="سید (Seed)", minimum=0, maximum=MAX_SEED, value=DEFAULT_SEED, step=1)
                     randomize_seed = gr.Checkbox(label="استفاده از سید تصادفی", value=True)
             
             with gr.Column(elem_id="step-column"):
-                # پارامتر interactive=False در اینجا اضافه شد تا امکان آپلود غیرفعال شود
                 output_video = gr.Video(label="ویدیوی ساخته شده", autoplay=True, height=512, interactive=False)
                 with gr.Row():
                     download_btn = gr.Button("📥 دانلود ویدیو", variant="secondary", size="sm", scale=0, visible=False)
                 with gr.Row(elem_id="controls-row"):
-                    # ۸ ثانیه به لیست اضافه شد
-                    duration_ui = CameraDropdown(choices=["3s", "5s", "8s", "10s"], value="5s", title="مدت زمان ویدیو", elem_id="duration_ui")
+                    duration_ui = CameraDropdown(choices=["3s", "5s", "7s", "10s"], value="5s", title="مدت زمان ویدیو", elem_id="duration_ui")
                     duration = gr.Slider(label="Duration (seconds)", minimum=1.0, maximum=10.0, value=5.0, step=0.1, visible=False)
                     ICON_16_9, ICON_1_1, ICON_9_16 = ("<svg viewBox='0 0 24 24' fill='none' aria-hidden='true'><rect x='3' y='7' width='18' height='10' rx='2' stroke='currentColor' stroke-width='2'/></svg>", "<svg viewBox='0 0 24 24' fill='none' aria-hidden='true'><rect x='6' y='6' width='12' height='12' rx='2' stroke='currentColor' stroke-width='2'/></svg>", "<svg viewBox='0 0 24 24' fill='none' aria-hidden='true'><rect x='7' y='3' width='10' height='18' rx='2' stroke='currentColor' stroke-width='2'/></svg>")
                     resolution_ui = CameraDropdown(choices=[{"label": "16:9", "value": "16:9", "icon": ICON_16_9}, {"label": "1:1", "value": "1:1", "icon": ICON_1_1}, {"label": "9:16", "value": "9:16", "icon": ICON_9_16}], value="16:9", title="ابعاد تصویر", elem_id="resolution_ui")
@@ -2658,9 +2665,17 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
                     camera_ui = CameraDropdown(choices=[name for name, _ in VISIBLE_RUNTIME_LORA_CHOICES], value="No LoRA", title="افکت دوربین (LoRA)", elem_id="camera_ui")
                     camera_lora = gr.Dropdown(label="Camera Control LoRA", choices=[name for name, _ in VISIBLE_RUNTIME_LORA_CHOICES], value="No LoRA", visible=False)
                 
+                # اضافه کردن سوییچ متحرک به شکل قرص برای تنظیمات پرامپت دقیقاً بالای دکمه تولید
+                gr.HTML("<style>#enhance-toggle-row { display: flex; justify-content: center; margin-bottom: 15px; margin-top: 5px; width: 100%; }</style>")
+                with gr.Row(elem_id="enhance-toggle-row"):
+                    enhance_toggle = RadioAnimated(
+                        choices=["✨ دستور بهینه شود", "✋ دستور بهینه نشود"], 
+                        value="✨ دستور بهینه شود", 
+                        elem_id="enhance_toggle"
+                    )
+
                 generate_btn = gr.Button("🤩 ساخت ویدیو", variant="primary", elem_classes="button-gradient", elem_id="generate_btn")
                 
-                # توجه مهم: این دکمه دیگر visible=False نیست تا از DOM حذف نشود. توسط CSS بالا مخفی شده است.
                 reset_btn = gr.Button("Reset", elem_id="reset_btn")
 
     # اتصال رویدادها
@@ -2672,8 +2687,6 @@ 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)
     
-    # *** رویداد جادویی برای شکستن قفل گاردریو ***
-    # پارامتر queue=False باعث می‌شود اگر گاردریو در صف گیر کرده باشد، این دستور معطل نشود و فوراً اجرا شود
     reset_btn.click(
         fn=lambda: gr.Button(value="🤩 ساخت ویدیو", interactive=True),
         inputs=[],
@@ -2682,16 +2695,16 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
         queue=False 
     )
     
-    # زنجیره اصلی ساخت ویدیو
+    # زنجیره اصلی ساخت ویدیو - حالا از مقدار کلید enhance_toggle استفاده می‌کند
     generate_btn.click(
         fn=ui_lock_and_enhance,
-        inputs=[first_frame, prompt, radioanimated_mode],
+        inputs=[first_frame, prompt, radioanimated_mode, enhance_toggle],
         outputs=[generate_btn, prompt_ui, prompt, output_video]
     ).then(
         fn=ui_generate_video,
         inputs=[
             first_frame, end_frame, prompt, 
-            duration, input_video, radioanimated_mode, enhance_prompt_flag_input, 
+            duration, input_video, radioanimated_mode, enhance_toggle, 
             seed, randomize_seed, height, width, camera_lora, audio_input
         ],
         outputs=[generate_btn, output_video, download_btn]