Update app.py

app.py

@@ -1,44 +1,65 @@
 import os
-import gradio as gr
+import json
+from datetime import date
 from gradio_client import Client, handle_file
 
-# ดึง Token จาก Secrets ของ Space (ตั้งชื่อว่า HF_TOKEN) เพื่อสิทธิพิเศษในการเรียก API
-hf_token = os.getenv("HF_TOKEN")
+# --- CONFIGURATION ---
+HF_TOKEN = "ใส่_TOKEN_ของคุณที่นี่" # เพื่อความเสถียรของ API
+SPACE_SOURCE = "selfit-camera/omni-image-editor"
+DAILY_QUOTA = 3  # จำกัดวันละ 3 ภาพ
+QUOTA_FILE = "usage_quota.json"
 
-# เชื่อมต่อไปยัง Space ต้นทางผ่าน API
-# การใช้ Client แบบนี้จะไม่กินทรัพยากรเครื่องของคุณ
-client = Client("selfit-camera/omni-image-editor", hf_token=hf_token)
+# 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 api_bridge(input_img, prompt):
-    if input_img is None:
+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} ภาพสำหรับวันนี้แล้ว")
         return None
+
+    print(f"🔄 กำลังประมวลผล (ภาพที่ {current_count + 1}/{DAILY_QUOTA})...")
+    
     try:
-        # ส่งข้อมูลไปยัง API ต้นทาง
-        # หมายเหตุ: ตรวจสอบ api_name จาก client.view_api() อีกครั้งหากต้นทางอัปเดต
+        client = Client(SPACE_SOURCE, hf_token=HF_TOKEN)
         result = client.predict(
-            image=handle_file(input_img),
+            image=handle_file(image_path),
             prompt=prompt,
             api_name="/predict"
         )
+        
+        update_quota(current_count)
+        print("✅ ประมวลผลสำเร็จ!")
         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():
-            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():
-            img_output = gr.Image(label="Result")
-
-    submit_btn.click(
-        fn=api_bridge,
-        inputs=[img_input, prompt_input],
-        outputs=img_output
-    )
-
-demo.launch()
+        print(f"❌ เกิดข้อผิดพลาดจาก API: {e}")
+        return None
+
+# --- ส่วนการใช้งาน ---
+# ใส่พาธรูปภาพ และ Prompt ที่ต้องการแก้ไข
+my_image = "input.jpg" 
+my_prompt = "change the background to a beach"
+
+output_path = run_editor_api(my_image, my_prompt)
+
+if output_path:
+    print(f"ดาวน์โหลดผลลัพธ์ได้ที่: {output_path}")