Update app.py

app.py

@@ -1,65 +1,47 @@
 import os
-import json
-from datetime import date
+import gradio as gr
 from gradio_client import Client, handle_file
 
-# --- CONFIGURATION ---
-HF_TOKEN = "ใส่_TOKEN_ของคุณที่นี่" # เพื่อความเสถียรของ API
-SPACE_SOURCE = "selfit-camera/omni-image-editor"
-DAILY_QUOTA = 3  # จำกัดวันละ 3 ภาพ
-QUOTA_FILE = "usage_quota.json"
+# Fetch Token from Space Secrets (Name it HF_TOKEN in Settings)
+HF_TOKEN = os.getenv("HF_TOKEN")
+TARGET_SPACE = "selfit-camera/omni-image-editor"
 
-# 1. ฟังก์ชันตรวจสอบโควต้า
-def check_quota():
-    today = str(date.today())
-    if not os.path.exists(QUOTA_FILE):
-        usage_data = {"date": today, "count": 0}
-    else:
-        with open(QUOTA_FILE, "r") as f:
-            usage_data = json.load(f)
-        
-        if usage_data["date"] != today:
-            usage_data = {"date": today, "count": 0}
-            
-    if usage_data["count"] >= DAILY_QUOTA:
-        return False, usage_data["count"]
-    return True, usage_data["count"]
-
-def update_quota(current_count):
-    with open(QUOTA_FILE, "w") as f:
-        json.dump({"date": str(date.today()), "count": current_count + 1}, f)
-
-# 2. ฟังก์ชันเรียก API
-def run_editor_api(image_path, prompt):
-    can_run, current_count = check_quota()
-    
-    if not can_run:
-        print(f"❌ ขออภัย: คุณใช้โควต้าครบ {DAILY_QUOTA} ภาพสำหรับวันนี้แล้ว")
+def bridge_api(image_path, prompt):
+    if not image_path:
         return None
-
-    print(f"🔄 กำลังประมวลผล (ภาพที่ {current_count + 1}/{DAILY_QUOTA})...")
     
     try:
-        client = Client(SPACE_SOURCE, hf_token=HF_TOKEN)
+        # Direct connection to the target API
+        client = Client(TARGET_SPACE, hf_token=HF_TOKEN)
+        
+        # API prediction call
+        # Note: Ensure parameter names match target space's client.view_api()
         result = client.predict(
             image=handle_file(image_path),
             prompt=prompt,
             api_name="/predict"
         )
-        
-        update_quota(current_count)
-        print("✅ ประมวลผลสำเร็จ!")
         return result
     except Exception as e:
-        print(f"❌ เกิดข้อผิดพลาดจาก API: {e}")
-        return None
-
-# --- ส่วนการใช้งาน ---
-# ใส่พาธรูปภาพ และ Prompt ที่ต้องการแก้ไข
-my_image = "input.jpg" 
-my_prompt = "change the background to a beach"
+        return f"API Bridge Error: {str(e)}"
 
-output_path = run_editor_api(my_image, my_prompt)
-
-if output_path:
-    print(f"ดาวน์โหลดผลลัพธ์ได้ที่: {output_path}")
+# UI Layout
+with gr.Blocks(title="Omni Editor Proxy") as demo:
+    gr.Markdown("## 🚀 Omni Image Editor (Proxy Mode)")
+    gr.Markdown("Processing is handled by the backend API and returned here.")
+    
+    with gr.Row():
+        with gr.Column():
+            input_img = gr.Image(type="filepath", label="Input Image")
+            input_prompt = gr.Textbox(label="Edit Prompt", placeholder="e.g., 'change background to sunset'")
+            submit_btn = gr.Button("Process", variant="primary")
+        with gr.Column():
+            output_img = gr.Image(label="Output")
+
+    submit_btn.click(
+        fn=bridge_api,
+        inputs=[input_img, input_prompt],
+        outputs=output_img
+    )
+
+demo.launch()