Update app.py

app.py

@@ -13,13 +13,30 @@ gen = Generator(handler)
 
 # 2. Define GPU-enabled Inference Function
 @spaces.GPU(duration=35)
-def process_img(image, prompt):
+def process_img(
+    image, 
+    prompt, 
+    cfg_scale, 
+    steps, 
+    img_strength, 
+    depth_strength, 
+    edge_strength
+):
     if image is None:
         raise gr.Error("Please upload an image first.")
     
     try:
         print("--- Starting Generation ---")
-        result = gen.predict(image, prompt)
+        # Pass all parameters to the generator
+        result = gen.predict(
+            image, 
+            prompt,
+            guidance_scale=cfg_scale,
+            num_inference_steps=steps,
+            img2img_strength=img_strength,
+            depth_strength=depth_strength,
+            lineart_strength=edge_strength
+        )
         print("--- Generation Complete ---")
         return result
         
@@ -37,7 +54,7 @@ with gr.Blocks(title="Face To Pixel Art", theme=gr.themes.Soft()) as demo:
     )
     
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=2):
             input_img = gr.Image(type="pil", label="Input Image")
             prompt = gr.Textbox(
                 label="Prompt (Optional)", 
@@ -45,25 +62,81 @@ with gr.Blocks(title="Face To Pixel Art", theme=gr.themes.Soft()) as demo:
                 info="The trigger words 'p1x3l4rt, pixel art' are added automatically."
             )
             run_btn = gr.Button("Generate Pixel Art", variant="primary")
-        
-        with gr.Column():
+            
+            gr.Examples(
+                examples=[
+                    # Add default values for new sliders
+                    ["./examples/monalisa.jpg", "mona lisa portrait", 1.5, 6, 0.3, 0.3, 0.3],
+                    ["./examples/man.jpg", "photo of a man", 1.5, 6, 0.3, 0.3, 0.3],
+                    ["./examples/landscape.jpg", "a beautiful landscape", 1.5, 6, 0.3, 0.3, 0.3]
+                ],
+                # Update inputs to match the new sliders
+                inputs=[input_img, prompt, "cfg_scale", "steps", "img_strength", "depth_strength", "edge_strength"]
+            )
+            
+        with gr.Column(scale=1):
             output_img = gr.Image(label="Pixel Art Result")
-
+            
+            with gr.Accordion("Advanced Settings", open=False):
+                cfg_scale = gr.Slider(
+                    elem_id="cfg_scale",
+                    minimum=1.0, 
+                    maximum=5.0, 
+                    step=0.1, 
+                    value=1.5, 
+                    label="CFG Scale"
+                )
+                steps = gr.Slider(
+                    elem_id="steps",
+                    minimum=4, 
+                    maximum=20, 
+                    step=1, 
+                    value=6, 
+                    label="Steps Number"
+                )
+                img_strength = gr.Slider(
+                    elem_id="img_strength",
+                    minimum=0.1, 
+                    maximum=1.0, 
+                    step=0.05, 
+                    value=0.3, 
+                    label="Image Strength (Img2Img)"
+                )
+                depth_strength = gr.Slider(
+                    elem_id="depth_strength",
+                    minimum=0.0, 
+                    maximum=1.0, 
+                    step=0.05, 
+                    value=0.3, 
+                    label="DepthMap Strength"
+                )
+                edge_strength = gr.Slider(
+                    elem_id="edge_strength",
+                    minimum=0.0, 
+                    maximum=1.0, 
+                    step=0.05, 
+                    value=0.3, 
+                    label="EdgeMap Strength (LineArt)"
+                )
+                
     # Event Handler
+    # Update inputs list for the click event
+    all_inputs = [
+        input_img, 
+        prompt, 
+        cfg_scale, 
+        steps, 
+        img_strength, 
+        depth_strength, 
+        edge_strength
+    ]
+    
     run_btn.click(
         fn=process_img, 
-        inputs=[input_img, prompt], 
+        inputs=all_inputs, 
         outputs=[output_img]
     )
 
-    gr.Examples(
-        examples=[
-            ["./examples/monalisa.jpg", "mona lisa portrait"],
-            ["./examples/man.jpg", "photo of a man"],
-            ["./examples/landscape.jpg", "a beautiful landscape"]
-        ],
-        inputs=[input_img, prompt]
-    )
 
 # 4. Launch the App
 if __name__ == "__main__":