Update yml/generate-editor.py

yml/generate-editor.py

@@ -185,36 +185,42 @@ except Exception as dl_err:
     sys.exit(1)
 
 # =========================================================
-# بخش افزوده شده: ارتباط با CodeFormer برای افزایش کیفیت و بزرگ‌نمایی تصویر خروجی
+# بخش بهبودیافته: تلاش چندجانبه برای ارتباط موفق با API اسپیس CodeFormer
 print('4.5. Connecting to CodeFormer to restore and upscale (Factor=4, Fidelity=1.0)...')
 try:
     cf_client = Client('sczhou/CodeFormer')
     upscale_result = None
     
-    # تست با الگوی ۶ پارامتری (حاوی پارامتر face_align در پلتفرم جدید)
-    try:
-        print('   -> Attempting CodeFormer with 6-parameter interface...')
-        upscale_result = cf_client.predict(
-            handle_file(filename), # تصویر ورودی ویرایش شده
-            True,                  # Face Align
-            True,                  # Background Enhance
-            True,                  # Face Upsample
-            4,                     # Rescaling Factor (تنظیم شده بر روی ۴)
-            1.0,                   # CodeFormer Fidelity (تنظیم شده بر روی ۱.۰)
-            api_name='/predict'
-        )
-    except Exception as cf_e6:
-        print(f'   -> 6-parameter attempt failed: {cf_e6}. Falling back to 5-parameter interface...')
-        # متد جایگزین ۵ پارامتری برای نسخه‌های قدیمی اسپیس CodeFormer
-        upscale_result = cf_client.predict(
-            handle_file(filename), # تصویر ورودی ویرایش شده
-            True,                  # Background Enhance
-            True,                  # Face Upsample
-            4,                     # Rescaling Factor (تنظیم شده بر روی ۴)
-            1.0,                   # CodeFormer Fidelity (تنظیم شده بر روی ۱.۰)
-            api_name='/predict'
-        )
-        
+    # ساختار مانیتورینگ متدهای متوالی برای جلوگیری از بروز خطای api_name
+    methods_to_try = [
+        # روش ۱: استفاده از ایندکس صفر به همراه ساختار ۶ پارامتری (ویرایش جدید اسپیس)
+        {"args": [handle_file(filename), True, True, True, 4, 1.0], "kwargs": {"fn_index": 0}},
+        # روش ۲: استفاده از ایندکس صفر به همراه ساختار ۵ پارامتری (نسخه‌های پایدار قدیمی)
+        {"args": [handle_file(filename), True, True, 4, 1.0], "kwargs": {"fn_index": 0}},
+        # روش ۳: تلاش با ایندکس یک به همراه ساختار ۶ پارامتری
+        {"args": [handle_file(filename), True, True, True, 4, 1.0], "kwargs": {"fn_index": 1}},
+        # روش ۴: تلاش با ایندکس یک به همراه ساختار ۵ پارامتری
+        {"args": [handle_file(filename), True, True, 4, 1.0], "kwargs": {"fn_index": 1}},
+        # روش ۵: فراخوانی اسمی بدون اسلش (predict) با ساختار ۶ پارامتری
+        {"args": [handle_file(filename), True, True, True, 4, 1.0], "kwargs": {"api_name": "predict"}},
+        # روش ۶: فراخوانی اسمی بدون اسلش (predict) با ساختار ۵ پارامتری
+        {"args": [handle_file(filename), True, True, 4, 1.0], "kwargs": {"api_name": "predict"}},
+        # روش ۷: فراخوانی اسمی با اسلش (/predict) با ساختار ۶ پارامتری
+        {"args": [handle_file(filename), True, True, True, 4, 1.0], "kwargs": {"api_name": "/predict"}},
+        # روش ۸: فراخوانی اسمی با اسلش (/predict) با ساختار ۵ پارامتری
+        {"args": [handle_file(filename), True, True, 4, 1.0], "kwargs": {"api_name": "/predict"}},
+    ]
+    
+    for idx, method in enumerate(methods_to_try):
+        try:
+            print(f'   -> Trying execution path {idx + 1} of {len(methods_to_try)}...')
+            upscale_result = cf_client.predict(*method["args"], **method["kwargs"])
+            if upscale_result:
+                print(f'   -> Success! CodeFormer handled by path {idx + 1}.')
+                break
+        except Exception as method_err:
+            print(f'   -> Path {idx + 1} bypassed: {method_err}')
+            
     upscale_path = None
     if isinstance(upscale_result, list) and len(upscale_result) > 0:
         upscale_path = upscale_result[0]
@@ -226,7 +232,7 @@ try:
         upscale_path = upscale_result
 
     if upscale_path and os.path.exists(str(upscale_path)):
-        print(f'   -> CodeFormer Success! Saved locally at {upscale_path}')
+        print(f'   -> CodeFormer Success! Output saved at {upscale_path}')
         upscale_ext = 'png'
         if str(upscale_path).endswith('.webp'):
             upscale_ext = 'webp'
@@ -246,7 +252,7 @@ try:
         else:
             print(f'   -> Enhanced image upload failed: {res_up.status_code} - {res_up.text}')
     else:
-        print('   -> Warning: CodeFormer did not return a valid image file path.')
+        print('   -> Warning: CodeFormer did not return a valid file path.')
 except Exception as cf_err:
     print(f'   -> Warning: CodeFormer process bypassed or failed: {cf_err}')
 # =========================================================