Update app.py

app.py

@@ -3,29 +3,35 @@ import gradio as gr
 import os
 from diffusers import AutoPipelineForText2Image, DPMSolverMultistepScheduler
 
-# ชื่อโมเดลที่ถูกต้องสำหรับเรียกผ่าน API ของ Hugging Face
-MODEL_ID = "Tongyi-MAI/Z-Image-Turbo" 
+MODEL_ID = "Lykon/dreamshaper-xl-turbo"
 
-print(f"🚀 Recovering Space with: {MODEL_ID}")
-
-pipe = AutoPipelineForText2Image.from_pretrained(
-    MODEL_ID, 
-    torch_dtype=torch.float32,
-    low_cpu_mem_usage=True
-)
+# โหลดโมเดลและตั้งค่า Never-OOM
+pipe = AutoPipelineForText2Image.from_pretrained(MODEL_ID, torch_dtype=torch.float32, low_cpu_mem_usage=True)
 pipe.to("cpu")
-
-# มาตรการ Never OOM สำหรับเครื่องฟรี
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.enable_attention_slicing("max") 
 pipe.enable_vae_tiling()             
 torch.set_num_threads(os.cpu_count()) 
 
-def gen(prompt, steps, cfg, width, height):
+# ดิกชันนารีเก็บสไตล์ (Prompt สำเร็จรูป)
+STYLE_MAP = {
+    "สมจริง (Cinematic)": "highly detailed film still, cinematic lighting, masterpiece, 8k, realistic skin, professional photography",
+    "อนิเมะ (Anime)": "anime style, high quality, vibrant colors, clean lines, masterpiece, aesthetic",
+    "วาดเขียน (Digital Art)": "detailed digital illustration, sharp focus, artistic, vibrant, concept art",
+    "ไม่เน้นสไตล์": ""
+}
+
+def gen(prompt, style_name, negative_prompt, steps, cfg, width, height):
     if not prompt: return None
+    
+    # รวม Prompt จากปุ่มเลือกสไตล์
+    style_prompt = STYLE_MAP.get(style_name, "")
+    full_prompt = f"{prompt}, {style_prompt}"
+    
     with torch.no_grad():
         image = pipe(
-            prompt=prompt,
+            prompt=full_prompt,
+            negative_prompt=negative_prompt,
             num_inference_steps=int(steps),
             guidance_scale=float(cfg),
             width=int(width),
@@ -33,20 +39,26 @@ def gen(prompt, steps, cfg, width, height):
         ).images[0]
     return image
 
-# UI แบบประหยัด RAM
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown(f"### 🛠️ Z-Image-Turbo CPU Mode")
+    gr.Markdown("### 🚀 Easy One-Click CPU Turbo")
+    
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox(label="Prompt", placeholder="ใส่คำที่อยากเจน...")
-            steps = gr.Slider(1, 10, 4, step=1, label="Steps")
-            cfg = gr.Slider(0.0, 3.0, 1.0, step=0.1, label="CFG (Z-Image ชอบ 1.0)")
-            width = gr.Slider(256, 512, 384, step=64, label="Width")
-            height = gr.Slider(256, 512, 512, step=64, label="Height")
-            btn = gr.Button("Generate", variant="primary")
+            prompt = gr.Textbox(label="1. พิมพ์คำสั้นๆ (เช่น woman, city)", lines=2)
+            style_name = gr.Radio(choices=list(STYLE_MAP.keys()), value="สมจริง (Cinematic)", label="2. เลือกสไตล์ที่ต้องการ")
+            
+            with gr.Accordion("Advanced Settings", open=False):
+                negative = gr.Textbox(label="Negative", value="low quality, blurry, deformed, messy")
+                steps = gr.Slider(1, 10, 4, step=1, label="Steps (แนะนำ 4-6)")
+                cfg = gr.Slider(0.0, 3.0, 1.2, step=0.1, label="CFG Scale")
+                width = gr.Slider(256, 512, 384, step=64, label="Width")
+                height = gr.Slider(256, 512, 512, step=64, label="Height")
+            
+            btn = gr.Button("Generate Now", variant="primary")
+            
         with gr.Column():
             output_img = gr.Image(label="Result")
 
-    btn.click(fn=gen, inputs=[prompt, steps, cfg, width, height], outputs=[output_img])
+    btn.click(fn=gen, inputs=[prompt, style_name, negative, steps, cfg, width, height], outputs=[output_img])
 
 demo.launch()