Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 import random
 from PIL import Image
+from io import BytesIO
 import base64
 import requests
 
@@ -13,7 +14,8 @@ def get_random_api_key():
 def swap_face_api(source_img, target_img, doFaceEnhancer):
     try:
         api_key = get_random_api_key()
-        api_url = "https://tuan2308-face-swap-predict.hf.space/api/predict/"  # URL вашего API
+        api_url = "https://tuan2308-face-swap-predict.hf.space/api/predict/"
+
 
         source_b64 = gr.encode_pil_to_base64(source_img)
         target_b64 = gr.encode_pil_to_base64(target_img)
@@ -24,30 +26,26 @@ def swap_face_api(source_img, target_img, doFaceEnhancer):
                 target_b64,
                 doFaceEnhancer
             ],
-            "api_key": api_key  # Передаем api_key в payload
+            "api_key": api_key
         }
 
         response = requests.post(api_url, json=payload)
-        response.raise_for_status()  # Вызывает исключение, если есть ошибка
+        response.raise_for_status()
 
-        # Получаем результат из ответа
         result = response.json()
-        result_data = result["data"]  # Извлекаем данные из ответа
-        result_decoded = base64.b64decode(result_data[0]) # Декодируем base64
-
-
+        result_data = result["data"][0]  #  Убедитесь, что индекс корректный
+        result_decoded = base64.b64decode(result_data)
         output_image = Image.open(BytesIO(result_decoded))
+
         return output_image
 
     except requests.exceptions.RequestException as e:
         print(f"Ошибка API: {e}")
         return None
-
     except Exception as e:
         print(f"Ошибка: {e}")
         return None
 
-
 iface = gr.Interface(
     fn=swap_face_api,
     inputs=[
@@ -59,4 +57,5 @@ iface = gr.Interface(
     title="Face Swap via API"
 )
 
-iface.launch()
+iface.launch(server_name="0.0.0.0", server_port=7860) #  Запускаем на всех интерфейсах
+