Update app.py

app.py

@@ -1,12 +1,61 @@
 import streamlit as st
 import os
-import json
+import time
 from utils.ingestion import DocumentProcessor
 from utils.llm import LLMProcessor
 from utils.qa import QAEngine
 
-st.set_page_config(page_title="AI-Powered Document QA", layout="wide")
-st.title("📄 AI-Powered Document QA")
+# Set up Streamlit page with modern UI
+st.set_page_config(page_title="AI-Powered Document Chat", layout="wide")
+
+# CSS for better UI styling
+st.markdown(
+    """
+    <style>
+    .chat-container {
+        max-width: 800px;
+        margin: auto;
+    }
+    .chat-bubble {
+        padding: 12px;
+        border-radius: 12px;
+        margin-bottom: 10px;
+        max-width: 75%;
+    }
+    .user-bubble {
+        background-color: #0078ff;
+        color: white;
+        text-align: right;
+        margin-left: auto;
+    }
+    .ai-bubble {
+        background-color: #f1f1f1;
+        color: black;
+    }
+    .chat-input-container {
+        position: fixed;
+        bottom: 10px;
+        width: 100%;
+        max-width: 800px;
+        background: white;
+        padding: 10px;
+        box-shadow: 0 -2px 10px rgba(0,0,0,0.1);
+        border-radius: 8px;
+    }
+    .chat-input {
+        width: 100%;
+        padding: 10px;
+        border-radius: 8px;
+        border: 1px solid #ccc;
+    }
+    .spinner {
+        text-align: center;
+        padding: 20px;
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
 
 # Initialize processors
 document_processor = DocumentProcessor()
@@ -16,7 +65,8 @@ qa_engine = QAEngine()
 # Ensure temp directory exists
 os.makedirs("temp", exist_ok=True)
 
-st.sidebar.header("Upload a PDF")
+# Sidebar - File Upload
+st.sidebar.header("📂 Upload a PDF")
 uploaded_file = st.sidebar.file_uploader("Choose a PDF file", type=["pdf"])
 
 if uploaded_file:
@@ -27,23 +77,45 @@ if uploaded_file:
 
     st.sidebar.success("✅ File uploaded successfully!")
 
-    with st.spinner("🔄 Processing document..."):
+    with st.spinner(""):
         document_processor.process_document(pdf_path)
 
     st.sidebar.success("✅ Document processed successfully!")
 
-    # Query input
-    question = st.text_input("Ask a question from the document:", placeholder="What are the key insights?")
+# Initialize chat history in session state
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
 
-    if st.button("🔍 Search & Answer"):
-        if question:
-            with st.spinner("🧠 Searching for relevant context..."):
-                answer = qa_engine.query(question)
+# Chat UI container
+st.title("💬 AI-Powered Document Chat")
+chat_container = st.container()
 
-            st.subheader("📝 Answer:")
-            st.write(answer)
+with chat_container:
+    st.markdown('<div class="chat-container">', unsafe_allow_html=True)
+    for role, text in st.session_state.chat_history:
+        if role == "user":
+            st.markdown(f'<div class="chat-bubble user-bubble">{text}</div>', unsafe_allow_html=True)
         else:
-            st.warning("⚠️ Please enter a question.")
+            st.markdown(f'<div class="chat-bubble ai-bubble">{text}</div>', unsafe_allow_html=True)
+    st.markdown('</div>', unsafe_allow_html=True)
+
+# User Input at the bottom with fixed position
+question = st.text_input(
+    "Ask a question:",
+    placeholder="Type your question and press Enter...",
+    key="user_input"
+)
+
+if question:
+    # Append user question to chat history
+    st.session_state.chat_history.append(("user", question))
+
+    with st.spinner('<div class="spinner">...</div>'):
+        time.sleep(0.5)  # Simulating delay
+        answer = qa_engine.query(question)
+
+    # Append AI answer to chat history
+    st.session_state.chat_history.append(("ai", answer))
 
-st.markdown("---")
-st.caption("🤖 Powered by ChromaDB + Groq LLM | Built with ❤️ using Streamlit")
+    # Rerun the app to update chat history
+    st.rerun()