Update main.py

main.py

@@ -1,54 +1,41 @@
-
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 import json
 import os
 
-app = FastAPI(title="TANZOS Update API")
+app = FastAPI()
 
-# อนุญาตให้เว็บจากภายนอกเข้าถึง API นี้ได้ (CORS)
+# --- แก้ไขจุดนี้: เพิ่ม CORS เพื่อให้หน้าเว็บเข้าถึง API ได้ ---
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # เปิดรับทุกโดเมน (หรือใส่เฉพาะโดเมนของเว็บคุณเพื่อความปลอดภัย)
-    allow_credentials=True,
+    allow_origins=["*"],
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-DATA_FILE = "tanzos_update.json"
+# ไฟล์เก็บข้อมูลอัพเดท
+DB_FILE = "update_data.json"
 
 class UpdateData(BaseModel):
     version: str
     description: str
     script: str
 
-@app.get("/")
-def read_root():
-    return {"message": "TANZOS Update API is running on Hugging Face!"}
-
-# ดึงข้อมูลอัพเดทล่าสุด
 @app.get("/api/update")
-def get_update():
-    if os.path.exists(DATA_FILE):
-        try:
-            with open(DATA_FILE, "r", encoding="utf-8") as f:
-                return json.load(f)
-        except Exception as e:
-            return {"error": str(e)}
-    # ถ้ายังไม่มีไฟล์ ให้ส่งค่า default กลับไป
-    return {"version": "1.0", "description": "No update available yet.", "script": ""}
+async def get_update():
+    if os.path.exists(DB_FILE):
+        with open(DB_FILE, "r") as f:
+            return json.load(f)
+    return {"version": "1.0", "description": "ระบบเริ่มต้น", "script": ""}
 
-# บันทึกข้อมูลอัพเดทจากหน้าแอดมิน (/a1234)
 @app.post("/api/update")
-def post_update(data: UpdateData):
-    update_dict = data.dict()
-    try:
-        with open(DATA_FILE, "w", encoding="utf-8") as f:
-            json.dump(update_dict, f, ensure_ascii=False, indent=4)
-        return {"status": "success", "message": "TANZOS Update published successfully!"}
-    except Exception as e:
-        return {"status": "error", "message": str(e)}
-
-# หากใช้งานบน Hugging Face Docker Space จะถูกรันอัตโนมัติผ่านคำสั่งใน Dockerfile
-# เช่น CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+async def post_update(data: UpdateData):
+    with open(DB_FILE, "w") as f:
+        json.dump(data.dict(), f)
+    return {"status": "success"}
+
+# ตรวจสอบว่าไฟล์รันได้ปกติ
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)