Update app.py

app.py

@@ -1,49 +1,58 @@
 import gradio as gr
-import requests
 import os
+import requests
 
-# Замените "YOUR_MODEL_ENDPOINT_HERE" и "YOUR_BEARER_TOKEN_HERE" на фактический адрес вашей модели Hugging Face и ваш Bearer токен
-MODEL_ENDPOINT = os.getenv("BASE_URL")
-BEARER_TOKEN = os.getenv("API_KEY")
-
-def generate_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed):
-    # Создаем заголовок авторизации Bearer
-    headers = {"Authorization": f"Bearer {BEARER_TOKEN}"}
-
-    # Отправляем запрос к API модели Hugging Face с использованием введенных параметров и заголовка авторизации
-    response = requests.post(
-        MODEL_ENDPOINT,
-        json={
-            "prompt": prompt,
-            "negative_prompt": negative_prompt,
-            "sampling_steps": sampling_steps,
-            "cfg_scale": cfg_scale,
-            "seed": seed
-        },
-        headers=headers
-    )
-
-    # Получаем сгенерированное изображение из ответа API
+# Загрузка переменных окружения
+BASE_URL = os.getenv("BASE_URL")
+API_KEY = os.getenv("API_KEY")
+
+# Функция для отправки запроса к модели
+def generate_image(prompt, neg_prompt, sampling_steps, cfg_scale, seed):
+    # Подготовка данных для запроса
+    data = {
+        "prompt": prompt,
+        "neg_prompt": neg_prompt,
+        "sampling_steps": int(sampling_steps),
+        "cfg_scale": int(cfg_scale),
+        "seed": int(seed)
+    }
+
+    # Установка заголовков для авторизации с использованием API-ключа
+    headers = {
+        "Authorization": f"Bearer {API_KEY}"
+    }
+
+    # Отправка запроса к модели
+    response = requests.post(BASE_URL, json=data, headers=headers)
+
+    # Получение сгенерированного изображения
     generated_image = response.content
 
     return generated_image
 
-def generate_on_button_click(prompt, negative_prompt, sampling_steps, cfg_scale, seed):
-    return generate_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed)
-
+# Определение интерфейса Gradio
 iface = gr.Interface(
-    fn=generate_on_button_click,
+    fn=generate_image,
     inputs=[
         gr.Textbox("text", label="Prompt"),
-        gr.Textbox("text", label="Negative Prompt"),
-        gr.Slider(minimum=1, maximum=30, label="Sampling Steps"),
-        gr.Slider(minimum=0.1, maximum=2.0, label="CFG Scale"),
-        gr.Textbox("number", label="Seed (default -1)"),
-        gr.Button("Generate", generate_on_button_click)
+        gr.Textbox("text", label="Negative Prompt", default=""),
     ],
-    outputs=gr.Image(label="Generated Image"),
+    outputs=gr.Image(),
     live=True,
-    title="Hugging Face Image Generator"
+    title="Image Generator",
+    layout="vertical",
+    theme="compact"
+)
+
+# Добавление второй вкладки
+iface.add_tab(
+    "Advanced Settings",
+    [
+        gr.Slider(minimum=1, maximum=30, default=1, label="Sampling Steps"),
+        gr.Slider(minimum=1, maximum=20, default=1, label="CFG Scale"),
+        gr.Number(default=-1, label="Seed")
+    ]
 )
 
+# Запуск веб-приложения
 iface.launch()