Update app.py

app.py

@@ -2492,10 +2492,14 @@ ENHANCE_NO_TEXT = "دستور بهینه نشود ✋"
 
 # --- توابع کنترل کننده (بدون وابستگی محلی) ---
 
-def ui_lock_and_enhance(first_frame, prompt, mode, enhance_toggle):
+def ui_lock_and_enhance(first_frame, prompt, mode, enhance_toggle, audio_input, duration):
     """
     مرحله ۱: قفل کردن UI و بررسی اینکه آیا پرامپت نیاز به بهینه‌سازی دارد یا خیر
     """
+    # بررسی محدودیت زمانی برای فایل صوتی
+    if audio_input is not None and float(duration) > 5.0:
+        raise gr.Error("در صورت وجود همزمان فایل صوتی امکان ساخت فقط تا پنج ثانیه امکان پذیر است")
+
     if enhance_toggle == ENHANCE_YES_TEXT:
         yield (
             gr.Button(value="⏳ در حال بهینه سازی پرامپت (CPU)...", interactive=False),
@@ -2530,6 +2534,10 @@ def ui_generate_video(
     """
     مرحله ۲: ساخت ویدیو با استفاده از GPU
     """
+    # بررسی مجدد محدودیت زمانی برای اطمینان در مرحله ساخت
+    if audio_input is not None and float(duration) > 5.0:
+        raise gr.Error("در صورت وجود همزمان فایل صوتی امکان ساخت فقط تا پنج ثانیه امکان پذیر است")
+
     video_path = generate_video(
         first_frame, end_frame, enhanced_prompt, duration, input_video,
         radioanimated_mode, False, seed, randomize_seed,
@@ -2574,7 +2582,7 @@ relocate_js = r"""
                 m.addedNodes.forEach(node => {
                     if (node.nodeType === 1) {
                         const txt = (node.textContent || "").toLowerCase();
-                        if (txt.includes("quota") || txt.includes("exceeded") || txt.includes("error") || txt.includes("خطا")) {
+                        if (txt.includes("quota") || txt.includes("exceeded") || txt.includes("error") || txt.includes("خطا") || txt.includes("فایل صوتی")) {
                             errorFound = true;
                         }
                         if (node.classList && node.classList.contains("toast-wrap")) {
@@ -2685,7 +2693,7 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
                 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", "6s", "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")
@@ -2778,7 +2786,7 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
     # زنجیره اصلی ساخت ویدیو
     generate_btn.click(
         fn=ui_lock_and_enhance,
-        inputs=[first_frame, prompt, radioanimated_mode, enhance_toggle],
+        inputs=[first_frame, prompt, radioanimated_mode, enhance_toggle, audio_input, duration],
         outputs=[generate_btn, prompt_ui, prompt, output_video]
     ).then(
         fn=ui_generate_video,