Update app.py

app.py

@@ -161,6 +161,8 @@ class StableDiffusionAgent:
     def get_history(self, user_id: str) -> List[Dict]:
         return [entry for entry in self.memory if entry["user_id"] == user_id]
 
+# Update the Gradio interface section as follows:
+
 # ------------------ Gradio Interface ------------------
 def create_web_interface(agent: StableDiffusionAgent):
     css = """
@@ -188,9 +190,19 @@ def create_web_interface(agent: StableDiffusionAgent):
                 steps = gr.Slider(10, 100, value=50, label="Steps")
                 guidance = gr.Slider(1.0, 20.0, value=7.5, label="Guidance Scale")
                 seed = gr.Number(label="Seed (optional)")
-        
+
+        # Modified click handler
         generate_btn.click(
-            fn=lambda *args: agent.generate(*args),
+            fn=lambda user_id, prompt, negative_prompt, style, steps, guidance, seed: 
+                agent.generate(
+                    user_id=user_id,
+                    prompt=prompt,
+                    negative_prompt=negative_prompt,
+                    style=style,
+                    num_inference_steps=steps,
+                    guidance_scale=guidance,
+                    seed=seed
+                ),
             inputs=[user_id, prompt, negative_prompt, style, steps, guidance, seed],
             outputs=[output_image, meta_info]
         )