Update app.py

app.py

@@ -3,35 +3,44 @@ import gradio as gr
 import os
 from diffusers import AutoPipelineForText2Image, DPMSolverMultistepScheduler
 
-# --- เปลี่ยนชื่อโมเดลที่พังๆ ตรงนี้ได้เลย ---
-# เช่น "stabilityai/sdxl-turbo", "RunDiffusion/Juggernaut-XL-v9", ฯลฯ
-MODEL_ID = "models/Tongyi-MAI/Z-Image-Turbo" 
+# ตั้งค่า Model ID เป็นตัวที่คุณต้องการ
+MODEL_ID = "Tongyi-MAI/Z-Image-Turbo" 
 
-print(f"Starting Recovery Mode for: {MODEL_ID}")
+print(f"🚀 Initializing Recovery Mode for: {MODEL_ID}")
 
-# 1. โหลดแบบประหยัด RAM ขั้นสุด
+# 1. โหลดโมเดลด้วยระบบ Low RAM Usage
 pipe = AutoPipelineForText2Image.from_pretrained(
     MODEL_ID, 
-    torch_dtype=torch.float32, # บังคับ float32 เพราะ CPU ไม่รองรับ float16 ในบางคำสั่ง
+    torch_dtype=torch.float32, 
     low_cpu_mem_usage=True
 )
 
-# 2. ย้ายเข้า CPU และตัดการเชื่อมต่อกับ CUDA Driver (เพื่อไม่ให้ Exit Code: 1)
+# 2. บังคับรันบน CPU เพื่อหลีกเลี่ยง Error เกี่ยวกับ CUDA Driver
 pipe.to("cpu")
 
-# 3. ใส่เกราะกันระเบิด (Never OOM)
+# 3. มาตรการ Never OOM: หั่นการประมวลผลเป็นส่วนย่อย
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.enable_attention_slicing("max") 
 pipe.enable_vae_tiling()             
-
-# 4. รีดพลัง CPU ทั้งหมดที่มี
 torch.set_num_threads(os.cpu_count()) 
 
-def gen(prompt, steps, cfg, width, height):
+# ระบบ Style สำเร็จรูปเพื่อความสะดวก
+STYLE_MAP = {
+    "สมจริง (Realistic)": "hyperrealistic, 8k, highly detailed, cinematic lighting",
+    "อนิเมะ (Anime)": "anime style, vibrant colors, masterpiece, clean lines",
+    "ศิลปะดิจิทัล (Digital Art)": "detailed digital painting, concept art, artistic",
+    "ดั้งเดิม (None)": ""
+}
+
+def gen(prompt, style, steps, cfg, width, height):
     if not prompt: return None
+    
+    # รวม Prompt กับ Style เข้าด้วยกัน
+    full_prompt = f"{prompt}, {STYLE_MAP[style]}" if style != "ดั้งเดิม (None)" else prompt
+    
     with torch.no_grad():
         image = pipe(
-            prompt=prompt,
+            prompt=full_prompt,
             num_inference_steps=int(steps),
             guidance_scale=float(cfg),
             width=int(width),
@@ -39,20 +48,31 @@ def gen(prompt, steps, cfg, width, height):
         ).images[0]
     return image
 
-# UI แบบคลีนๆ ใช้ง่าย
-with gr.Blocks() as demo:
-    gr.Markdown(f"### 🛠️ Space Recovery: {MODEL_ID}")
+# จัดหน้าจอ UI ให้ใช้ง่ายและดูดี
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown(f"## 🛠️ Space Recovery: {MODEL_ID}")
+    gr.Markdown("สถานะ: **CPU Mode (Never OOM Enabled)**")
+    
     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.2, step=0.1, label="CFG")
-            width = gr.Slider(256, 512, 384, step=64, label="Width")
-            height = gr.Slider(256, 512, 512, step=64, label="Height")
-            btn = gr.Button("Recover & Generate", variant="primary")
-        with gr.Column():
-            output_img = gr.Image(label="Result")
-
-    btn.click(fn=gen, inputs=[prompt, steps, cfg, width, height], outputs=[output_img])
+        with gr.Column(scale=1):
+            prompt = gr.Textbox(label="คำสั่ง (Prompt)", placeholder="เช่น futuristic city, cyberpunk style", lines=3)
+            style = gr.Radio(choices=list(STYLE_MAP.keys()), value="ดั้งเดิม (None)", label="เลือกสไตล์")
+            
+            with gr.Accordion("การตั้งค่าขั้นสูง (Advanced)", open=False):
+                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 (แนะนำ 1.0-1.5)")
+                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")
+            
+        with gr.Column(scale=1):
+            output_img = gr.Image(label="ผลลัพธ์", interactive=False)
+
+    btn.click(
+        fn=gen, 
+        inputs=[prompt, style, steps, cfg, width, height], 
+        outputs=[output_img]
+    )
 
 demo.launch()