Update app.py

app.py

@@ -375,7 +375,7 @@ def resize_image(image, max_size=512):
     return image
 
 # ===== Functions =====
-def colorize(sketch, base_model, anime_model, prompt, seed, steps, scale, cn_weight):
+def colorize(sketch, base_model, anime_model, prompt, negative_prompt, seed, steps, scale, cn_weight):
     try:
         # ตรวจสอบว่าเป็น SDXL model หรือไม่
         if is_sdxl_model(base_model):
@@ -410,6 +410,7 @@ def colorize(sketch, base_model, anime_model, prompt, seed, steps, scale, cn_wei
         with torch.inference_mode():
             out = pipe(
                 prompt, 
+                negative_prompt=negative_prompt,
                 image=lineart, 
                 num_inference_steps=int(steps),
                 guidance_scale=float(scale), 
@@ -427,7 +428,7 @@ def colorize(sketch, base_model, anime_model, prompt, seed, steps, scale, cn_wei
         error_img = Image.new('RGB', (512, 512), color='red')
         return error_img, Image.new('RGB', (512, 512), color='gray')
 
-def t2i(prompt, model, seed, steps, scale, w, h):
+def t2i(prompt, negative_prompt, model, seed, steps, scale, w, h):
     try:
         load_t2i_model(model)
         print(f"🖼️ Using T2I model: {model}")
@@ -442,6 +443,7 @@ def t2i(prompt, model, seed, steps, scale, w, h):
                 height = max(int(h), 512)
                 result = CURRENT_T2I_PIPE(
                     prompt, 
+                    negative_prompt=negative_prompt,
                     width=width, 
                     height=height,
                     num_inference_steps=int(steps), 
@@ -451,6 +453,7 @@ def t2i(prompt, model, seed, steps, scale, w, h):
             else:
                 result = CURRENT_T2I_PIPE(
                     prompt, 
+                    negative_prompt=negative_prompt,
                     width=int(w), 
                     height=int(h),
                     num_inference_steps=int(steps), 
@@ -573,9 +576,14 @@ with gr.Blocks(title="🎨 Advanced Image Generation Suite", theme=gr.themes.Sof
                 placeholder="e.g., 1girl, blonde hair, blue eyes, beautiful, masterpiece",
                 lines=2
             )
-            seed = gr.Number(value=42, label="Seed")
+            negative_prompt = gr.Textbox(
+                label="Negative Prompt", 
+                placeholder="e.g., ugly, deformed, bad anatomy, blurry",
+                lines=2
+            )
         
         with gr.Row():
+            seed = gr.Number(value=42, label="Seed")
             steps = gr.Slider(10, 50, 25, step=1, label="Steps")
             scale = gr.Slider(1, 20, 8, step=0.5, label="CFG Scale")
             cn_weight = gr.Slider(0.5, 1.5, 1.0, step=0.1, label="ControlNet Weight")
@@ -583,7 +591,7 @@ with gr.Blocks(title="🎨 Advanced Image Generation Suite", theme=gr.themes.Sof
         run = gr.Button("🎨 Colorize", variant="primary")
         run.click(
             colorize, 
-            [inp, base_model, anime_chk, prompt, seed, steps, scale, cn_weight], 
+            [inp, base_model, anime_chk, prompt, negative_prompt, seed, steps, scale, cn_weight], 
             [out, sketch_out]
         )
 
@@ -603,13 +611,18 @@ with gr.Blocks(title="🎨 Advanced Image Generation Suite", theme=gr.themes.Sof
                 lines=3,
                 placeholder="e.g., a beautiful landscape with mountains and a lake at sunset, highly detailed, 4k"
             )
+            t2i_negative_prompt = gr.Textbox(
+                label="Negative Prompt", 
+                lines=2,
+                placeholder="e.g., blurry, ugly, deformed, low quality"
+            )
+        
+        with gr.Row():
             t2i_model = gr.Dropdown(
                 choices=ALL_MODELS,  # ใช้ทั้ง SD1.5 และ SDXL
                 value="digiplay/ChikMix_V3",
                 label="Model"
             )
-        
-        with gr.Row():
             t2i_seed = gr.Number(value=42, label="Seed")
             t2i_steps = gr.Slider(10, 50, 30, step=1, label="Steps")
             t2i_scale = gr.Slider(1, 20, 7.5, step=0.5, label="CFG Scale")
@@ -622,7 +635,7 @@ with gr.Blocks(title="🎨 Advanced Image Generation Suite", theme=gr.themes.Sof
         gen_btn = gr.Button("🖼️ Generate", variant="primary")
         gen_btn.click(
             t2i, 
-            [t2i_prompt, t2i_model, t2i_seed, t2i_steps, t2i_scale, w, h], 
+            [t2i_prompt, t2i_negative_prompt, t2i_model, t2i_seed, t2i_steps, t2i_scale, w, h], 
             t2i_out
         )