Update yml/avatar.py

yml/avatar.py

@@ -78,6 +78,7 @@ print(f'   -> User Run ID: {user_run_id}')
 print(f'   -> Target prompt: "{prompt}"')
 print(f'   -> Params: Seed={seed}, Res={res}, Vocal={vocal_mode}, Accel={acceleration}')
 
+# مسیر فایل‌ها در سرور اصلی شما
 img_url = f"{space_url}/static/images/{user_run_id}_avatar_img.{img_ext}"
 aud_url = f"{space_url}/static/images/{user_run_id}_avatar_aud.{aud_ext}"
 state_url = f"{space_url}/static/images/{user_run_id}_state.pt"
@@ -87,56 +88,74 @@ local_aud = f"input_aud.{aud_ext}"
 local_state = "input_state.pt"
 
 print('2. Downloading assets from your Space...')
-try:
-    req_img = requests.get(img_url, timeout=45)
-    if req_img.status_code != 200:
-        raise Exception(f"Image download failed. Status code: {req_img.status_code}")
-    with open(local_img, 'wb') as f:
-        f.write(req_img.content)
-
-    req_aud = requests.get(aud_url, timeout=45)
-    if req_aud.status_code != 200:
-        raise Exception(f"Audio download failed. Status code: {req_aud.status_code}")
-    with open(local_aud, 'wb') as f:
-        f.write(req_aud.content)
-
-except Exception as download_err:
-    err_str = f"Error downloading source files: {download_err}"
-    print(err_str)
-    report_failure(err_str)
+# دانلود عکس با تلاش مجدد
+for att in range(3):
+    try:
+        req_img = requests.get(img_url, timeout=45)
+        if req_img.status_code == 200:
+            with open(local_img, 'wb') as f:
+                f.write(req_img.content)
+            break
+    except:
+        time.sleep(2)
+else:
+    report_failure("Image download failed.")
+    sys.exit(1)
+
+# دانلود صدای اصلی با تلاش مجدد
+for att in range(3):
+    try:
+        req_aud = requests.get(aud_url, timeout=45)
+        if req_aud.status_code == 200:
+            with open(local_aud, 'wb') as f:
+                f.write(req_aud.content)
+            break
+    except:
+        time.sleep(2)
+else:
+    report_failure("Audio download failed.")
     sys.exit(1)
 
-# بررسی هوشمند: آیا فایل وضعیت از اکشن قبلی در سرور موجود است؟
+# بررسی وجود فایل پی‌تی (نشان‌دهنده این که این اکشن، ادامه اکشن قبلی است)
 has_state = False
 try:
-    req_state_head = requests.head(state_url, timeout=10)
-    if req_state_head.status_code == 200:
-        print('   -> Found existing state file! Downloading for continuation...')
-        req_state = requests.get(state_url, timeout=60)
-        if req_state.status_code == 200:
-            with open(local_state, 'wb') as f:
-                f.write(req_state.content)
-            has_state = True
-            print('   -> State file loaded successfully.')
+    req_state = requests.get(state_url, timeout=45)
+    if req_state.status_code == 200:
+        with open(local_state, 'wb') as f:
+            f.write(req_state.content)
+        has_state = True
+        print('   -> State file (.pt) found and loaded to resume rendering.')
 except Exception as e:
-    print(f"   -> No valid state file found. Starting fresh generation...")
+    print("   -> No state file found. Starting fresh generation...")
 
 print('3. Connecting to LongCat-Video-Avatar-1.5 Space on ZeroGPU...')
 try:
     client = Client("https://opera8-longcat-video-avatar-1-5-2nd2.hf.space")
     
