Update app.py

app.py

@@ -5,12 +5,15 @@ import os
 # API ссылка
 url = "https://stablediffusionapi.com/api/v4/dreambooth"
 
+# API ключ
+api_key = os.getenv("KEY")
+
 
 # Функция для отправки запроса
-def render(prompt, negative_prompt, width, height, upscale, api_key):
+def render(model, prompt, negative_prompt, width, height, guidance, seed, upscale):
     data = {
         "key":  api_key,  
-        "model_id": "realistic-vision-51",  
+        "model_id": model,  
         "prompt": prompt, 
         "negative_prompt": negative_prompt,  
         "width": width,  
@@ -19,8 +22,8 @@ def render(prompt, negative_prompt, width, height, upscale, api_key):
         "num_inference_steps": "40",  
         "safety_checker": "no",  
         "enhance_prompt": "yes",  
-        "seed": None,  
-        "guidance_scale": 7.5,  
+        "seed": seed,  
+        "guidance_scale": guidance,  
         "multi_lingual": "no",  
         "panorama": "no",  
         "self_attention": "no",  
@@ -40,12 +43,14 @@ def render(prompt, negative_prompt, width, height, upscale, api_key):
 gr.Interface(
     render,
     inputs=[
+        gr.Dropdown(["realistic-vision-51"], label="Модель", value="realistic-vision-51"),
         gr.Textbox(label="Описание изображения:", placeholder="Введите описание изображения", lines=3),
         gr.Textbox(label="Negative Prompt:", placeholder="Введите Negative Prompt", value="[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry"),
         gr.Slider(show_label=True, minimum=256, maximum=2048, label="Ширина:", value="512", step=1),
         gr.Slider(show_label=True, minimum=256, maximum=2048, label="Высота:", value="512", step=1),
+        gr.Slider(show_label=True, minimum=1, maximum=20, label="CFG Scale:", value="7.5", step=0.5),
+        gr.Slider(show_label=True, label="Seed:", minimum=-1, maximum=1000000, value=-1, step=1),
         gr.Dropdown(["yes", "no"], label="Upscale", value="no"),
-        gr.Textbox(label="Ваш API ключ", placeholder="API_KEY"),
     ],
     outputs=gr.Image()
 ).launch()