Update app.py

app.py

@@ -1,20 +1,29 @@
 import os
-# ติดตั้ง Gradio เวอร์ชันล่าสุด (จำเป็นสำหรับการเชื่อมต่อข้าม Space)
+import sys
+
+# --- ส่วนแก้ไขการ Import ---
 try:
-    import gradio
+    import gradio as gr
 except ImportError:
+    # ถ้ายังไม่มี Gradio ให้ติดตั้งและ import ใหม่
+    print("=== ไม่พบ Gradio กำลังติดตั้ง... ===")
     os.system('pip install gradio --upgrade')
     import gradio as gr
 
 print("=== กำลังเชื่อมต่อกับ Space ต้นทาง (selfit-camera/omni-image-editor) ===")
 print("=== ระบบจะใช้ทรัพยากรจาก Space ต้นทาง ไม่ได้ใช้ GPU ของคุณ ===")
 
-# ใช้ gr.load เพื่อดึงหน้าเว็บและ API จาก Space ต้นทางมาแสดงผลที่นี่
-# src="spaces" บอกว่าเรากำลังดึงมาจาก Spaces ไม่ใช่ Models
-demo = gr.load(name="spaces/selfit-camera/omni-image-editor")
+# โหลด Interface จาก Space ต้นทาง
+# src="spaces" คือคีย์สำคัญที่บอกว่าให้ดึงหน้าเว็บมาแสดงผล ไม่ใช่โหลดโมเดลมาลงเครื่อง
+try:
+    demo = gr.load(name="selfit-camera/omni-image-editor", src="spaces")
+except Exception as e:
+    # เผื่อกรณีใส่ชื่อเต็มแล้ว error ให้ลองใส่แบบนี้
+    print(f"เกิดข้อผิดพลาดในการโหลดแบบแรก: {e}")
+    print("กำลังลองโหลดด้วยวิธีสำรอง...")
+    demo = gr.load(name="spaces/selfit-camera/omni-image-editor")
 
 # รันแอปพลิเคชัน
-# share=True จะสร้างลิงก์ public ชั่วคราว (สำหรับ Colab)
-# ถ้าเอาไปใส่ใน HF Space ของตัวเอง ไม่ต้องใส่ share=True ก็ได้
 if __name__ == "__main__":
-    demo.launch(share=True)
+    # ใน HF Space ไม่จำเป็นต้องใช้ share=True แต่ถ้าเผื่อรันใน Colab ก็ใส่ไว้ได้
+    demo.launch()