Update app.py

app.py

@@ -1,26 +1,61 @@
-from fastapi import FastAPI, Request
+from fastapi import FastAPI, Request, HTTPException
 import time
 import uvicorn
+import httpx
 
 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"""
+    if ip == "unknown" or ip.startswith("127."):
+        return "Local/Unknown"
+    
+    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()
+            if data.get("status") == "success":
+                return f"{data.get('city')}, {data.get('country')}"
+    except Exception:
+        pass
+    return "Unknown"
+
 @app.get("/")
 async def root(request: Request):
+    # 1. Lấy Client IP chuẩn
     headers = request.headers
-    client_ip = headers.get("x-forwarded-for", "unknown").split(",")[0]
+    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)
+    
+    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
+
+    # 3. Lấy vị trí thật (City, Country)
+    real_loc = await get_real_location(client_ip)
     
     return {
         "fl": "998f87",
         "h": headers.get("host", "huggingface.co"),
         "ip": client_ip,
-        "ts": time.time(),
+        "ts": current_time,
         "visit_scheme": "https",
         "uag": headers.get("user-agent", "unknown"),
-        "loc": headers.get("cf-ipcountry", "VN"),
+        "loc": real_loc, # Vị trí thật từ API
         "tls": "TLSv1.3",
         "status": "active"
     }
 
 if __name__ == "__main__":
-    # Quan trọng: Port phải là 7860 và host là 0.0.0.0
+    # Giữ nguyên cấu hình cho Hugging Face Spaces
     uvicorn.run(app, host="0.0.0.0", port=7860)