Update app.py

app.py

@@ -3,175 +3,121 @@ import tempfile
 import gc
 import logging
 import streamlit as st
-from groq import Groq, APIError
+from groq import Groq
 from langchain_community.document_loaders import PyPDFLoader
 from langchain_text_splitters import RecursiveCharacterTextSplitter
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain_community.vectorstores import Chroma
 import torch
 
-# ---------------- CONFIGURATION ----------------
+# ---------------- CONFIG ----------------
 logging.basicConfig(level=logging.INFO)
 
-# Load API key from Hugging Face secrets
 GROQ_API_KEY = st.secrets.get("GROQ_API_KEY", os.environ.get("GROQ_API_KEY"))
 GROQ_MODEL = "llama-3.1-8b-instant"
 
-# Initialize Groq client
 client = None
 if GROQ_API_KEY:
     try:
         client = Groq(api_key=GROQ_API_KEY)
-        logging.info("✅ Groq client initialized successfully.")
     except Exception as e:
-        st.error(f"❌ Failed to initialize Groq client: {e}")
-        client = None
-else:
-    st.warning("⚠️ GROQ_API_KEY not found. Please add it to Hugging Face secrets.")
+        st.error(e)
 
-# ---------------- STREAMLIT UI SETUP ----------------
+# ---------------- PAGE CONFIG ----------------
 st.set_page_config(
-    page_title="PDF Assistant", 
-    page_icon="📘", 
-    layout="wide", 
-    initial_sidebar_state="expanded"
+    page_title="PDF Assistant",
+    page_icon="📘",
+    layout="wide"
 )
 
