Update app.py

app.py

@@ -1,158 +1,71 @@
 import gradio as gr
-import os
 from rag_engine import RAGEngine
 
-# Initialize engine
 engine = RAGEngine()
 
-# ── Helpers ──────────────────────────────────────────────────────────────────
-
-def process_pdf(file):
-    if file is None:
-        return "⚠️ No file uploaded.", gr.update(interactive=False)
-    try:
-        msg = engine.load_pdf(file.name)
-        return msg, gr.update(interactive=True)
-    except Exception as e:
-        return f"❌ Error: {e}", gr.update(interactive=False)
-
-def process_url(url):
-    if not url.strip():
-        return "⚠️ Please enter a URL.", gr.update(interactive=False)
-    try:
-        msg = engine.load_url(url.strip())
-        return msg, gr.update(interactive=True)
-    except Exception as e:
-        return f"❌ Error: {e}", gr.update(interactive=False)
-
-def process_text(text):
-    if not text.strip():
-        return "⚠️ Please paste some text.", gr.update(interactive=False)
-    try:
-        msg = engine.load_text(text.strip())
-        return msg, gr.update(interactive=True)
-    except Exception as e:
-        return f"❌ Error: {e}", gr.update(interactive=False)
-
-def chat(message, history):
-    if not message.strip():
+# ── Functions ─────────────────────────────
+
+def load_pdf(file):
+    if not file:
+        return "Upload file", gr.update(interactive=False)
+    return engine.load_pdf(file.name), gr.update(interactive=True)
+
+def load_url(url):
+    if not url:
+        return "Enter URL", gr.update(interactive=False)
+    return engine.load_url(url), gr.update(interactive=True)
+
+def load_text(text):
+    if not text:
+        return "Enter text", gr.update(interactive=False)
+    return engine.load_text(text), gr.update(interactive=True)
+
+def chat(msg, history):
+    if not msg:
         return history, ""
-    answer = engine.answer(message)
+    ans = engine.answer(msg)
     history = history or []
-    history.append((message, answer))
+    history.append((msg, ans))
     return history, ""
 
-def reset_all():
+def reset():
     engine.reset()
-    return [], "", "🔄 Knowledge base cleared. Load a new source to begin."
+    return [], "", "Reset done"
 
-# ── UI ────────────────────────────────────────────────────────────────────────
+# ── UI ───────────────────────────────────
 
-DESCRIPTION = """
-# 🤖 RAG Chatbot — Chat with Your Data
-Upload a **PDF**, paste a **website URL**, or drop in **raw text** — then ask anything.
+with gr.Blocks(title="RAG Bot") as demo:
+    gr.Markdown("# 🤖 RAG Chatbot")
 
-Built with 🤗 HuggingFace · FAISS · spaCy · Sentence-Transformers
-"""
+    with gr.Row():
+        with gr.Column():
+            pdf = gr.File()
+            btn_pdf = gr.Button("Load PDF")
 
-with gr.Blocks(
-    title="RAG Chatbot",
-    theme=gr.themes.Soft(primary_hue="violet", secondary_hue="slate"),
-    css="""
-    #chatbot { min-height: 420px; }
-    .source-box { border: 1px solid #e2e8f0; border-radius: 12px; padding: 16px; background: #f8fafc; }
-    .status-bar { font-size: 13px; padding: 6px 12px; border-radius: 8px; }
-    footer { display: none !important; }
-    """,
-) as demo:
-    gr.Markdown(DESCRIPTION)
+            url = gr.Textbox()
+            btn_url = gr.Button("Load URL")
 
