Update app.py

app.py

@@ -3,62 +3,104 @@ import gradio as gr
 import os
 from diffusers import AutoPipelineForText2Image, DPMSolverMultistepScheduler
 
+# ใช้โมเดลเดิม
 MODEL_ID = "Lykon/dreamshaper-xl-turbo"
 
-# โหลดโมเดลและตั้งค่า Never-OOM
-pipe = AutoPipelineForText2Image.from_pretrained(MODEL_ID, torch_dtype=torch.float32, low_cpu_mem_usage=True)
+print("Loading for Pure CPU Environment (No GPU)...")
+
+# โหลดโมเดลแบบ CPU
+pipe = AutoPipelineForText2Image.from_pretrained(
+    MODEL_ID, 
+    torch_dtype=torch.float32,
+    low_cpu_mem_usage=True,
+    safety_checker=None,
+    requires_safety_checker=False
+)
 pipe.to("cpu")
+
+# [SAFE CPU OPTIMIZATIONS]
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.enable_attention_slicing("max") 
-pipe.enable_vae_tiling()             
-torch.set_num_threads(os.cpu_count()) 
+pipe.enable_vae_tiling()
+torch.set_num_threads(os.cpu_count())
 
-# ดิกชันนารีเก็บสไตล์ (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",
-    "ไม่เน้นสไตล์": ""
+# พจนานุกรมสำหรับ Style
+STYLE_TEMPLATES = {
+    "None": "{prompt}",
+    "Cinematic": "cinematic photo, {prompt}, high resolution, detailed, 8k, masterpiece",
+    "Digital Art": "digital art, {prompt}, vibrant colors, detailed, trending on artstation",
+    "Anime": "anime style, {prompt}, vivid, expressive, high quality art",
+    "Oil Painting": "oil painting, {prompt}, brush strokes, classical style, textured canvas",
+    "Photography": "professional photography, {prompt}, realistic, 8k uhd, dslr, high detail"
 }
 
-def gen(prompt, style_name, negative_prompt, steps, cfg, width, height):
+# ฟังก์ชันคำนวณสัดส่วนภาพ
+def get_dimensions(ratio_str):
+    ratios = {
+        "1:1 (Square)": (512, 512),
+        "4:3 (Classic)": (512, 384),
+        "3:4 (Portrait)": (384, 512),
+        "16:9 (Widescreen)": (512, 288),
+        "9:16 (Vertical)": (288, 512)
+    }
+    return ratios.get(ratio_str, (512, 512))
+
+def gen(prompt, negative_prompt, style, aspect_ratio, steps, cfg):
     if not prompt: return None
     
-    # รวม Prompt จากปุ่มเลือกสไตล์
-    style_prompt = STYLE_MAP.get(style_name, "")
-    full_prompt = f"{prompt}, {style_prompt}"
+    # รวม Prompt กับ Style
+    final_prompt = STYLE_TEMPLATES[style].format(prompt=prompt)
+    
+    # กำหนดขนาดภาพจาก Aspect Ratio
+    width, height = get_dimensions(aspect_ratio)
     
     with torch.no_grad():
         image = pipe(
-            prompt=full_prompt,
+            prompt=final_prompt,
             negative_prompt=negative_prompt,
             num_inference_steps=int(steps),
             guidance_scale=float(cfg),
-            width=int(width),
-            height=int(height)
+            width=width,
+            height=height
         ).images[0]
     return image
 
+# UI
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("### 🚀 Easy One-Click CPU Turbo")
+    gr.Markdown("### 🚀 Pure CPU Turbo (Style & Aspect Ratio Support)")
     
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox(label="1. พิมพ์คำสั้นๆ (เช่น woman, city)", lines=2)
-            style_name = gr.Radio(choices=list(STYLE_MAP.keys()), value="สมจริง (Cinematic)", label="2. เลือกสไตล์ที่ต้องการ")
+            prompt = gr.Textbox(label="Prompt", lines=3, placeholder="What do you want to create?")
+            
+            with gr.Row():
+                style_dropdown = gr.Dropdown(
+                    choices=list(STYLE_TEMPLATES.keys()), 
+                    value="None", 
+                    label="Select Style"
+                )
+                ratio_dropdown = gr.Dropdown(
+                    choices=["1:1 (Square)", "4:3 (Classic)", "3:4 (Portrait)", "16:9 (Widescreen)", "9:16 (Vertical)"], 
+                    value="1:1 (Square)", 
+                    label="Aspect Ratio"
+                )
+
+            negative = gr.Textbox(label="Negative", value="deformed, lowres, blurry, nsfw")
             
             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)")
+                steps = gr.Slider(1, 10, 4, step=1, label="Steps (Turbo recommended 1-4)")
                 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")
+                
+            btn = gr.Button("Generate", variant="primary")
             
         with gr.Column():
             output_img = gr.Image(label="Result")
 
-    btn.click(fn=gen, inputs=[prompt, style_name, negative, steps, cfg, width, height], outputs=[output_img])
+    btn.click(
+        fn=gen,
+        inputs=[prompt, negative, style_dropdown, ratio_dropdown, steps, cfg],
+        outputs=[output_img]
+    )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()