Update app.py

app.py

@@ -1,106 +1,44 @@
-import torch
-import gradio as gr
 import os
-from diffusers import AutoPipelineForText2Image, DPMSolverMultistepScheduler
-
-# ใช้โมเดลเดิม
-MODEL_ID = "Lykon/dreamshaper-xl-turbo"
-
-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())
-
-# พจนานุกรมสำหรับ 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 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
-    final_prompt = STYLE_TEMPLATES[style].format(prompt=prompt)
-    
-    # กำหนดขนาดภาพจาก Aspect Ratio
-    width, height = get_dimensions(aspect_ratio)
-    
-    with torch.no_grad():
-        image = pipe(
-            prompt=final_prompt,
-            negative_prompt=negative_prompt,
-            num_inference_steps=int(steps),
-            guidance_scale=float(cfg),
-            width=width,
-            height=height
-        ).images[0]
-    return image
-
-# UI
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("### 🚀 Pure CPU Turbo (Style & Aspect Ratio Support)")
-    
+import gradio as gr
+from gradio_client import Client, handle_file
+
+# ดึง Token จาก Secrets ของ Space (ตั้งชื่อว่า HF_TOKEN) เพื่อสิทธิพิเศษในการเรียก API
+hf_token = os.getenv("HF_TOKEN")
+
+# เชื่อมต่อไปยัง Space ต้นทางผ่าน API
+# การใช้ Client แบบนี้จะไม่กินทรัพยากรเครื่องของคุณ
+client = Client("selfit-camera/omni-image-editor", hf_token=hf_token)
+
+def api_bridge(input_img, prompt):
+    if input_img is None:
+        return None
+    try:
+        # ส่งข้อมูลไปยัง API ต้นทาง
+        # หมายเหตุ: ตรวจสอบ api_name จาก client.view_api() อีกครั้งหากต้นทางอัปเดต
+        result = client.predict(
+            image=handle_file(input_img),
+            prompt=prompt,
+            api_name="/predict"
+        )
+        return result
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# สร้างหน้า Interface ของคุณเอง
+with gr.Blocks(title="Omni Editor API Bridge") as demo:
+    gr.Markdown("### Omni Image Editor (via API Bridge)")
     with gr.Row():
         with gr.Column():
-            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):
-                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")
-                
-            btn = gr.Button("Generate", variant="primary")
-            
+            img_input = gr.Image(type="filepath", label="Upload Image")
+            prompt_input = gr.Textbox(label="Edit Prompt", placeholder="เช่น 'make it look like a sketch'")
+            submit_btn = gr.Button("Send to API", variant="primary")
         with gr.Column():
-            output_img = gr.Image(label="Result")
+            img_output = gr.Image(label="Result")
 
-    btn.click(
-        fn=gen,
-        inputs=[prompt, negative, style_dropdown, ratio_dropdown, steps, cfg],
-        outputs=[output_img]
+    submit_btn.click(
+        fn=api_bridge,
+        inputs=[img_input, prompt_input],
+        outputs=img_output
     )
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()