-    with gr.Row():
-        # ── Left column: source loader ──────────────────────────────────────
-        with gr.Column(scale=1, elem_classes="source-box"):
-            gr.Markdown("### 📂 Load a Knowledge Source")
-
-            with gr.Tab("📄 PDF"):
-                pdf_input = gr.File(
-                    label="Upload PDF",
-                    file_types=[".pdf"],
-                    type="filepath",
-                )
-                pdf_btn = gr.Button("Load PDF", variant="primary")
-
-            with gr.Tab("🌐 Website URL"):
-                url_input = gr.Textbox(
-                    label="Website URL",
-                    placeholder="https://example.com/page",
-                )
-                url_btn = gr.Button("Load URL", variant="primary")
-
-            with gr.Tab("📝 Raw Text"):
-                text_input = gr.Textbox(
-                    label="Paste your text",
-                    lines=8,
-                    placeholder="Paste any text here...",
-                )
-                text_btn = gr.Button("Load Text", variant="primary")
-
-            status = gr.Textbox(
-                label="Status",
-                value="⏳ Load a source to start chatting.",
-                interactive=False,
-                elem_classes="status-bar",
-            )
-
-            with gr.Accordion("ℹ️ NLP Insights", open=False):
-                nlp_output = gr.JSON(label="Last query analysis")
-
-        # ── Right column: chat ──────────────────────────────────────────────
-        with gr.Column(scale=2):
-            gr.Markdown("### 💬 Chat")
-            chatbot = gr.Chatbot(
-                elem_id="chatbot",
-                bubble_full_width=False,
-                avatar_images=(None, "https://huggingface.co/front/assets/huggingface_logo-noborder.svg"),
-            )
-
-            with gr.Row():
-                msg_box = gr.Textbox(
-                    placeholder="Ask a question about your loaded content...",
-                    show_label=False,
-                    scale=4,
-                    interactive=False,
-                )
-                send_btn = gr.Button("Send ➤", variant="primary", scale=1)
-
-            reset_btn = gr.Button("🗑️ Clear & Reset", variant="secondary")
-
-    # ── Example questions ────────────────────────────────────────────────────
-    gr.Examples(
-        examples=[
-            "What is the main topic of this document?",
-            "Summarise the key points in 3 bullet points.",
-            "What are the most important entities mentioned?",
-            "Give me a short conclusion based on the content.",
-        ],
-        inputs=msg_box,
-        label="💡 Example questions",
-    )
-
-    # ── Wire events ──────────────────────────────────────────────────────────
-    def chat_with_nlp(message, history):
-        if not message.strip():
-            return history, "", {}
-        answer, nlp_info = engine.answer_with_nlp(message)
-        history = history or []
-        history.append((message, answer))
-        return history, "", nlp_info
-
-    pdf_btn.click(process_pdf, pdf_input, [status, msg_box])
-    url_btn.click(process_url, url_input, [status, msg_box])
-    text_btn.click(process_text, text_input, [status, msg_box])
-
-    send_btn.click(chat_with_nlp, [msg_box, chatbot], [chatbot, msg_box, nlp_output])
-    msg_box.submit(chat_with_nlp, [msg_box, chatbot], [chatbot, msg_box, nlp_output])
-    reset_btn.click(reset_all, outputs=[chatbot, msg_box, status])
+            text = gr.Textbox(lines=5)
+            btn_text = gr.Button("Load Text")
+
+            status = gr.Textbox(value="Load data", interactive=False)
+
+        with gr.Column():
+            chatbot = gr.Chatbot()  # ✅ FIXED (no bubble_full_width)
+
+            msg = gr.Textbox(placeholder="Ask something...", interactive=False)
+            send = gr.Button("Send")
+
+            reset_btn = gr.Button("Reset")
+
+    btn_pdf.click(load_pdf, pdf, [status, msg])
+    btn_url.click(load_url, url, [status, msg])
+    btn_text.click(load_text, text, [status, msg])
+
+    send.click(chat, [msg, chatbot], [chatbot, msg])
+    msg.submit(chat, [msg, chatbot], [chatbot, msg])
+
+    reset_btn.click(reset, outputs=[chatbot, msg, status])
 
 if __name__ == "__main__":
-    demo.launch(share=False)
+    demo.launch()