Update app.py

app.py

@@ -1,14 +1,24 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
+import subprocess
+import os
+from huggingface_hub import InferenceClient, HfApi
 from rag_query import retrieve
+from embed_index import main as build_index
 
 from config.settings import *
 
+INDEX_FILES = [
+    FAISS_INDEX_PATH,
+    METADATA_PATH
+]
+
+
 def load_prompt():
     with open("prompts/rag_prompt.txt", "r", encoding="utf-8") as f:
         return f.read()
 
 
+# ---------- RAG CHAT ----------
 def respond(
     message,
     history,
@@ -26,13 +36,10 @@ def respond(
     retrieved = retrieve(message)
 
     context_blocks = []
-    sources = set()
-
     for item in retrieved:
         context_blocks.append(
             f"[{item['condition']} – {item['section']}]\n{item}"
         )
-        sources.add(item["source_id"])
 
     context = "\n\n".join(context_blocks)
 
@@ -47,7 +54,6 @@ def respond(
     ]
 
     response = ""
-
     for chunk in client.chat_completion(
         messages,
         max_tokens=max_tokens,
@@ -60,6 +66,36 @@ def respond(
             yield response
 
 
+# ---------- BUILD INDEX ----------
+def rebuild_index_ui():
+    build_index()
+    return "✅ Index rebuilt successfully."
+
+
+# ---------- COMMIT TO HF ----------
+def commit_index_ui(hf_token: gr.OAuthToken):
+    api = HfApi(token=hf_token.token)
+
+    repo_id = os.environ.get("SPACE_ID")  # HF injects this
+    if not repo_id:
+        return "❌ Could not determine Space repo ID."
+
+    for file_path in INDEX_FILES:
+        if not os.path.exists(file_path):
+            return f"❌ Missing file: {file_path}"
+
+        api.upload_file(
+            path_or_fileobj=file_path,
+            path_in_repo=file_path,
+            repo_id=repo_id,
+            repo_type="space",
+            commit_message="Update FAISS index"
+        )
+
+    return "⬆ Index committed to Hugging Face successfully."
+
+
+# ---------- UI ----------
 chatbot = gr.ChatInterface(
     respond,
     type="messages",
@@ -77,7 +113,22 @@ chatbot = gr.ChatInterface(
 with gr.Blocks() as demo:
     with gr.Sidebar():
         gr.LoginButton()
+        build_btn = gr.Button("🔨 Build Index")
+        commit_btn = gr.Button("⬆ Commit to HF")
+        status_box = gr.Markdown()
+
+        build_btn.click(
+            rebuild_index_ui,
+            outputs=status_box
+        )
+
+        commit_btn.click(
+            commit_index_ui,
+            outputs=status_box
+        )
+
     chatbot.render()
 
+
 if __name__ == "__main__":
     demo.launch()