Update app.py

app.py

@@ -1,61 +1,60 @@
 from fastapi import FastAPI, Request, HTTPException
 import time
 import uvicorn
-import httpx
+import json
+from urllib.request import urlopen
 
 app = FastAPI()
 
-# Lưu trữ thời gian request cuối cùng của mỗi IP
-# { "1.2.3.4": 1712345678.9 }
 last_request_time = {}
 
-async def get_real_location(ip: str):
-    """Gọi API bên thứ 3 để lấy thông tin vị trí thật từ IP"""
+def get_real_location_sync(ip: str):
+    """Lấy vị trí bằng thư viện chuẩn urllib (không cần cài thêm gì)"""
     if ip == "unknown" or ip.startswith("127."):
-        return "Local/Unknown"
-    
+        return "VN"
     try:
-        async with httpx.AsyncClient() as client:
-            # Sử dụng ip-api.com (miễn phí cho thử nghiệm)
-            response = await client.get(f"http://ip-api.com/json/{ip}")
-            data = response.json()
+        # Gọi API lấy vị trí (timeout 2s để không làm chậm app)
+        with urlopen(f"http://ip-api.com/json/{ip}", timeout=2) as response:
+            data = json.loads(response.read().decode())
             if data.get("status") == "success":
                 return f"{data.get('city')}, {data.get('country')}"
-    except Exception:
+    except:
         pass
-    return "Unknown"
+    return "VN"
 
 @app.get("/")
 async def root(request: Request):
-    # 1. Lấy Client IP chuẩn
     headers = request.headers
-    forwarded = headers.get("x-forwarded-for")
-    client_ip = forwarded.split(",")[0] if forwarded else "unknown"
-
-    # 2. Rate Limiting: 1 request / 0.5s
-    current_time = time.time()
-    last_time = last_request_time.get(client_ip, 0)
+    client_ip = headers.get("x-forwarded-for", "unknown").split(",")[0]
     
-    if current_time - last_time < 0.5:
-        raise HTTPException(status_code=429, detail="Too Many Requests - Wait 0.5s")
-    
-    last_request_time[client_ip] = current_time
+    # --- Rate Limiting (0.5s) ---
+    now = time.time()
+    if client_ip in last_request_time and (now - last_request_time[client_ip] < 0.5):
+        raise HTTPException(status_code=429, detail="Slow down!")
+    last_request_time[client_ip] = now
 
-    # 3. Lấy vị trí thật (City, Country)
-    real_loc = await get_real_location(client_ip)
+    # --- Lấy vị trí thật ---
+    # Nếu chạy trên Hugging Face, nó có sẵn header 'cf-ipcity' và 'cf-ipcountry'
+    # Bạn có thể ưu tiên lấy từ header cho nhanh, nếu không có mới gọi API
+    city = headers.get("cf-ipcity", "")
+    country = headers.get("cf-ipcountry", "VN")
     
+    if city:
+        real_loc = f"{city}, {country}"
+    else:
+        real_loc = get_real_location_sync(client_ip)
+
     return {
         "fl": "998f87",
         "h": headers.get("host", "huggingface.co"),
         "ip": client_ip,
-        "ts": current_time,
+        "ts": now,
         "visit_scheme": "https",
         "uag": headers.get("user-agent", "unknown"),
-        "loc": real_loc, # Vị trí thật từ API
+        "loc": real_loc,
         "tls": "TLSv1.3",
         "status": "active"
     }
 
 if __name__ == "__main__":
-    # Giữ nguyên cấu hình cho Hugging Face Spaces
     uvicorn.run(app, host="0.0.0.0", port=7860)