Update app.py

app.py

@@ -51,7 +51,7 @@ if not os.path.exists(CHARACTERS_FOLDER):
     os.makedirs(CHARACTERS_FOLDER)
 
 
-def generate_flux_image(final_prompt, is_negative, steps, cfg_scale, seed, strength):
+def generate_flux_image(final_prompt, is_negative, steps, cfg_scale, seed, strength, width, height, sampler):
     """
     Generate an image using the FLUX.1-Schnell-Serverless-enhanced model via Gradio client.
     Returns the image path, seed used, and prompt.
@@ -67,11 +67,11 @@ def generate_flux_image(final_prompt, is_negative, steps, cfg_scale, seed, stren
             is_negative=is_negative,
             steps=steps,
             cfg_scale=cfg_scale,
-            sampler="Heun",  # Default to Heun sampler
+            sampler=sampler,
             seed=seed,
             strength=strength,
-            width=512,  # Default width
-            height=512,  # Default height
+            width=width,
+            height=height,
             api_name="/query"
         )
         
@@ -85,6 +85,7 @@ def generate_flux_image(final_prompt, is_negative, steps, cfg_scale, seed, stren
         print(f"Error in image generation: {str(e)}")
         raise gr.Error(f"Image generation failed: {str(e)}")
 
+        
 
 
 def get_character_images():