Update app.py

app.py

@@ -587,19 +587,123 @@ def chat_respond(message: str, history: list) -> str:
 
 
 # -----------------------------
-# GRADIO UI
+# KB FILE MANAGEMENT
 # -----------------------------
 
-description = f"""
-🚀 **{CONFIG['client']['name']}**
+ALLOWED_EXTENSIONS = {".txt", ".md"}
+MAX_FILE_SIZE_MB = 10
 
-Ask questions and get instant answers from the knowledge base using semantic search.
 
-**Tips for better results:**
-- Be specific in your questions
-- Use keywords related to your topic
-- Ask one question at a time
-"""
+def list_kb_files() -> str:
+    """Return a formatted markdown table of current KB files."""
+    paths = sorted(
+        glob.glob(os.path.join(KB_DIR, "*.txt")) +
+        glob.glob(os.path.join(KB_DIR, "*.md"))
+    )
+    if not paths:
+        return "_No files in knowledge base yet. Upload some documents to get started._"
+
+    rows = ["| File | Size |", "|------|------|"]
+    for p in paths:
+        size_kb = os.path.getsize(p) / 1024
+        rows.append(f"| {os.path.basename(p)} | {size_kb:.1f} KB |")
+    return "\n".join(rows)
+
+
+def upload_kb_files(files) -> Tuple[str, str]:
+    """
+    Accept uploaded files, validate them, save to KB_DIR, and rebuild the index.
+    Returns (status_message, updated_file_list).
+    """
+    if not files:
+        return "⚠️ No files selected.", list_kb_files()
+
+    os.makedirs(KB_DIR, exist_ok=True)
+
+    saved = []
+    skipped = []
+
+    for file in files:
+        # Gradio passes either a filepath string or a NamedString object
+        filepath = file if isinstance(file, str) else file.name
+        original_name = os.path.basename(filepath)
+        ext = os.path.splitext(original_name)[1].lower()
+
+        if ext not in ALLOWED_EXTENSIONS:
+            skipped.append(f"{original_name} (unsupported type — use .txt or .md)")
+            continue
+
+        size_mb = os.path.getsize(filepath) / (1024 * 1024)
+        if size_mb > MAX_FILE_SIZE_MB:
+            skipped.append(f"{original_name} (too large — max {MAX_FILE_SIZE_MB} MB)")
+            continue
+
+        dest = os.path.join(KB_DIR, original_name)
+        try:
+            with open(filepath, "r", encoding="utf-8", errors="replace") as f:
+                content = f.read()
+            if not content.strip():
+                skipped.append(f"{original_name} (empty file)")
+                continue
+            with open(dest, "w", encoding="utf-8") as f:
+                f.write(content)
+            saved.append(original_name)
+            logger.info(f"KB file saved: {dest}")
+        except Exception as e:
+            skipped.append(f"{original_name} (error: {e})")
+            logger.error(f"Failed to save {original_name}: {e}")
+
+    if saved:
+        logger.info("Rebuilding KB index after upload...")
+        try:
+            kb_index.reload()
+            reload_msg = "✅ Index rebuilt successfully."
+        except Exception as e:
+            reload_msg = f"⚠️ Index rebuild failed: {e}"
+            logger.error(reload_msg)
+    else:
+        reload_msg = ""
+
+    status_parts = []
+    if saved:
+        status_parts.append(f"✅ **Uploaded:** {', '.join(saved)}")
+    if skipped:
+        status_parts.append(f"⚠️ **Skipped:** {', '.join(skipped)}")
+    if reload_msg:
+        status_parts.append(reload_msg)
+
+    return "\n\n".join(status_parts), list_kb_files()
+
+
+def delete_kb_file(filename: str) -> Tuple[str, str]:
+    """Delete a file from the KB directory and rebuild the index."""
+    filename = filename.strip()
+    if not filename:
+        return "⚠️ No filename provided.", list_kb_files()
+
+    # Safety: strip any path components — only allow plain filenames
+    filename = os.path.basename(filename)
+    ext = os.path.splitext(filename)[1].lower()
+    if ext not in ALLOWED_EXTENSIONS:
+        return f"⚠️ Invalid file type: {filename}", list_kb_files()
+
+    target = os.path.join(KB_DIR, filename)
+    if not os.path.exists(target):
+        return f"⚠️ File not found: {filename}", list_kb_files()
+
+    try:
+        os.remove(target)
+        logger.info(f"Deleted KB file: {target}")
+        kb_index.reload()
+        return f"🗑️ **Deleted** `{filename}` and rebuilt index.", list_kb_files()
+    except Exception as e:
+        logger.error(f"Failed to delete {filename}: {e}")
+        return f"❌ Could not delete `{filename}`: {e}", list_kb_files()
+
+
+# -----------------------------
+# GRADIO UI
+# -----------------------------
 
 examples = [
     q["query"]
@@ -611,13 +715,70 @@ examples = [
     "Why is self-service important for customer support?",
 ]
 
-chat_interface = gr.ChatInterface(
-    fn=chat_respond,
-    title=f"🤖 {CONFIG['client']['name']}",
-    description=description,
-    examples=examples,
-    cache_examples=False,
-)
+with gr.Blocks(title=f"🤖 {CONFIG['client']['name']}") as demo:
+    gr.Markdown(f"# 🤖 {CONFIG['client']['name']}")
+
+    with gr.Tabs():
+
+        # ── Tab 1: Chat ──────────────────────────────────────────────
+        with gr.TabItem("💬 Chat"):
+            gr.Markdown("""
+Ask questions and get instant answers from the knowledge base using semantic search.
+
+**Tips:** Be specific · Use keywords · Ask one question at a time
+""")
+            gr.ChatInterface(
+                fn=chat_respond,
+                examples=examples,
+                cache_examples=False,
+            )
+
+        # ── Tab 2: KB Manager ────────────────────────────────────────
+        with gr.TabItem("📁 KB Manager"):
+            gr.Markdown("""
+### Manage Knowledge Base Documents
+
+Upload `.txt` or `.md` files to expand what the assistant can answer.
+The index is rebuilt automatically after every upload or deletion.
+""")
+
+            with gr.Row():
+                with gr.Column(scale=2):
+                    upload_input = gr.File(
+                        label="Upload documents (.txt or .md, max 10 MB each)",
+                        file_count="multiple",
+                        file_types=[".txt", ".md"],
+                    )
+                    upload_btn = gr.Button("⬆️ Upload & Index", variant="primary")
+                    upload_status = gr.Markdown(label="Upload Status")
+
+                with gr.Column(scale=1):
+                    gr.Markdown("#### Current KB Files")
+                    kb_file_list = gr.Markdown(value=list_kb_files)
+
+                    gr.Markdown("#### Delete a File")
+                    delete_input = gr.Textbox(
+                        placeholder="exact filename e.g. my_doc.md",
+                        label="Filename to delete",
+                        show_label=False,
+                    )
+                    delete_btn = gr.Button("🗑️ Delete", variant="stop")
+                    delete_status = gr.Markdown(label="Delete Status")
+
+            # Wire up upload
+            upload_btn.click(
+                fn=upload_kb_files,
+                inputs=[upload_input],
+                outputs=[upload_status, kb_file_list],
+            )
+
+            # Wire up delete
+            delete_btn.click(
+                fn=delete_kb_file,
+                inputs=[delete_input],
+                outputs=[delete_status, kb_file_list],
+            )
+
 
 # -----------------------------
 # LAUNCH
@@ -630,7 +791,7 @@ if __name__ == "__main__":
     if is_huggingface or is_container:
         env_label = "🤗 HuggingFace Spaces" if is_huggingface else "🐳 container"
         logger.info(f"Launching on {env_label}...")
-        chat_interface.launch(server_name="0.0.0.0", server_port=7860)
+        demo.launch(server_name="0.0.0.0", server_port=7860)
     else:
         logger.info("💻 Launching locally...")
-        chat_interface.launch(share=False)
+        demo.launch(share=False)