Update app.py

app.py

@@ -1,61 +1,40 @@
 import gradio as gr
-import os
 import requests
+import os
 
-# Загрузка переменных окружения
 BASE_URL = os.getenv("BASE_URL")
 API_KEY = os.getenv("API_KEY")
 
-# Функция для отправки запроса к модели
-def generate_image(prompt, neg_prompt, sampling_steps=1, cfg_scale=1, seed=-1):
-    # Подготовка данных для запроса
-    data = {
+def generate_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed):
+    url = f"{BASE_URL}/generate"
+    headers = {"Authorization": f"Bearer {API_KEY}"}
+    params = {
         "prompt": prompt,
-        "neg_prompt": neg_prompt,
-        "sampling_steps": int(sampling_steps),
-        "cfg_scale": int(cfg_scale),
-        "seed": int(seed)
+        "negative_prompt": negative_prompt,
+        "sampling_steps": sampling_steps,
+        "cfg_scale": cfg_scale,
+        "seed": seed
     }
 
-    # Установка заголовков для авторизации с использованием API-ключа
-    headers = {
-        "Authorization": f"Bearer {API_KEY}"
-    }
-
-    # Отправка запроса к модели
-    response = requests.post(BASE_URL, json=data, headers=headers)
-
-    # Получение сгенерированного изображения
-    generated_image = response.content
+    response = requests.get(url, headers=headers, params=params)
 
-    return generated_image
+    if response.status_code == 200:
+        return response.content
+    else:
+        return None
 
-# Определение интерфейса Gradio
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.Textbox("text", label="Prompt"),
-        gr.Textbox("text", label="Negative Prompt"),
-        gr.Slider(1, 30, 1, label="Sampling Steps"),
-        gr.Slider(1, 20, 1, label="CFG Scale"),
-        gr.Number(label="Seed")
+        gr.Textbox("prompt", label="Prompt"),
+        gr.Textbox("negative_prompt", label="Negative Prompt"),
+        gr.Slider(1, 30, 1, label="Sampling Steps", default=10),
+        gr.Slider(1, 20, 1, label="CFG Scale", default=10),
+        gr.Textbox("seed", label="Seed (Default: -1)"),
     ],
-    outputs=gr.Image(),
+    outputs=gr.Image(label="Generated Image"),
     live=True,
-    title="Image Generator",
-    theme="compact",
-    interpretation="default",
-    allow_flagging=False,
-    allow_screenshot=False,
-    allow_download=False,
-    allow_duplicate=False,
-    show_input=True,
-    show_output=True,
-    show_interpretation=True,
-    show_flag=False,
-    show_screenshot=False,
-    show_download=False
+    theme="compact"
 )
 
-# Запуск веб-приложения
 iface.launch()