Update app.py

app.py

@@ -38,12 +38,7 @@ def download_weights():
     if not os.path.exists(motion_model_file):
         print("📥 در حال دانلود مدل‌ها...")
         try:
-            snapshot_download(
-                repo_id=REPO_ID,
-                local_dir=WEIGHTS_DIR,
-                local_dir_use_symlinks=False,
-                resume_download=True,
-            )
+            snapshot_download(repo_id=REPO_ID, local_dir=WEIGHTS_DIR, local_dir_use_symlinks=False, resume_download=True)
         except Exception as e:
             print(f"Error downloading: {e}")
 
@@ -75,37 +70,52 @@ except Exception as e:
 
 emo_name_to_id = {v: k for k, v in emo_map.items()}
 
-# --- جاوا اسکریپت برای ترجمه خطای ZeroGPU ---
-# این اسکریپت دائماً صفحه را چک می‌کند و اگر خطای انگلیسی را دید، آن را فارسی می‌کند
+# --- جاوا اسکریپت هوشمند برای ترجمه خطای ZeroGPU ---
 js_func = """
-function refresh() {
-    const url = new URL(window.location);
-    if (url.searchParams.get('__theme') !== 'light') {
-        url.searchParams.set('__theme', 'light');
-        window.location.href = url.href;
-    }
-    
-    // Observer برای پیدا کردن پیام خطا و ترجمه آن
-    const observer = new MutationObserver(() => {
-        document.querySelectorAll('.toast-body, .error, .prose').forEach(el => {
-            if (el.innerText.includes('ZeroGPU quota exceeded')) {
-                el.innerHTML = '<b>🔴 سهمیه GPU شما تمام شده است!</b><br>لطفاً مدتی صبر کنید تا سهمیه شما شارژ شود و دوباره تلاش کنید.';
-                el.style.direction = 'rtl';
-                el.style.textAlign = 'right';
-                el.style.fontSize = '1.1em';
-            }
-            if (el.innerText.includes('exceeded your GPU quota')) {
-                 el.style.display = 'none'; // مخفی کردن متن انگلیسی اضافی
+() => {
+    const observer = new MutationObserver((mutations) => {
+        mutations.forEach((mutation) => {
+            if (mutation.addedNodes.length) {
+                mutation.addedNodes.forEach((node) => {
+                    if (node.nodeType === 1 && (node.classList.contains('toast-body') || node.classList.contains('error'))) {
+                        const originalText = node.innerText;
+                        
+                        // Regex برای پیدا کردن اعداد در متن خطا
+                        const regex = /(\d+)s requested vs. (\d+)s left/;
+                        const match = originalText.match(regex);
+                        
+                        if (match && !node.dataset.translated) {
+                            const requested = match[1];
+                            const left = match[2];
+                            
+                            // ساخت پیام فارسی با استفاده از اعداد استخراج شده
+                            node.innerHTML = `
+                                <b>🔴 سهمیه GPU کافی نیست!</b>
+                                <hr style='margin: 5px 0; border-top: 1px solid #c82333;'>
+                                <div style='font-size: 0.9em;'>
+                                    زمان درخواستی شما: <b>${requested} ثانیه</b><br>
+                                    اعتبار باقیمانده: <b>${left} ثانیه</b>
+                                </div>
+                            `;
+                            
+                            node.style.direction = 'rtl';
+                            node.style.textAlign = 'right';
+                            node.style.fontFamily = 'Tahoma, sans-serif';
+                            node.dataset.translated = 'true'; // جلوگیری از ترجمه مجدد
+                        }
+                    }
+                });
             }
         });
     });
+
     observer.observe(document.body, { childList: true, subtree: true });
 }
 """
 
 # --- تابع اصلی ---
