Update app.py

app.py

@@ -1,11 +1,19 @@
 import gradio as gr
 from gradio_client import Client, handle_file
+import os
+import random
 
-# Создаем клиента для API
-client = Client("tuan2308/face-swap")
+KEYS = os.getenv("KEYS").split(",")
+
+def get_random_api_key():
+    return random.choice(KEYS)
 
 def swap_face_api(source_img, target_img, doFaceEnhancer):
     try:
+        api_key = get_random_api_key()
+
+        client = Client("tuan2308/face-swap", api_key=api_key)
+
         result = client.predict(
             source_file=handle_file(source_img),
             target_file=handle_file(target_img),
@@ -15,9 +23,8 @@ def swap_face_api(source_img, target_img, doFaceEnhancer):
         return result
     except Exception as e:
         print(f"Ошибка при вызове API: {e}")
-        return None  # Или какое-то изображение-заглушку
+        return None
 
-# Создаем интерфейс
 iface = gr.Interface(
     fn=swap_face_api,
     inputs=[
@@ -30,4 +37,3 @@ iface = gr.Interface(
 )
 
 iface.launch()
-