fix UI

app.py

@@ -1,6 +1,8 @@
 from fastapi import FastAPI, Request
 from rag_core.business import answer_query, rescan_index
 from ui import app_ui
+import gradio as gr
+import logging
 
 app = FastAPI()
 
@@ -14,6 +16,7 @@ async def ask_api(req: Request):
 async def rescan_api():
     return rescan_index()
 
-# Gắn Gradio UI vào FastAPI
-import gradio as gr
-app = gr.mount_gradio_app(app, app_ui, path="/")
+# Mount Gradio UI vào FastAPI tại root
+app = gr.mount_gradio_app(app, app_ui, path="")
+
+logging.info("✅ Gradio UI đã mount vào root /")

rag_core/business.py

@@ -6,55 +6,56 @@ from rag_core.retriever import Retriever
 from rag_core.llm import generate_answer
 
 retriever = Retriever()
-ready = retriever.index is not None
+
+def is_ready():
+    ready = retriever.index is not None
+    logging.info(f"✅ Kiểm tra index sẵn sàng: {ready}")
+    return ready
 
 def build_index():
-    global ready
     logging.info("🔄 Bắt đầu xây FAISS index từ đầu...")
     try:
         with open("data/raw_law.txt", "r", encoding="utf-8") as f:
             text = f.read()
-        logging.info("📄 Đọc xong file văn bản luật.")
+        logging.info(f"📄 Đọc văn bản thành công, độ dài: {len(text)} ký tự")
         chunks = chunk_legal_text(text)
-        logging.info(f"🔪 Chunking hoàn tất: {len(chunks)} đoạn.")
+        logging.info(f"✂️ Đã chunk thành {len(chunks)} đoạn")
         retriever.build(chunks, get_embedding)
-        ready = True
-        logging.info("✅ Đã xây xong FAISS index.")
+        logging.info("✅ Xây FAISS index thành công.")
     except Exception as e:
         logging.exception(f"❌ Lỗi khi xây index: {e}")
-        raise
 
 def answer_query(query):
-    logging.info("💬 Nhận yêu cầu hỏi đáp.")
-    if not ready:
-        logging.warning("⚠️ Index chưa sẵn sàng.")
-        return {"error": "Index chưa sẵn sàng. Vui lòng bấm 'Xây Index' hoặc gọi API rescan."}
+    logging.info(f"❓ Nhận câu hỏi: {query}")
+    if not is_ready():
+        logging.warning("⚠️ Index chưa sẵn sàng khi truy vấn.")
+        return {"error": "Index chưa sẵn sàng. Vui lòng bấm 'Rebuild Index'."}
     try:
-        logging.info(f"🔍 Truy vấn: {query}")
         docs = retriever.query(query, get_embedding)
-        logging.info(f"📚 Top {len(docs)} đoạn liên quan được truy xuất.")
+        logging.info(f"📚 Truy xuất được {len(docs)} đoạn liên quan")
         prompt = "\n\n".join(docs) + f"\n\nCâu hỏi: {query}\nTrả lời:"
         answer = generate_answer(prompt)
-        logging.info("✅ Đã nhận được phản hồi từ LLM.")
+        logging.info("✅ Đã tạo câu trả lời thành công.")
         return {"answer": answer}
     except Exception as e:
-        logging.exception(f"❌ Lỗi khi xử lý truy vấn: {e}")
-        return {"error": "Lỗi khi xử lý truy vấn."}
+        logging.exception(f"❌ Lỗi khi trả lời câu hỏi: {e}")
+        return {"error": str(e)}
 
 def rescan_index():
-    logging.info("🔄 Thực hiện rescan & bổ sung FAISS index.")
-    if not ready:
-        logging.warning("⚠️ Index chưa sẵn sàng.")
+    if not is_ready():
+        logging.info("⚠️ Chưa có index, gọi build_index() thay vì rescan.")
         build_index()
+        return {"status": "✅ Đã tạo mới FAISS index."}
+    logging.info("♻️ Tiến hành rescan và cập nhật các chunk mới...")
     try:
         with open("data/raw_law.txt", "r", encoding="utf-8") as f:
             text = f.read()
-        logging.info("📄 Đọc xong file văn bản luật.")
+        logging.info(f"📄 Đọc văn bản thành công, độ dài: {len(text)} ký tự")
         chunks = chunk_legal_text(text)
-        logging.info(f"🔍 Số chunk sau chunking: {len(chunks)}")
+        logging.info(f"✂️ Đã chunk thành {len(chunks)} đoạn")
         retriever.rescan_and_append(chunks, get_embedding)
-        logging.info("✅ Đã rescan & cập nhật thành công.")
-        return {"status": "Rescan & update thành công."}
+        logging.info("✅ Rescan hoàn tất.")
+        return {"status": "✅ Rescan & update thành công."}
     except Exception as e:
-        logging.exception(f"❌ Lỗi khi rescan index: {e}")
-        return {"error": "Lỗi khi rescan index."}
+        logging.exception(f"❌ Lỗi trong quá trình rescan: {e}")
+        return {"status": f"Lỗi: {str(e)}"}

ui.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import logging
-from rag_core.business import build_index, rescan_index, answer_query, retriever, ready
+from rag_core.business import build_index, rescan_index, answer_query, retriever, is_ready
 
 logging.info("🎨 Khởi tạo Gradio UI...")