Add Gradio RAG UI

day3/gradio_rag.py

@@ -0,0 +1,44 @@
+# day3/gradio_rag.py
+import gradio as gr
+from dotenv import load_dotenv
+from rag_system import RAGPipeline
+import traceback
+import os
+
+load_dotenv()  # ensure GROQ_API_KEY is loaded for ChatGroq
+
+# Use a writable persistent dir
+rag = RAGPipeline(persist_dir="./chroma_db_space", collection_name="pdf_docs")
+
+def chat_with_pdf(pdf_path: str, question: str):
+    try:
+        if not pdf_path:
+            return "Please upload a PDF."
+        if not question or not question.strip():
+            return "Please enter a question."
+
+        # Index the uploaded PDF (path is a string because of type='filepath')
+        rag.index_document(pdf_path, doc_id_prefix="upload")
+
+        # Ask
+        out = rag.query(question, k=4)
+        return out["answer"]
+    except Exception as e:
+        # Surface the exact error to the UI for debugging
+        return f"Error: {e}\n\n{traceback.format_exc()}"
+
+demo = gr.Interface(
+    fn=chat_with_pdf,
+    inputs=[
+        gr.File(label="Upload PDF", file_types=[".pdf"], type="filepath"),
+        gr.Textbox(label="Ask a question", placeholder="What does the PDF say?"),
+    ],
+    outputs=gr.Textbox(label="Answer"),
+    title="PDF RAG (Chroma + Groq)",
+    description="Upload a PDF and ask a question. Uses Chroma for retrieval and Groq LLM for answers."
+)
+
+if __name__ == "__main__":
+    # Optional: show whether the env var is visible to this process
+    print("GROQ key present:", bool(os.getenv("GROQ_API_KEY")))
+    demo.launch()

day3/requirements.txt

@@ -0,0 +1,6 @@
+gradio
+chromadb
+sentence-transformers
+langchain-groq
+pypdf
+python-dotenv