diff --git a/README.md b/README.md index 703779b..9c02a24 100644 --- a/README.md +++ b/README.md @@ -266,9 +266,9 @@ documind/ --- -## Screenshots +## Demo -> *Screenshots coming soon — upload a PDF, ask questions, see sources cited inline.* +https://github.com/user-attachments/assets/290e7caf-6676-43c2-9f9f-9df63d28c3f9 --- diff --git a/rag/agents/graph.py b/rag/agents/graph.py index 725c648..ab451dd 100644 --- a/rag/agents/graph.py +++ b/rag/agents/graph.py @@ -52,12 +52,13 @@ def retrieve(state: GraphState) -> GraphState: def direct_response(state: GraphState) -> GraphState: """Return a response for general conversation (no retrieval needed).""" - return { - "generation": ( - "Hello! I'm a PDF assistant. Please ask me a question about your " - "document and I'll look it up for you." - ) - } + question = state.get("question", "").strip().lower() + greetings = {"hi", "hello", "hey", "hiya", "howdy", "sup", "what's up", "whats up"} + if any(question == g or question.startswith(g + " ") for g in greetings): + reply = "Hey! Ask me anything about your document and I'll find the answer for you." + else: + reply = "I can only answer questions about the document you've uploaded. What would you like to know?" + return {"generation": reply} def fallback(state: GraphState) -> GraphState: diff --git a/requirements.txt b/requirements.txt index 2ac4715..2d67b69 100644 --- a/requirements.txt +++ b/requirements.txt @@ -36,5 +36,5 @@ streamlit watchdog # Evaluation -ragas +ragas==0.0.22 datasets diff --git a/ui/components/chat.py b/ui/components/chat.py index ce4943b..c52dbda 100644 --- a/ui/components/chat.py +++ b/ui/components/chat.py @@ -23,21 +23,83 @@ def render_chat() -> None: if not st.session_state.current_doc_id: # ── Welcome screen ──────────────────────────────────────────────────── st.markdown(""" -
-

Welcome to DocuMind 🧠

-

- Upload a PDF from the sidebar to start a conversation with your document.

- This assistant uses hybrid search (semantic + keyword), - cross-encoder reranking, and Gemini 2.5 Flash to give you - precise, document-grounded answers. +

+
+
🧠
+
+
DocuMind
+
+ Agentic Document Intelligence +
+
+
+
+ Ask anything about your document.
+ Get answers you can trust. +
+

+ Upload a PDF from the sidebar. Ask a question and DocuMind will find the relevant parts, + generate an answer based only on your document, and show you exactly where it came from.

+
+ ⚡ Hybrid Search + 🎯 Cross-Encoder Reranking + ✨ Gemini 2.5 Flash + 🔗 LangGraph Agent + 🛡 Hallucination Guard +
""", unsafe_allow_html=True) @@ -92,14 +154,12 @@ def render_chat() -> None: st.session_state.chat_history.pop() return - if sources: - _render_sources(sources) - st.session_state.chat_history.append({ "role": "assistant", "content": answer, "sources": sources, }) + st.rerun() def _stream_query(question: str): diff --git a/ui/components/sidebar.py b/ui/components/sidebar.py index 54dcdc9..18e3a2e 100644 --- a/ui/components/sidebar.py +++ b/ui/components/sidebar.py @@ -9,7 +9,40 @@ def render_sidebar() -> None: - st.sidebar.markdown("## 📂 Documents") + st.sidebar.markdown(""" +
+
🧠
+
+
DocuMind
+
v1.0.0
+
+
+
+ 📂 Your Documents +
+ """, unsafe_allow_html=True) # ── Upload ──────────────────────────────────────────────────────────────── uploaded_file = st.sidebar.file_uploader( diff --git a/ui/streamlit_app.py b/ui/streamlit_app.py index 0029b85..8e3625f 100644 --- a/ui/streamlit_app.py +++ b/ui/streamlit_app.py @@ -97,8 +97,34 @@ # ── Header ───────────────────────────────────────────────────────────────────── header_left, header_right = st.columns([6, 1]) with header_left: - st.markdown("## 📄 RAG PDF Assistant") - st.caption("Ask questions about your PDF — answers are grounded strictly in your document.") + st.markdown(""" +
+
🧠
+
+
DocuMind
+
+ Intelligent answers  ·  grounded in your document +
+
+
+ """, unsafe_allow_html=True) with header_right: if st.session_state.current_doc_id: label = "Hide PDF 📕" if st.session_state.show_pdf else "View PDF 📖"