Update app.py

app.py

@@ -6,35 +6,39 @@ BASE_URL = os.getenv("BASE_URL")
 API_KEY = os.getenv("API_KEY")
 
 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,
-        "negative_prompt": negative_prompt,
-        "sampling_steps": sampling_steps,
-        "cfg_scale": cfg_scale,
-        "seed": seed
+    headers = {
+        'Authorization': f'Bearer {API_KEY}',
     }
 
-    response = requests.get(url, headers=headers, params=params)
+    data = {
+        'prompt': prompt,
+        'negative_prompt': negative_prompt,
+        'sampling_steps': sampling_steps,
+        'cfg_scale': cfg_scale,
+        'seed': seed,
+    }
+
+    response = requests.post(BASE_URL, headers=headers, json=data)
 
     if response.status_code == 200:
-        return response.content
+        image_data = response.json()['image']
+        return image_data
     else:
         return None
 
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.Textbox("prompt", label="Prompt"),
-        gr.Textbox("negative_prompt", label="Negative Prompt"),
-        gr.Slider(1, 30, 1, label="Sampling Steps", value=10),
-        gr.Slider(1, 20, 1, label="CFG Scale", value=10),
-        gr.Textbox("seed", label="Seed (Default: -1)"),
+        gr.Textbox('text', label='Prompt', default='Enter your prompt...'),
+        gr.Textbox('text', label='Negative Prompt', default='Enter your negative prompt...'),
     ],
-    outputs=gr.Image(label="Generated Image"),
+    outputs=gr.Image(label='Generated Image'),
     live=True,
-    theme="compact"
 )
 
-iface.launch()
+with iface.launch(debug=True):
+    iface.add_slider('Sampling Steps', 1, 30, 1)
+    iface.add_slider('CFG Scale', 1, 20, 1)
+    iface.add_textbox('Seed', default='-1')
+    iface.add_button('Generate', None)
+    iface.add_download_button('Download Image')