Update app.py

app.py

@@ -1,11 +1,12 @@
 import streamlit as st
 import os
+import json
 from utils.ingestion import DocumentProcessor
 from utils.llm import LLMProcessor
 from utils.qa import QAEngine
 
-# Set up Streamlit page
-st.set_page_config(page_title="AI-Powered Document Chat", layout="wide")
+st.set_page_config(page_title="AI-Powered Document QA", layout="wide")
+st.title("📄 AI-Powered Document QA")
 
 # Initialize processors
 document_processor = DocumentProcessor()
@@ -15,7 +16,6 @@ qa_engine = QAEngine()
 # Ensure temp directory exists
 os.makedirs("temp", exist_ok=True)
 
-# Sidebar - File Upload
 st.sidebar.header("Upload a PDF")
 uploaded_file = st.sidebar.file_uploader("Choose a PDF file", type=["pdf"])
 
@@ -27,40 +27,23 @@ if uploaded_file:
 
     st.sidebar.success("✅ File uploaded successfully!")
 
-    with st.spinner(""):
+    with st.spinner("🔄 Processing document..."):
         document_processor.process_document(pdf_path)
 
     st.sidebar.success("✅ Document processed successfully!")
 
-# Initialize chat history in session state
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []
+    # Query input
+    question = st.text_input("Ask a question from the document:", placeholder="What are the key insights?")
 
-# Display chat history
-st.title("💬 AI-Powered Document Chat")
+    if st.button("🔍 Search & Answer"):
+        if question:
+            with st.spinner("🧠 Searching for relevant context..."):
+                answer = qa_engine.query(question)
 
-chat_container = st.container()
-
-with chat_container:
-    for message in st.session_state.chat_history:
-        role, text = message
-        if role == "user":
-            st.markdown(f"**🧑‍💻 You:** {text}")
+            st.subheader("📝 Answer:")
+            st.write(answer)
         else:
-            st.markdown(f"**🤖 AI:** {text}")
-
-# User Input at the bottom
-question = st.text_input("Ask a question:", placeholder="Type your question and press Enter...", key="user_input")
-
-if question:
-    # Append user question to history
-    st.session_state.chat_history.append(("user", question))
-
-    with st.spinner(""):
-        answer = qa_engine.query(question)
-
-    # Append AI answer to history
-    st.session_state.chat_history.append(("ai", answer))
+            st.warning("⚠️ Please enter a question.")
 
-    # Rerun the app to update chat history
-    st.rerun()
+st.markdown("---")
+st.caption("🤖 Powered by ChromaDB + Groq LLM | Built with ❤️ using Streamlit")