Update app.py

app.py

@@ -17,12 +17,15 @@ gen = Generator(handler)
 @spaces.GPU(duration=35)
 def process_img(
     image, 
-    prompt, 
+    prompt,
+    negative_prompt,    # <-- ADDED
     cfg_scale, 
     steps, 
     img_strength, 
     depth_strength, 
-    edge_strength
+    edge_strength,
+    color_strength,     # <-- ADDED
+    seed                # <-- ADDED
 ):
     if image is None:
         raise gr.Error("Please upload an image first.")
@@ -33,11 +36,14 @@ def process_img(
         result = gen.predict(
             image, 
             prompt,
+            negative_prompt=negative_prompt,    # <-- ADDED
             guidance_scale=cfg_scale,
             num_inference_steps=steps,
             img2img_strength=img_strength,
             depth_strength=depth_strength,
-            lineart_strength=edge_strength
+            lineart_strength=edge_strength,
+            color_strength=color_strength,      # <-- ADDED
+            seed=seed                           # <-- ADDED
         )
         print("--- Generation Complete ---")
         return result
@@ -64,14 +70,26 @@ with gr.Blocks(title="Face To Pixel Art", theme=gr.themes.Soft()) as demo:
                 info="The trigger words 'p1x3l4rt, pixel art' are added automatically."
             )
             
-            # --- MOVED ACCORDION HERE ---
+            # --- ADDED: Negative Prompt ---
+            negative_prompt = gr.Textbox(
+                label="Negative Prompt (Optional)", 
+                placeholder="e.g., blurry, text, watermark, bad art..."
+            )
+            
             with gr.Accordion("Advanced Settings", open=False):
+                # --- ADDED: Seed ---
+                seed = gr.Number(
+                    label="Seed", 
+                    value=-1, 
+                    info="-1 for random", 
+                    precision=0
+                )
+                
                 cfg_scale = gr.Slider(
                     elem_id="cfg_scale",
                     minimum=1.0, 
                     maximum=5.0, 
                     step=0.1, 
-                    # --- USE CONFIG DEFAULT ---
                     value=Config.CGF_SCALE, 
                     label="CFG Scale"
                 )
@@ -80,7 +98,6 @@ with gr.Blocks(title="Face To Pixel Art", theme=gr.themes.Soft()) as demo:
                     minimum=4, 
                     maximum=20, 
                     step=1, 
-                    # --- USE CONFIG DEFAULT ---
                     value=Config.STEPS_NUMBER, 
                     label="Steps Number"
                 )
@@ -89,7 +106,6 @@ with gr.Blocks(title="Face To Pixel Art", theme=gr.themes.Soft()) as demo:
                     minimum=0.1, 
                     maximum=1.0, 
                     step=0.05, 
-                    # --- USE CONFIG DEFAULT ---
                     value=Config.IMG_STRENGTH, 
                     label="Image Strength (Img2Img)"
                 )
@@ -98,7 +114,6 @@ with gr.Blocks(title="Face To Pixel Art", theme=gr.themes.Soft()) as demo:
                     minimum=0.0, 
                     maximum=1.0, 
                     step=0.05, 
-                    # --- USE CONFIG DEFAULT ---
                     value=Config.DEPTH_STRENGTH, 
                     label="DepthMap Strength"
                 )
@@ -107,31 +122,37 @@ with gr.Blocks(title="Face To Pixel Art", theme=gr.themes.Soft()) as demo:
                     minimum=0.0, 
                     maximum=1.0, 
                     step=0.05, 
-                    # --- USE CONFIG DEFAULT ---
                     value=Config.EDGE_STRENGTH, 
                     label="EdgeMap Strength (LineArt)"
                 )
-            # --- END OF MOVED BLOCK ---
-
-            run_btn = gr.Button("Generate Pixel Art", variant="primary")
+                # --- ADDED: Color Strength ---
+                color_strength = gr.Slider(
+                    elem_id="color_strength",
+                    minimum=0.0, 
+                    maximum=1.0, 
+                    step=0.05, 
+                    value=Config.COLOR_STRENGTH, 
+                    label="ColorMap Strength"
+                )
             
-            # --- REMOVED gr.Examples BLOCK ---
+            run_btn = gr.Button("Generate Pixel Art", variant="primary")
             
         with gr.Column(scale=1):
             output_img = gr.Image(label="Pixel Art Result")
-            
-            # The accordion was here, but has been moved to the left column
                 
     # Event Handler
     # Update inputs list for the click event
     all_inputs = [
         input_img, 
         prompt, 
+        negative_prompt,    # <-- ADDED
         cfg_scale, 
         steps, 
         img_strength, 
         depth_strength, 
-        edge_strength
+        edge_strength,
+        color_strength,     # <-- ADDED
+        seed                # <-- ADDED
     ]
     
     run_btn.click(