-# نکته مهم: duration را به 60 کاهش دادم تا با سهمیه باقی‌مانده شما یکی باشد و کمتر ارور بدهد
-@spaces.GPU(duration=60)
+# زمان را روی ۱۲۰ ثانیه تنظیم می‌کنیم تا حداکثر مقدار ممکن باشد
+@spaces.GPU(duration=120)
 def generate_motion(source_image_path, driving_audio_path, persian_emotion_name, cfg_scale, progress=gr.Progress(track_tqdm=True)):
     if pipeline is None:
         raise gr.Error("❌ مدل هنوز بارگذاری نشده است. لطفاً صفحه را رفرش کنید.")
@@ -122,8 +132,6 @@ def generate_motion(source_image_path, driving_audio_path, persian_emotion_name,
         
         wav_audio_path = ensure_wav_format(driving_audio_path)
         
-        print(f"🚀 شروع پردازش: {english_emo_name}, CFG: {cfg_scale}")
-        
         result_video_path = pipeline.driven_sample(
             image_path=source_image_path,
             audio_path=wav_audio_path,
@@ -143,32 +151,14 @@ def generate_motion(source_image_path, driving_audio_path, persian_emotion_name,
     except Exception as e:
         print(f"Error Detail: {e}")
         traceback.print_exc()
-        error_str = str(e).lower()
-        
-        if "quota" in error_str:
-            raise gr.Error("⏳ سهمیه GPU شما تمام شده است.")
-        elif "cuda" in error_str:
-            raise gr.Error("💥 خطای سیستم. لطفاً دوباره تلاش کنید.")
-        else:
-            raise gr.Error(f"❌ خطا: {e}")
+        raise gr.Error(f"❌ خطا در پردازش: {e}")
 
 # --- رابط کاربری ---
 css = """
 .gradio-container {max-width: 900px !important; margin: auto !important; font-family: 'Tahoma', sans-serif;}
 h1, h2, h3, p, span, div {direction: rtl; text-align: right;}
 .center-text {text-align: center !important;}
-
-/* استایل مخصوص برای راست‌چین کردن پیام‌های خطا */
-.toast-wrap {
-    direction: rtl !important;
-    text-align: right !important;
-    font-family: 'Tahoma', sans-serif !important;
-    right: 20px !important;
-    left: auto !important;
-}
-.toast-body {
-    font-size: 16px !important;
-}
+.toast-wrap { direction: rtl !important; }
 """
 
 with gr.Blocks(theme=gr.themes.Soft(), css=css, title="MoDA Farsi", js=js_func) as demo:
@@ -183,10 +173,20 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, title="MoDA Farsi", js=js_func)
 
     with gr.Row():
         with gr.Column():
-            source_image = gr.Image(label="۱. تصویر چهره", type="filepath", height=250)
-            driving_audio = gr.Audio(label="۲. فایل صوتی", type="filepath")
+            # **اضافه شدن مقادیر پیش‌فرض**
+            source_image = gr.Image(
+                label="۱. تصویر چهره", 
+                type="filepath", 
+                value="src/examples/reference_images/7.jpg",
+                height=250
+            )
+            driving_audio = gr.Audio(
+                label="۲. فایل صوتی", 
+                type="filepath", 
+                value="src/examples/driving_audios/5.wav"
+            )
             
-            with gr.Accordion("⚙️ تنظیمات پیشرفته", open=False):
+            with gr.Accordion("⚙️ تنظیمات پیشرفته", open=True):
                 with gr.Row():
                     emotion_dropdown = gr.Dropdown(
                         label="حالت چهره", 
@@ -195,14 +195,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, title="MoDA Farsi", js=js_func)
                     )
                     cfg_slider = gr.Slider(label="دقت (CFG)", minimum=1.0, maximum=3.0, value=1.2, step=0.1)
             
-            gr.Markdown(
-                """
-                <div style="background-color: #fff3cd; color: #856404; padding: 10px; border-radius: 5px; margin-top: 10px; font-size: 0.9em; text-align: right;">
-                ⚠️ <b>نکته:</b> اگر ارور <b>Quota Exceeded</b> دیدید، یعنی سهمیه رایگان تمام شده است.
-                </div>
-                """
-            )
-            
             submit_btn = gr.Button("🎥 ساخت ویدیو", variant="primary", size="lg")
 
         with gr.Column():