Update app.py

app.py

@@ -2,14 +2,14 @@ import gradio as gr
 import requests
 import os
 
-# Замените URL и API_KEY на ваши значения
-BASE_URL = os.getenv("BASE_URL")
-API_KEY = os.getenv("API_KEY")
-
-# Функция для выполнения запроса к Inference 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}"}
+    url = os.getenv("BASE_URL")
+    api_key = os.getenv("API_KEY")
+    
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
 
     data = {
         "prompt": prompt,
@@ -20,40 +20,29 @@ def generate_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed):
     }
 
     response = requests.post(url, headers=headers, json=data)
+    generated_image = response.content  # Получаем изображение из ответа API
+    return generated_image
 
-    # Обработка ответа и возврат сгенерированного изображения
-    if response.status_code == 200:
-        generated_image = response.json()["image"]
-        return generated_image
-    else:
-        return "Ошибка при генерации изображения"
+# Создание интерфейса Gradio
+inputs_tab1 = [
+    gr.inputs.Textbox(label="Prompt", lines=3),
+    gr.inputs.Textbox(label="Negative Prompt", lines=3)
+]
 
-# Определение интерфейса Gradio
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=[
-        gr.Textbox("text", label="Prompt", default=""),
-        gr.Textbox("text", label="Negative Prompt", default=""),
-    ],
-    outputs=gr.Image(label="Generated Image"),
-    live=True,
-    layout="vertical",
-    title="Huggingface Image Generator",
-    description="Генератор изображений с использованием Huggingface Inference API",
-    theme="default"
-)
+inputs_tab2 = [
+    gr.inputs.Slider(1, 30, default=1, label="Sampling Steps", step=1),
+    gr.inputs.Slider(1, 20, default=1, label="CFG Scale", step=1),
+    gr.inputs.Number(default=-1, label="Seed")
+]
 
-# Добавление второй вкладки
-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")
-    )
-)
+outputs = gr.outputs.Image(label="Generated Image")
 
-iface.add_tab("Advanced Settings", cfg_tab)
-
-# Запуск приложения
-iface.launch()
+gr.Interface(
+    fn=generate_image,
+    inputs=[inputs_tab1, inputs_tab2],
+    outputs=outputs,
+    title="HuggingFace Image Generator",
+    allow_flagging=False,  # Отключаем возможность пометить результат
+    theme="compact",  # Выбираем компактную тему интерфейса
+    server_name="gradio.app"
+).launch()