-    print(f"--- Generation Phase Started ---")
-    job = client.submit(
-        handle_file(local_img),
-        handle_file(local_aud),
-        prompt,
-        res,
-        seed,
-        vocal_mode,
-        acceleration,
-        handle_file(local_state) if has_state else None,
-        api_name="/generate"
-    )
+    # ارتباط با اسپیس هوش مصنوعی با قابلیت تلاش مجدد
+    max_submit_retries = 3
+    job = None
+    for att in range(max_submit_retries):
+        try:
+            print(f"   -> Submitting job to ZeroGPU (Attempt {att+1})...")
+            job = client.submit(
+                handle_file(local_img),
+                handle_file(local_aud),
+                prompt,
+                res,
+                seed,
+                vocal_mode,
+                acceleration,
+                handle_file(local_state) if has_state else None,
+                api_name="/generate"
+            )
+            break
+        except Exception as e:
+            print(f"   -> Submission error: {e}")
+            time.sleep(5)
+    
+    if not job:
+        raise Exception("Failed to connect to ZeroGPU Space after multiple attempts.")
     
     while not job.done():
         time.sleep(5)
@@ -146,7 +165,7 @@ try:
     state_path = result[1]
     status_msg = result[2]
     
-    # سناریوی اول: ساخت ویدیو به صورت کامل به اتمام رسید
+    # سناریوی اول: ویدیو با موفقیت ساخته شد
     if video_path is not None:
         print('4. Final video generated successfully! Uploading back to your Space...')
         actual_video_path = video_path.get('video') or video_path.get('path') if isinstance(video_path, dict) else video_path
@@ -155,7 +174,8 @@ try:
             res_upload = requests.post(
                 f'{space_url}/api/webhook/upload', 
                 data={'run_id': user_run_id, 'github_run_id': github_run_id, 'ext': 'mp4'}, 
-                files={'file': f}
+                files={'file': f},
+                timeout=120
             )
 
         if res_upload.status_code == 200:
@@ -164,25 +184,26 @@ try:
         else:
             raise Exception(f"Webhook upload failed. Status code: {res_upload.status_code}")
             
-    # سناریوی دوم: مهلت زمانی رانر در ZeroGPU تمام شد و فایل pt دریافت شد (هدایت به اکشن جدید)
+    # سناریوی دوم: نشست تمام شد و فایل pt دریافت شد (هدایت به اکشن جدید)
     elif state_path is not None:
         print(f"4. Timeout reached securely. State file (.pt) received. Preparing next phase...")
         actual_state_path = state_path.get('path') if isinstance(state_path, dict) else state_path
         
-        # ۱. آپلود فایل وضعیت جدید برای ذخیره در سرور اصلی
+        # آپلود فایل وضعیت جدید برای ذخیره در سرور اصلی
         with open(actual_state_path, 'rb') as f:
             res_state_upload = requests.post(
                 f'{space_url}/api/webhook/upload', 
                 data={'run_id': f"{user_run_id}_state", 'github_run_id': github_run_id, 'ext': 'pt'}, 
-                files={'file': f}
+                files={'file': f},
+                timeout=120
             )
         
         if res_state_upload.status_code == 200:
-            print('   -> State file uploaded to server. Triggering new GitHub Action runner...')
+            print('   -> State file uploaded. Triggering a NEW GitHub Action runner to continue...')
             
-            # ۲. ارسال دستور به سرور میانی برای اجرای مجدد یک گیت‌هاب اکشن کاملاً جدید
+            # ارسال دستور به سرور میانی برای اجرای مجدد یک گیت‌هاب اکشن کاملاً جدید
             dispatch_payload = {
-                "prompt": raw_prompt,  # ارسال دقیق همان کانفیگ اولیه
+                "prompt": raw_prompt,  # ارسال دقیق همان کانفیگ اولیه (شامل آیدی یکسان)
                 "width": 1024,
                 "height": 1024,
                 "action_name": "avatar"
@@ -194,14 +215,13 @@ try:
                 timeout=20
             )
             
-            if trigger_res.status_code == 200:
+            if trigger_res.status_code in [200, 204]:
                 print('5. SUCCESS! Next phase dispatched to a new Action runner.')
-                sys.exit(0)  # خروج موفقیت‌آمیز از اکشن فعلی
+                sys.exit(0)  # خروج کاملا موفق و بدون خطا از اکشن فعلی
             else:
                 raise Exception(f"Failed to trigger next phase. Server returned {trigger_res.status_code}")
         else:
             raise Exception(f"Failed to upload state file. Server returned {res_state_upload.status_code}")
-            
     else:
         raise Exception(f"Unexpected result from model: {status_msg}")