Update app.py

app.py

@@ -37,8 +37,9 @@ SUPABASE_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJ
 model = SentenceTransformer("sentence-transformers/paraphrase-MiniLM-L3-v2")
 
 # 🔍 Ambil FAQ berdasarkan UID
-def get_faq_from_supabase(uid):
-    url = f"{SUPABASE_URL}/rest/v1/faq_texts?uid=eq.{uid}"
+def get_faq_from_supabase(admin_id):
+    """Ambil FAQ khusus untuk admin tertentu"""
+    url = f"{SUPABASE_URL}/rest/v1/faq_items?admin_id=eq.{admin_id}"
     headers = {
         "apikey": SUPABASE_KEY,
         "Authorization": f"Bearer {SUPABASE_KEY}",
@@ -47,10 +48,9 @@ def get_faq_from_supabase(uid):
     try:
         r = requests.get(url, headers=headers)
         r.raise_for_status()
-        data = r.json()
-        return [{"q": d["question"], "a": d["answer"]} for d in data]
+        return r.json()
     except Exception as e:
-        print("❌ Supabase error:", e)
+        print(f"❌ Error mengambil FAQ untuk admin {admin_id}:", e)
         return []
 
 # 🔮 Endpoint prediksi jawaban dari pertanyaan user
@@ -58,26 +58,94 @@ def get_faq_from_supabase(uid):
 @limiter.limit("5/minute")
 async def predict(request: Request):
     body = await request.json()
-    uid, question = body.get("data", [None, None])
+    admin_id, question = body.get("data", [None, None])  # admin_id dari UID di link
 
-    if not uid or not question:
-        return {"data": ["UID atau pertanyaan tidak valid."]}
+    if not admin_id or not question:
+        return {"data": ["Admin ID atau pertanyaan tidak valid."]}
 
-    faqs = get_faq_from_supabase(uid)
+    # Ambil FAQ khusus admin ini saja
+    faqs = get_faq_from_supabase(admin_id)
     if not faqs:
-        return {"data": ["FAQ tidak ditemukan untuk UID ini."]}
+        return {"data": ["Maaf, belum ada FAQ yang tersedia."]}
 
-    questions = [f["q"] for f in faqs]
-    answers = [f["a"] for f in faqs]
+    # Proses pencarian jawaban
+    questions = [f["question"] for f in faqs]
+    answers = [f["answer"] for f in faqs]
 
-    embeddings = model.encode(questions, convert_to_tensor=True)
-    query_embedding = model.encode(question, convert_to_tensor=True)
+    try:
+        embeddings = model.encode(questions, convert_to_tensor=True)
+        query_embedding = model.encode(question, convert_to_tensor=True)
+        similarity = util.pytorch_cos_sim(query_embedding, embeddings)
+        best_idx = torch.argmax(similarity).item()
+        
+        # Threshold similarity (minimal 0.5)
+        if similarity[0][best_idx] < 0.5:
+            return {"data": ["Maaf, saya tidak mengerti pertanyaan Anda."]}
+            
+        return {"data": [answers[best_idx]]}
+    except Exception as e:
+        print("❌ Error processing question:", e)
+        return {"data": ["Terjadi kesalahan saat memproses pertanyaan."]}
+
+# Save chat
+@app.post("/save_chat")
+async def save_chat(request: Request):
+    body = await request.json()
+    admin_id = body.get("admin_id")
+    session_id = body.get("session_id")
+    is_bot = body.get("is_bot", False)
+    is_admin = body.get("is_admin", False)
+    message = body.get("message")
+
+    if not all([admin_id, session_id, message]):
+        return JSONResponse({"error": "Data tidak lengkap"}, status_code=400)
 
-    similarity = util.pytorch_cos_sim(query_embedding, embeddings)
-    best_idx = torch.argmax(similarity).item()
+    url = f"{SUPABASE_URL}/rest/v1/chat_logs"
+    headers = {
+        "apikey": SUPABASE_KEY,
+        "Authorization": f"Bearer {SUPABASE_KEY}",
+        "Content-Type": "application/json",
+        "Prefer": "return=representation"
+    }
+    payload = {
+        "admin_id": admin_id,
+        "session_id": session_id,
+        "is_bot": is_bot,
+        "is_admin": is_admin,
+        "message": message
+    }
 
-    return {"data": [answers[best_idx]]}
+    try:
+        r = requests.post(url, json=payload, headers=headers)
+        r.raise_for_status()
+        return {"message": "Pesan berhasil disimpan", "id": r.json()[0]["id"]}
+    except Exception as e:
+        print("❌ Gagal menyimpan pesan:", e)
+        return JSONResponse({"error": "Gagal menyimpan pesan"}, status_code=500)
+
+# Chat history
+@app.get("/chat_history")
+async def get_chat_history(request: Request, admin_id: str, session_id: str):
+    """Ambil history chat spesifik untuk sesi ini"""
+    url = f"{SUPABASE_URL}/rest/v1/chat_logs"
+    params = {
+        "admin_id": f"eq.{admin_id}",
+        "or": f"(session_id.eq.{session_id},is_bot.eq.true)",
+        "order": "created_at.asc"
+    }
+    headers = {
+        "apikey": SUPABASE_KEY,
+        "Authorization": f"Bearer {SUPABASE_KEY}"
+    }
 
+    try:
+        r = requests.get(url, params=params, headers=headers)
+        r.raise_for_status()
+        return r.json()
+    except Exception as e:
+        print(f"❌ Error mengambil history untuk {session_id}:", e)
+        return JSONResponse({"error": "Gagal mengambil history"}, status_code=500)
+        
 # 🧹 Hapus satu pesan berdasarkan ID
 @app.post("/delete_chat")
 async def delete_chat(request: Request):