fix UI

app.py

@@ -1,10 +1,6 @@
-import logging
 from fastapi import FastAPI, Request
-import gradio as gr
-from rag_core.business import build_index, answer_query, rescan_index
-
-logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
-logging.info("🚀 Khởi chạy FastAPI + Gradio tại cổng 7860")
+from rag_core.business import answer_query, rescan_index
+from ui import app_ui
 
 app = FastAPI()
 
@@ -18,18 +14,6 @@ async def ask_api(req: Request):
 async def rescan_api():
     return rescan_index()
 
-def gradio_ui():
-    with gr.Blocks() as ui:
-        build_btn = gr.Button("🔄 Xây Index")
-        query_box = gr.Textbox(label="Nhập câu hỏi")
-        output_box = gr.Textbox(label="Trả lời")
-
-        build_btn.click(fn=lambda: build_index() or "✅ Đã xây xong Index.", outputs=None)
-        query_box.submit(fn=lambda q: answer_query(q)["answer"], inputs=query_box, outputs=output_box)
-
-    return ui
-
-# Mount Gradio vào FastAPI
-app = gr.mount_gradio_app(app, gradio_ui(), path="/")
-
-# Run với: uvicorn app:app --host 0.0.0.0 --port 7860
+# Gắn Gradio UI vào FastAPI
+import gradio as gr
+app = gr.mount_gradio_app(app, app_ui, path="/")

rag_core/business.py

@@ -45,7 +45,7 @@ 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.")
-        return {"error": "Index chưa sẵn sàng."}
+        build_index()
     try:
         with open("data/raw_law.txt", "r", encoding="utf-8") as f:
             text = f.read()

ui.py

@@ -0,0 +1,33 @@
+import gradio as gr
+import logging
+from rag_core.business import build_index, rescan_index, answer_query, retriever, ready
+
+logging.info("🎨 Khởi tạo Gradio UI...")
+
+def handle_rebuild():
+    if retriever.index is None:
+        logging.info("⚙️ Chưa có index, gọi build_index()")
+        build_index()
+        return "✅ Đã tạo mới FAISS index."
+    else:
+        logging.info("♻️ Đã có index, gọi rescan_index()")
+        result = rescan_index()
+        return result["status"]
+
+def handle_query(query):
+    result = answer_query(query)
+    return result.get("answer", result.get("error", "❌ Lỗi không xác định."))
+
+with gr.Blocks(title="Luật Giao Thông RAG") as iface:
+    gr.Markdown("## 📚 Trợ lý hỏi đáp văn bản luật")
+
+    query_box = gr.Textbox(label="❓ Nhập câu hỏi", lines=2)
+    answer_box = gr.Textbox(label="💬 Trả lời", lines=6)
+
+    submit_btn = gr.Button("📤 Gửi câu hỏi")
+    rebuild_btn = gr.Button("🔄 Rebuild Index")
+
+    submit_btn.click(fn=handle_query, inputs=query_box, outputs=answer_box)
+    rebuild_btn.click(fn=handle_rebuild, outputs=answer_box)
+
+app_ui = iface