Update app.py

app.py

@@ -1,29 +1,37 @@
+import gradio as gr
 import os
-import sys
 
-# --- ส่วนแก้ไขการ Import ---
-try:
-    import gradio as gr
-except ImportError:
-    # ถ้ายังไม่มี Gradio ให้ติดตั้งและ import ใหม่
-    print("=== ไม่พบ Gradio กำลังติดตั้ง... ===")
-    os.system('pip install gradio --upgrade')
-    import gradio as gr
+# ชื่อ Space ต้นทาง
+source_space = "selfit-camera/omni-image-editor"
+src_url = f"https://huggingface.co/spaces/{source_space}"
+
+print(f"=== กำลังพยายามเชื่อมต่อกับ {source_space} ===")
 
-print("=== กำลังเชื่อมต่อกับ Space ต้นทาง (selfit-camera/omni-image-editor) ===")
-print("=== ระบบจะใช้ทรัพยากรจาก Space ต้นทาง ไม่ได้ใช้ GPU ของคุณ ===")
+# ดึง HF Token จาก Settings (ถ้ามี) เพื่อช่วยในการ Wake up space
+hf_token = os.getenv("HF_TOKEN")
 
-# โหลด Interface จาก Space ต้นทาง
-# src="spaces" คือคีย์สำคัญที่บอกว่าให้ดึงหน้าเว็บมาแสดงผล ไม่ใช่โหลดโมเดลมาลงเครื่อง
 try:
-    demo = gr.load(name="selfit-camera/omni-image-editor", src="spaces")
+    # วิธีที่ 1: พยายามโหลดแบบ Native API (ดีที่สุด)
+    # ใส่ hf_token เพื่อช่วยปลุก Space ที่หลับอยู่ได้ดีขึ้น
+    demo = gr.load(name=source_space, src="spaces", hf_token=hf_token)
+    print("=== เชื่อมต่อ API สำเร็จ! ===")
+
 except Exception as e:
-    # เผื่อกรณีใส่ชื่อเต็มแล้ว error ให้ลองใส่แบบนี้
-    print(f"เกิดข้อผิดพลาดในการโหลดแบบแรก: {e}")
-    print("กำลังลองโหลดด้วยวิธีสำรอง...")
-    demo = gr.load(name="spaces/selfit-camera/omni-image-editor")
+    # วิธีที่ 2: ถ้า API พัง หรือ Space หลับ (เจอ JSONDecodeError) -> ใช้ Iframe แทน
+    print(f"=== ไม่สามารถโหลด API ได้ ({e}) ===")
+    print("=== กำลังสลับไปใช้โหมด Iframe เพื่อแสดงผลหน้าเว็บโดยตรง ===")
+    
+    with gr.Blocks() as demo:
+        gr.Markdown(f"### ⚠️ เชื่อมต่อ API โดยตรงไม่ได้ (Space ต้นทางอาจกำลังหลับ) \nระบบสลับมาใช้โหมดแสดงผลหน้าเว็บแทน คุณยังสามารถใช้งานได้ตามปกติ")
+        # ฝังหน้าเว็บต้นทางลงไปเลย (วิธีนี้ไม่มีวัน Error)
+        gr.HTML(f"""
+            <iframe src="{src_url}" 
+            width="100%" 
+            height="800px" 
+            style="border:none;">
+            </iframe>
+        """)
 
 # รันแอปพลิเคชัน
 if __name__ == "__main__":
-    # ใน HF Space ไม่จำเป็นต้องใช้ share=True แต่ถ้าเผื่อรันใน Colab ก็ใส่ไว้ได้
     demo.launch()