Update app.py

app.py

@@ -1,10 +1,9 @@
 import gradio as gr
-from gradio_client import Client
 import os
 import random
-from io import BytesIO
 from PIL import Image
 import base64
+import requests
 
 KEYS = os.getenv("KEYS").split(",")
 
@@ -14,32 +13,38 @@ def get_random_api_key():
 def swap_face_api(source_img, target_img, doFaceEnhancer):
     try:
         api_key = get_random_api_key()
-        client = Client("tuan2308/face-swap")
+        api_url = "https://tuan2308-face-swap-predict.hf.space/api/predict/"  # URL вашего API
 
-        # Кодируем PIL изображения в base64, используя gr.processing_utils
         source_b64 = gr.encode_pil_to_base64(source_img)
         target_b64 = gr.encode_pil_to_base64(target_img)
 
-        result = client.predict(
-            source_file=source_b64,
-            target_file=target_b64,
-            doFaceEnhancer=doFaceEnhancer,
-            api_name="/predict",
-            api_key=api_key #  Передаем api_key в predict
-        )
-        print(result)
-
-        # Декодируем результат из base64 в PIL Image
-        result_decoded = base64.b64decode(result) # Декодируем результат
+        payload = {
+            "data": [
+                source_b64,
+                target_b64,
+                doFaceEnhancer
+            ],
+            "api_key": api_key  # Передаем api_key в payload
+        }
+
+        response = requests.post(api_url, json=payload)
+        response.raise_for_status()  # Вызывает исключение, если есть ошибка
+
+        # Получаем результат из ответа
+        result = response.json()
+        result_data = result["data"]  # Извлекаем данные из ответа
+        result_decoded = base64.b64decode(result_data[0]) # Декодируем base64
+
+
         output_image = Image.open(BytesIO(result_decoded))
         return output_image
 
-    except Exception as e:
-        print(f"Ошибка при вызове API: {e}")
+    except requests.exceptions.RequestException as e:
+        print(f"Ошибка API: {e}")
         return None
 
     except Exception as e:
-        print(f"Ошибка при вызове API: {e}")
+        print(f"Ошибка: {e}")
         return None