Update app.py

app.py

@@ -16,12 +16,7 @@ logging.basicConfig(level=logging.INFO)
 GROQ_API_KEY = st.secrets.get("GROQ_API_KEY", os.environ.get("GROQ_API_KEY"))
 GROQ_MODEL = "llama-3.1-8b-instant"
 
-client = None
-if GROQ_API_KEY:
-    try:
-        client = Groq(api_key=GROQ_API_KEY)
-    except Exception as e:
-        st.error(e)
+client = Groq(api_key=GROQ_API_KEY) if GROQ_API_KEY else None
 
 # ---------------- PAGE CONFIG ----------------
 st.set_page_config(
@@ -30,124 +25,120 @@ st.set_page_config(
     layout="wide"
 )
 
-# ---------------- FORCE DARK UI (THEME INDEPENDENT) ----------------
+# ---------------- FORCE DARK UI ----------------
 st.markdown("""
 <style>
+/* ================= FORCE DARK THEME ================= */
 :root {
-    --primary-color: #1e3a8a;
-    --background-color: #0e1117;
-    --secondary-background-color: #161b22;
-    --text-color: #f0f2f6;
+    --bg: #0e1117;
+    --sidebar: #161b22;
+    --input: #1f2933;
+    --border: #374151;
+    --text: #f0f2f6;
+    --primary: #1e3a8a;
 }
 
 html, body, [data-testid="stAppViewContainer"], .main {
-    background-color: #0e1117 !important;
-    color: #f0f2f6 !important;
+    background-color: var(--bg) !important;
+    color: var(--text) !important;
 }
 
-/* Hide Streamlit default UI */
-header[data-testid="stHeader"], footer {
-    display: none;
-}
+/* Hide Streamlit chrome */
+header[data-testid="stHeader"], footer { display: none; }
 
 /* Sidebar */
 [data-testid="stSidebar"] {
-    background-color: #161b22 !important;
+    background-color: var(--sidebar) !important;
     width: 20rem;
-    position: fixed;
-    height: 100vh;
-    overflow-y: auto;
 }
 
-/* Fixed header */
+/* FIX: File uploader white background */
+section[data-testid="stFileUploader"],
+section[data-testid="stFileUploader"] > div,
+section[data-testid="stFileUploader"] label {
+    background-color: var(--sidebar) !important;
+    color: var(--text) !important;
+    border-radius: 12px;
+}
+
+/* Fixed Header */
 .fixed-header {
     position: fixed;
     top: 0;
     left: 0;
-    width: 100%;
     height: 6rem;
-    background: #0e1117;
-    z-index: 9999;
+    width: 100%;
+    background: var(--bg);
+    border-bottom: 1px solid rgba(255,255,255,0.08);
     display: flex;
     flex-direction: column;
     align-items: center;
     justify-content: center;
-    border-bottom: 1px solid rgba(255,255,255,0.1);
-}
-
-.title-text {
-    font-size: 2.5rem;
-    font-weight: 800;
-}
-
-.creator-text {
-    font-size: 1rem;
-    color: #9ca3af;
+    z-index: 9999;
 }
 
-.creator-text a {
-    color: #4da6ff;
-    text-decoration: none;
-}
+.title-text { font-size: 2.5rem; font-weight: 800; }
+.creator-text { color: #9ca3af; }
 
-/* Main content spacing */
+/* Main spacing */
 .main .block-container {
     margin-top: 6rem;
-    padding-bottom: 5rem;
+    padding-bottom: 6rem;
 }
 
-/* Inputs */
+/* ================= INPUT + BUTTON ALIGNMENT ================= */
 input, textarea {
-    background-color: #1f2933 !important;
-    color: #f0f2f6 !important;
-    border: 1px solid #374151 !important;
+    background-color: var(--input) !important;
+    color: var(--text) !important;
+    border: 1px solid var(--border) !important;
+    height: 3.2rem !important;
+    border-radius: 10px !important;
+}
+
+/* Placeholder FIX (same in light & dark) */
+input::placeholder {
+    color: #9ca3af !important;
+    opacity: 1 !important;
 }
 
-/* Buttons */
+/* Button (Arrow) */
 button {
-    background-color: #1e3a8a !important;
+    height: 3.2rem !important;
+    background-color: var(--primary) !important;
     color: white !important;
-    border-radius: 8px !important;
-    font-weight: 600;
+    border-radius: 10px !important;
+    font-weight: 700 !important;
 }
 
-button:hover {
-    background-color: #2563eb !important;
-}
+button:hover { background-color: #2563eb !important; }
 
 /* Chat bubbles */
 .chat-user {
     background: #2d3748;
-    padding: 12px;
-    border-radius: 10px 10px 2px 10px;
+    padding: 14px;
+    border-radius: 12px 12px 2px 12px;
     margin: 6px 0 6px auto;
     max-width: 85%;
     text-align: right;
 }
 
 .chat-bot {
-    background: #1e3a8a;
-    padding: 12px;
-    border-radius: 10px 10px 10px 2px;
+    background: var(--primary);
+    padding: 14px;
+    border-radius: 12px 12px 12px 2px;
     margin: 6px auto 6px 0;
     max-width: 85%;
 }
 
-/* Hide file list */
-[data-testid="stFileUploaderFile"],
-section[data-testid="stFileUploader"] ul,
-section[data-testid="stFileUploader"] small {
-    display: none;
-}
-
-/* Scrollbar */
-*::-webkit-scrollbar {
-    width: 8px;
-}
+/* Scrollbars */
+*::-webkit-scrollbar { width: 8px; }
 *::-webkit-scrollbar-thumb {
     background: #2d3748;
     border-radius: 4px;
 }
+
+/* Hide uploaded filename */
+[data-testid="stFileUploaderFile"] { display: none; }
 </style>
 """, unsafe_allow_html=True)
 
@@ -156,7 +147,7 @@ st.markdown("""
 <div class="fixed-header">
     <div class="title-text">📘 PDF Assistant</div>
     <div class="creator-text">
-        by <a href="https://www.linkedin.com/in/abhishek-iitr/" target="_blank">
+        by <a href="https://www.linkedin.com/in/abhishek-iitr/" target="_blank" style="color:#4da6ff">
         Abhishek Saxena</a>
     </div>
 </div>
@@ -192,53 +183,38 @@ def process_pdf(uploaded):
         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)
+    docs = PyPDFLoader(path).load()
+    chunks = RecursiveCharacterTextSplitter(
+        chunk_size=800, chunk_overlap=60
+    ).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})
+    vs = Chroma.from_documents(chunks, embeddings)
+    st.session_state.retriever = vs.as_retriever(search_kwargs={"k": 5})
     os.unlink(path)
 
-def ask(question):
-    docs = st.session_state.retriever.invoke(question)
+def ask(q):
+    docs = st.session_state.retriever.invoke(q)
     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 the answer is 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}],
+        messages=[{
+            "role": "user",
+            "content": f"Use ONLY the PDF content.\n\n{context}\n\nQuestion: {q}"
+        }],
         temperature=0.1
     )
     return res.choices[0].message.content.strip()
 
 # ---------------- SIDEBAR ----------------
 with st.sidebar:
-    if st.button("🗑️ Clear Chat History", use_container_width=True):
-        clear_chat()
-    if st.button("🔥 Clear PDF Memory", use_container_width=True):
-        clear_memory()
-        st.success("Memory cleared")
+    st.button("🗑️ Clear Chat History", on_click=clear_chat, use_container_width=True)
+    st.button("🔥 Clear PDF Memory", on_click=clear_memory, use_container_width=True)
 
     st.markdown("---")
 
@@ -262,8 +238,8 @@ with st.sidebar:
 # ---------------- INPUT ----------------
 disabled = not st.session_state.uploaded_file_name or not client
 
-with st.form(key="chat_form", clear_on_submit=True):
-    col1, col2 = st.columns([0.85, 0.15])
+with st.form("chat_form", clear_on_submit=True):
+    col1, col2 = st.columns([0.9, 0.1])
     with col1:
         question = st.text_input(
             "",
@@ -276,13 +252,10 @@ with st.form(key="chat_form", clear_on_submit=True):
 if send and question:
     st.session_state.messages.append(("user", question))
     with st.spinner("Thinking..."):
-        answer = ask(question)
-        st.session_state.messages.append(("bot", answer))
+        st.session_state.messages.append(("bot", ask(question)))
     st.rerun()
 
-# ---------------- CHAT HISTORY ----------------
+# ---------------- CHAT ----------------
 for role, msg in reversed(st.session_state.messages):
-    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)
+    css = "chat-user" if role == "user" else "chat-bot"
+    st.markdown(f"<div class='{css}'>{msg}</div>", unsafe_allow_html=True)