-# ---------------- CSS ----------------
+# ---------------- FORCE DARK UI (THEME INDEPENDENT) ----------------
 st.markdown("""
 <style>
-/* 0. FORCE DARK THEME (OVERRIDES LIGHT MODE) */
-[data-testid="stAppViewContainer"] {
-    background-color: #0e1117;
-    color: #fafafa;
-}
-[data-testid="stSidebar"] {
-    background-color: #262730;
-    color: #fafafa;
-}
-/* Force text colors for standard elements */
-p, h1, h2, h3, h4, h5, h6, li, span, div, label {
-    color: #fafafa !important;
-}
 
-/* 1. GLOBAL RESET & SCROLL LOCK */
-html, body {
-    overflow: hidden;
-    height: 100%;
-    margin: 0;
-    background-color: #0e1117 !important;
+/* ========= FORCE DARK THEME ========= */
+:root {
+    --primary-color: #1e3a8a;
+    --background-color: #0e1117;
+    --secondary-background-color: #161b22;
+    --text-color: #f0f2f6;
 }
 
-/* 2. HIDE DEFAULT STREAMLIT ELEMENTS & SIDEBAR TOGGLES */
-header[data-testid="stHeader"] {
-    display: none;
-}
-footer {
-    display: none;
+html, body, [data-testid="stAppViewContainer"], .main {
+    background-color: #0e1117 !important;
+    color: #f0f2f6 !important;
 }
 
-/* --- HIDE SIDEBAR TOGGLE BUTTONS --- */
-section[data-testid="stSidebar"] > div > div:first-child {
-    display: none;
-}
-[data-testid="collapsedControl"] {
+/* Hide Streamlit chrome */
+header[data-testid="stHeader"], footer {
     display: none;
 }
 
-/* 3. SIDEBAR STYLING */
+/* Sidebar */
 [data-testid="stSidebar"] {
+    background-color: #161b22 !important;
+    width: 20rem;
     position: fixed;
-    top: 0;
-    left: 0;
     height: 100vh;
-    width: 20rem;
-    overflow-y: auto !important;
-    z-index: 99999;
-}
-[data-testid="stSidebar"]::-webkit-scrollbar {
-    width: 6px;
-}
-[data-testid="stSidebar"]::-webkit-scrollbar-thumb {
-    background: #2d3748;
-    border-radius: 3px;
+    overflow-y: auto;
 }
 
-/* 4. FIXED HEADER STYLING */
+/* Fixed Header */
 .fixed-header {
     position: fixed;
     top: 0;
     left: 0;
     width: 100%;
     height: 6rem;
-    background-color: #0e1117; 
-    z-index: 99998;
+    background: #0e1117;
+    z-index: 9999;
     display: flex;
     flex-direction: column;
-    justify-content: center;
     align-items: center;
-    border-bottom: 1px solid rgba(255, 255, 255, 0.1);
+    justify-content: center;
+    border-bottom: 1px solid rgba(255,255,255,0.1);
 }
 
-/* 5. MAIN CONTENT SCROLLING */
-.main .block-container {
-    margin-top: 6rem;
-    height: calc(100vh - 6rem);
-    overflow-y: auto;
-    padding-top: 1rem;
-    padding-bottom: 5rem;
-}
-.main .block-container::-webkit-scrollbar {
-    width: 8px;
-}
-.main .block-container::-webkit-scrollbar-thumb {
-    background: #2d3748;
-    border-radius: 4px;
+.title-text {
+    font-size: 2.5rem;
+    font-weight: 800;
 }
 
-/* 6. SIDEBAR BUTTON STYLING */
-[data-testid="stSidebar"] .stButton button {
-    width: 100%;
-    border-radius: 8px;
-    font-weight: 600;
-    margin-bottom: 6px;
-    background-color: #2d3748; 
-    color: white; 
-    border: 1px solid #4a5568;
-}
-[data-testid="stSidebar"] .stButton button:hover {
-    background-color: #4a5568;
-    border-color: #cbd5e0;
-    color: white;
+.creator-text {
+    font-size: 1rem;
+    color: #9ca3af;
 }
 
-/* 7. UPLOADED FILE FIXES (FORCE DARK DROPDOWN) */
-/* This fixes the white box issue seen in your screenshot */
-[data-testid="stFileUploader"] {
-    background-color: #262730; /* Dark background matching sidebar */
-    border-radius: 10px;
-    padding: 10px;
+.creator-text a {
+    color: #4da6ff;
+    text-decoration: none;
 }
-[data-testid="stFileUploader"] section {
-    background-color: #262730 !important;
-    color: #fafafa !important;
+
+/* Main container */
+.main .block-container {
+    margin-top: 6rem;
+    padding-bottom: 5rem;
 }
-[data-testid="stFileUploader"] button {
-    color: #fafafa !important; /* Browse files button text */
-    border-color: #4a5568 !important;
+
+/* Inputs */
+input, textarea {
+    background-color: #1f2933 !important;
+    color: #f0f2f6 !important;
+    border: 1px solid #374151 !important;
 }
-[data-testid="stFileUploader"] .uploadedFile {
-    background-color: #2d3748 !important;
+
+/* Buttons */
+button {
+    background-color: #1e3a8a !important;
     color: white !important;
+    border-radius: 8px !important;
+    font-weight: 600;
 }
-/* Hiding file list details if needed, but keeping the uploader visible */
-section[data-testid="stFileUploader"] ul {
-    display: none;
-}
-section[data-testid="stFileUploader"] small {
-    display: none;
+
+button:hover {
+    background-color: #2563eb !important;
 }
 
-/* 8. CHAT BUBBLES */
+/* Chat bubbles */
 .chat-user {
     background: #2d3748;
     padding: 12px;
@@ -179,52 +125,36 @@ section[data-testid="stFileUploader"] small {
     margin: 6px 0 6px auto;
     max-width: 85%;
     text-align: right;
-    color: #f0f2f6;
 }
+
 .chat-bot {
     background: #1e3a8a;
     padding: 12px;
     border-radius: 10px 10px 10px 2px;
     margin: 6px auto 6px 0;
     max-width: 85%;
-    text-align: left;
-    color: #ffffff;
-}
-.sources {
-    display: none; 
 }
 
-/* 9. TITLE TEXT */
-.title-text {
-    font-size: 2.5rem;
-    font-weight: 800;
-    margin: 0;
-    line-height: 1.2;
-    color: #ffffff !important; 
-}
-.creator-text {
-    font-size: 1rem;
-    font-weight: 500;
-    color: #cccccc !important;
-}
-.creator-text a {
-    color: #4da6ff;
-    text-decoration: none;
+/* Hide file list */
+[data-testid="stFileUploaderFile"],
+section[data-testid="stFileUploader"] ul,
+section[data-testid="stFileUploader"] small {
+    display: none;
 }
 
-/* 10. INPUT FORM STYLING */
-[data-testid="stForm"] {
-    border: none;
-    padding: 0;
+/* Scrollbars */
+*::-webkit-scrollbar {
+    width: 8px;
 }
-[data-testid="stTextInput"] input {
-    color: white;
-    background-color: #2d3748;
+*::-webkit-scrollbar-thumb {
+    background: #2d3748;
+    border-radius: 4px;
 }
+
 </style>
 """, unsafe_allow_html=True)
 
