Update backend/api.py

backend/api.py

@@ -13,7 +13,7 @@ app = FastAPI()
 # === CORS untuk frontend (misal dari Vercel) ===
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],      # bisa disesuaikan agar lebih aman
+    allow_origins=["*"],      # Bisa dibatasi ke domain Vercel kamu untuk keamanan
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -34,6 +34,7 @@ try:
     tokenizer = AutoTokenizer.from_pretrained(str(TOKENIZER_PATH))
     session = ort.InferenceSession(str(MODEL_PATH), providers=["CPUExecutionProvider"])
     print("✅ Model dan tokenizer berhasil dimuat!")
+    print("📥 Model expects inputs:", [i.name for i in session.get_inputs()])
 except Exception as e:
     print("❌ ERROR saat memuat model/tokenizer:", e)
     traceback.print_exc()
@@ -49,8 +50,6 @@ id2label = {
     6: "skills",
 }
 
-label2id = {v: k for k, v in id2label.items()}
-
 # === Kamus respon SAMA dengan training ===
 responses = {
     "about_me": "I am a passionate developer specializing in AI and web development.",
@@ -81,10 +80,11 @@ async def chatbot(req: ChatRequest):
         # === Tokenisasi input ===
         inputs = tokenizer(req.text, return_tensors="np", padding=True, truncation=True, max_length=128)
 
-        # === Siapkan input untuk ONNX ===
-        ort_inputs = {k: v.astype(np.int64) for k, v in inputs.items()}
+        # === Ambil hanya input yang cocok dengan model ONNX ===
+        expected_inputs = [i.name for i in session.get_inputs()]
+        ort_inputs = {k: v.astype(np.int64) for k, v in inputs.items() if k in expected_inputs}
 
-        # === Inferensi ===
+        # === Inferensi ONNX ===
         ort_outputs = session.run(None, ort_inputs)
         logits = ort_outputs[0]
         pred_id = np.argmax(logits, axis=1)[0]
@@ -98,5 +98,6 @@ async def chatbot(req: ChatRequest):
         return {"reply": reply, "intent": intent}
 
     except Exception as e:
+        print("❌ Runtime error:", e)
         traceback.print_exc()
         return {"reply": "⚠️ Internal server error.", "intent": intent}