Update app.py

app.py

@@ -4,6 +4,16 @@ from fastapi import FastAPI, Request
 from sentence_transformers import SentenceTransformer, util
 import torch
 import requests
+# Rate_limit
+from slowapi import Limiter, _rate_limit_exceeded_handler
+from slowapi.util import get_remote_address
+from slowapi.errors import RateLimitExceeded
+
+# Inisialisasi FastAPI dan Limiter
+limiter = Limiter(key_func=get_remote_address)
+app = FastAPI()
+app.state.limiter = limiter
+app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
 
 # 🔐 Paksa semua cache ke path aman di Hugging Face Spaces
 HF_CACHE = "/tmp/hf"
@@ -21,6 +31,13 @@ if os.path.exists(f"{HF_CACHE}/models--sentence-transformers--paraphrase-MiniLM-
 SUPABASE_URL = "https://olbjfxlclotxtnpjvpfj.supabase.co"
 SUPABASE_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6Im9sYmpmeGxjbG90eHRucGp2cGZqIiwicm9sZSI6ImFub24iLCJpYXQiOjE3NTIyMzYwMDEsImV4cCI6MjA2NzgxMjAwMX0.7q_o5DCFEAAysnWXMChH4MI5qNhIVc4OgpT5JvgYxc0"
 
+
+# Inisialisasi rate limiter dan FastAPI
+limiter = Limiter(key_func=get_remote_address)
+app = FastAPI()
+app.state.limiter = limiter
+app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
+
 # 🔄 Gunakan model kecil dan cepat
 model = SentenceTransformer("sentence-transformers/paraphrase-MiniLM-L3-v2")
 
@@ -43,6 +60,7 @@ def get_faq_from_supabase(uid):
         return []
 
 @app.post("/predict")
+@limiter.limit("5/minute")
 async def predict(request: Request):
     body = await request.json()
     uid, question = body.get("data", [None, None])