-# ---------------- FIXED HEADER ----------------
+# ---------------- HEADER ----------------
 st.markdown("""
 <div class="fixed-header">
     <div class="title-text">📘 PDF Assistant</div>
@@ -237,17 +167,21 @@ st.markdown("""
 # ---------------- SESSION STATE ----------------
 if "chat" not in st.session_state:
     st.session_state.chat = []
+
 if "vectorstore" not in st.session_state:
     st.session_state.vectorstore = None
+
 if "retriever" not in st.session_state:
     st.session_state.retriever = None
+
 if "uploaded_file_name" not in st.session_state:
     st.session_state.uploaded_file_name = None
+
 if "uploader_key" not in st.session_state:
     st.session_state.uploader_key = 0
 
 # ---------------- FUNCTIONS ----------------
-def clear_chat_history():
+def clear_chat():
     st.session_state.chat = []
 
 def clear_memory():
@@ -259,133 +193,100 @@ def clear_memory():
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
 
-def process_pdf(uploaded_file):
-    try:
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
-            tmp.write(uploaded_file.getvalue())
-            path = tmp.name
-        
-        loader = PyPDFLoader(path)
-        docs = loader.load()
-        
-        splitter = RecursiveCharacterTextSplitter(chunk_size=800, chunk_overlap=60)
-        chunks = splitter.split_documents(docs)
-        
-        embeddings = HuggingFaceEmbeddings(
-            model_name="sentence-transformers/all-MiniLM-L6-v2",
-            model_kwargs={"device": "cpu"},
-            encode_kwargs={"normalize_embeddings": True}
-        )
-        
-        vectorstore = Chroma.from_documents(chunks, embeddings)
-        retriever = vectorstore.as_retriever(search_kwargs={"k": 5})
-        
-        st.session_state.vectorstore = vectorstore
-        st.session_state.retriever = retriever
-        
-        if os.path.exists(path):
-            os.unlink(path)
-        
-        return len(chunks)
-    except Exception as e:
-        st.error(f"Error processing PDF: {str(e)}")
-        return None
-
-def ask_question(question):
-    if not client:
-        return None, 0, "Groq client is not initialized."
-    if not st.session_state.retriever:
-        return None, 0, "Upload PDF first."
-    
-    try:
-        docs = st.session_state.retriever.invoke(question)
-        context = "\n\n".join(d.page_content for d in docs)
-        
-        prompt = f"""You are a strict RAG Q&A assistant.
-Summarize the context based on user question and return best answer. 
-If the answer is not found, reply: "I cannot find this in the PDF." 
-
-CONTEXT = {context}
-QUESTION = {question}
-
-Answer on your behalf, don't say based on the context..."""
-        
-        response = client.chat.completions.create(
-            model=GROQ_MODEL,
-            messages=[
-                {"role": "system", "content": "Use only the PDF content."},
-                {"role": "user", "content": prompt}
-            ],
-            temperature=0.1
-        )
-        return response.choices[0].message.content.strip(), len(docs), None
-    except Exception as e:
-        return None, 0, f"Error: {str(e)}"
+def process_pdf(uploaded):
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
+        tmp.write(uploaded.getvalue())
+        path = tmp.name
+
+    loader = PyPDFLoader(path)
+    docs = loader.load()
+
+    splitter = RecursiveCharacterTextSplitter(chunk_size=800, chunk_overlap=60)
+    chunks = splitter.split_documents(docs)
+
+    embeddings = HuggingFaceEmbeddings(
+        model_name="sentence-transformers/all-MiniLM-L6-v2",
+        model_kwargs={"device": "cpu"},
+        encode_kwargs={"normalize_embeddings": True}
+    )
+
+    vectorstore = Chroma.from_documents(chunks, embeddings)
+    st.session_state.retriever = vectorstore.as_retriever(search_kwargs={"k": 5})
+    os.unlink(path)
+
+def ask(question):
+    docs = st.session_state.retriever.invoke(question)
+    context = "\n\n".join(d.page_content for d in docs)
+
+    prompt = f"""
+You are a strict RAG assistant.
+Use ONLY the PDF content.
+If not found, say: I cannot find this in the PDF.
+
+CONTEXT:
+{context}
+
+QUESTION:
+{question}
+"""
+
+    res = client.chat.completions.create(
+        model=GROQ_MODEL,
+        messages=[{"role": "user", "content": prompt}],
+        temperature=0.1
+    )
+    return res.choices[0].message.content.strip()
 
 # ---------------- SIDEBAR ----------------
 with st.sidebar:
-    st.write("") 
     if st.button("🗑️ Clear Chat History", use_container_width=True):
-        clear_chat_history()
-    
-    if st.button("🔥 Clear PDF Memory", on_click=clear_memory, use_container_width=True): 
-        st.success("Memory Cleared!")
-    
+        clear_chat()
+    if st.button("🔥 Clear PDF Memory", use_container_width=True):
+        clear_memory()
+        st.success("Memory cleared")
+
     st.markdown("---")
-    
-    upload_label = "✅ PDF Uploaded!" if st.session_state.uploaded_file_name else "Upload PDF"
+
     uploaded = st.file_uploader(
-        upload_label, type=["pdf"], key=st.session_state.uploader_key, label_visibility="collapsed"
+        "Upload PDF", type=["pdf"],
+        key=st.session_state.uploader_key,
+        label_visibility="collapsed"
     )
-    
-    if uploaded:
-        if uploaded.name != st.session_state.uploaded_file_name:
-            st.session_state.uploaded_file_name = None
+
+    if uploaded and uploaded.name != st.session_state.uploaded_file_name:
+        with st.spinner("Processing PDF..."):
+            process_pdf(uploaded)
+            st.session_state.uploaded_file_name = uploaded.name
             st.session_state.chat = []
-            with st.spinner(f"Processing '{uploaded.name}'..."):
-                chunks = process_pdf(uploaded)
-                if chunks:
-                    st.session_state.uploaded_file_name = uploaded.name
-                    st.success("✅ PDF Processed!") 
-                else:
-                    st.error("❌ Failed.")
-        else:
-            st.success(f"✅ **Active:** `{uploaded.name}`")
-    else:
-        st.warning("⬆️ Upload a PDF to start chatting!")
+            st.success("PDF ready")
 
-# ---------------- INPUT AREA ----------------
-disabled_input = st.session_state.uploaded_file_name is None or client is None
+    if not uploaded:
+        st.warning("⬆️ Upload a PDF to start chatting")
 
-# Input Form
-with st.form(key='chat_form', clear_on_submit=True):
-    col_input, col_btn = st.columns([0.85, 0.15], gap="small")
-    with col_input:
-        user_question = st.text_input(
-            "Ask a question", 
+# ---------------- INPUT ----------------
+disabled = not st.session_state.uploaded_file_name or not client
+
+with st.form("chat", clear_on_submit=True):
+    col1, col2 = st.columns([0.85, 0.15])
+    with col1:
+        q = st.text_input(
+            "",
             placeholder="Ask a question about the loaded PDF...",
-            label_visibility="collapsed",
-            disabled=disabled_input
+            disabled=disabled
         )
-    with col_btn:
-        submit_btn = st.form_submit_button("➤", disabled=disabled_input, use_container_width=True)
+    with col2:
+        send = st.form_submit_button("➤", disabled=disabled)
 
-if submit_btn and user_question:
-    st.session_state.chat.append(("user", user_question))
+if send and q:
+    st.session_state.chat.append(("user", q))
     with st.spinner("Thinking..."):
-        answer, sources, error = ask_question(user_question)
-        if answer:
-            bot_msg = answer
-            st.session_state.chat.append(("bot", bot_msg))
-        else:
-            st.session_state.chat.append(("bot", f"🔴 **Error:** {error}"))
+        ans = ask(q)
+        st.session_state.chat.append(("bot", ans))
     st.rerun()
 
-# ---------------- CHAT HISTORY (REVERSED) ----------------
-if st.session_state.chat:
-    st.markdown("---")
-    for role, msg in reversed(st.session_state.chat):
-        if role == "user":
-            st.markdown(f"<div class='chat-user'>{msg}</div>", unsafe_allow_html=True)
-        else:
-            st.markdown(f"<div class='chat-bot'>{msg}</div>", unsafe_allow_html=True)
+# ---------------- CHAT ----------------
+for role, msg in reversed(st.session_state.chat):
+    if role == "user":
+        st.markdown(f"<div class='chat-user'>{msg}</div>", unsafe_allow_html=True)
+    else:
+        st.markdown(f"<div class='chat-bot'>{msg}</div>", unsafe_allow_html=True)