Update app.py

app.py

@@ -1,55 +1,59 @@
-import os
 import gradio as gr
 import requests
+import os
 
-# Получение URL модели и API-ключа из переменных окружения
+# Замените URL и API_KEY на ваши значения
 BASE_URL = os.getenv("BASE_URL")
 API_KEY = os.getenv("API_KEY")
 
-# Функция для отправки запроса к API модели
+# Функция для выполнения запроса к Inference API
 def generate_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed):
+    url = f"{BASE_URL}/generate"
     headers = {"Authorization": f"Bearer {API_KEY}"}
-    
-    # Параметры для запроса
-    params = {
+
+    data = {
         "prompt": prompt,
         "negative_prompt": negative_prompt,
-        "sampling_steps": int(sampling_steps),
-        "cfg_scale": int(cfg_scale),
-        "seed": int(seed)
+        "sampling_steps": sampling_steps,
+        "cfg_scale": cfg_scale,
+        "seed": seed
     }
-    
-    # Отправка запроса к API модели
-    response = requests.post(BASE_URL, headers=headers, json=params)
-    
-    # Получение сгенерированного изображения
-    generated_image = response.content
-    
-    return generated_image
+
+    response = requests.post(url, headers=headers, json=data)
+
+    # Обработка ответа и возврат сгенерированного изображения
+    if response.status_code == 200:
+        generated_image = response.json()["image"]
+        return generated_image
+    else:
+        return "Ошибка при генерации изображения"
 
 # Определение интерфейса Gradio
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.Textbox(default="Prompt"),
-        gr.Textbox(default="Negative Prompt"),
+        gr.Textbox("text", label="Prompt", default=""),
+        gr.Textbox("text", label="Negative Prompt", default=""),
     ],
-    outputs=gr.Image(type="pil"),
+    outputs=gr.Image(label="Generated Image"),
     live=True,
     layout="vertical",
-    title="Hugging Face Image Generator",
-    description="Generate images using Hugging Face Inference API.",
-    theme="compact",
-    tabs=[
-        "Prompt Input",
-        [
-            gr.Slider(minimum=1, maximum=30, step=1, default=15, label="Sampling Steps"),
-            gr.Slider(minimum=1, maximum=20, step=1, default=10, label="CFG Scale"),
-            gr.Textbox(default="-1", label="Seed"),
-        ]
-    ],
-    interpretation="default"
+    title="Huggingface Image Generator",
+    description="Генератор изображений с использованием Huggingface Inference API",
+    theme="default"
+)
+
+# Добавление второй вкладки
+cfg_tab = gr.Tab(
+    gr.Row(
+        gr.Slider(minimum=1, maximum=30, default=15, label="Sampling Steps", key="sampling_steps"),
+        gr.Slider(minimum=1, maximum=20, default=10, label="CFG Scale", key="cfg_scale"),
+        gr.Textbox("text", label="Seed", default="-1", key="seed"),
+        gr.Button("Сгенерировать", key="generate_button")
+    )
 )
 
-# Запуск интерфейса Gradio
+iface.add_tab("Advanced Settings", cfg_tab)
+
+# Запуск приложения
 iface.launch()