Update app.py

app.py

@@ -97,22 +97,17 @@ def infer(prompt, negative="low_quality", style_name=None, guidance_scale=None):
     return image_b64
 
 
-@app.route("/", methods=["POST", "GET"])
-async def generate_image(request: Request):
-    if request.method == "POST":
-        body = await request.body()
-        data = json.loads(body)
-        if 'prompt' in data and 'style_name' in data and 'guidance_scale' in data:
-            prompt = data['prompt']
-            style_name = data['style_name']
-            guidance_scale = data['guidance_scale']
-            
-            image_urls = infer(prompt, style_name=style_name, guidance_scale=guidance_scale)
+@app.post("/")
+def generate_image(data: dict):
+    if 'prompt' in data and 'style_name' in data and 'guidance_scale' in data:
+        prompt = data['prompt']
+        style_name = data['style_name']
+        guidance_scale = data['guidance_scale']]
+        image_urls = infer(prompt, style_name=style_name, guidance_scale=guidance_scale)
             
             # Convert the first generated image to base64
-            image_b64 = image_urls[0].split(",")[1]
-            
-            return {"image_base64": image_b64}
+        image_b64 = image_urls[0].split(",")[1]
+        return {"image_base64": image_b64}
         else:
             raise HTTPException(status_code=400, detail="Missing required parameters")
     else: