Update main.py

main.py

@@ -127,10 +127,19 @@ def generate_prompts(prompt_text: str):
 
 
 
+from fastapi import Query
+
 @app.get("/send_inputs")
-def send_inputs(inputs: str, noise_level: float):
+def send_inputs(
+    inputs: str,
+    noise_level: float,
+    is_negative: bool = False,
+    steps: int = 1,
+    cfg_scale: int = 6,
+    seed: int = None
+):
     try:
-        generated_image = generate_image(inputs)
+        generated_image = generate_image(inputs, is_negative, steps, cfg_scale, seed)
         if generated_image is not None:
             image_bytes = io.BytesIO()
             generated_image.save(image_bytes, format="JPEG")