Update app.py

app.py

@@ -22,7 +22,6 @@ client = None
 if GROQ_API_KEY:
     try:
         client = Groq(api_key=GROQ_API_KEY)
-        # ✅ Removed the st.success message (Problem #1)
         logging.info("✅ Groq client initialized successfully.")
     except Exception as e:
         st.error(f"❌ Failed to initialize Groq client: {e}")
@@ -36,10 +35,7 @@ st.set_page_config(page_title="PDF Assistant", page_icon="📘", layout="wide")
 # ---------------- CSS ----------------
 st.markdown("""
 <style>
-/* -------------------------------------- */
-/* 1. SCROLL FIX FOR SIDEBAR & MAIN AREA  */
-/* -------------------------------------- */
-/* Fix the sidebar and prevent it from scrolling */
+/* 1. SCROLL FIX FOR SIDEBAR & MAIN AREA */
 [data-testid="stSidebar"] {
     position: fixed;
     overflow-y: hidden !important;
@@ -47,28 +43,13 @@ st.markdown("""
     z-index: 99;
 }
 
-/* Allow the main content area to scroll independently */
 .main .block-container {
     overflow-y: auto;
     height: 100vh;
-    padding-bottom: 5rem; /* Space for scrolling */
+    padding-bottom: 5rem;
 }
 
-/* Hide main page scrollbar for cleaner look */
-::-webkit-scrollbar {
-    width: 8px;
-}
-::-webkit-scrollbar-track {
-    background: #0e1117;
-}
-::-webkit-scrollbar-thumb {
-    background: #2d3748;
-    border-radius: 4px;
-}
-
-/* -------------------------------------- */
-/* 2. CHAT & UI STYLING                   */
-/* -------------------------------------- */
+/* 2. CHAT & UI STYLING */
 :root {
     --primary-color: #1e3a8a;
     --background-color: #0e1117;
@@ -103,10 +84,28 @@ st.markdown("""
     padding-top: 5px;
 }
 
-/* -------------------------------------- */
-/* 3. INPUT FORM STYLING                  */
-/* -------------------------------------- */
-/* Align the input box and button properly */
+/* 3. CENTER TITLE STYLING */
+.title-container {
+    text-align: center;
+    margin-bottom: 20px;
+}
+.title-text {
+    font-size: 3rem;
+    font-weight: 800;
+    margin: 0;
+}
+.creator-text {
+    font-size: 1.1rem;
+    font-weight: 500;
+    margin-top: -5px;
+    color: #cccccc;
+}
+.creator-text a {
+    color: #4da6ff;
+    text-decoration: none;
+}
+
+/* 4. INPUT FORM STYLING */
 [data-testid="stForm"] {
     border: none;
     padding: 0;
@@ -142,7 +141,7 @@ def clear_memory():
     gc.collect()
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
-    st.success("Memory cleared. Please upload a new PDF.")
+    # Notification moved to UI logic to avoid rerun issues
 
 def process_pdf(uploaded_file):
     """Process uploaded PDF and create vectorstore."""
@@ -224,58 +223,64 @@ FINAL ANSWER:
     except Exception as e:
         return None, 0, f"General error: {str(e)}"
 
-# ---------------- UI COMPONENTS ----------------
-
-# ✅ Problem #2: Title and Creator info at Top Left
-col_header, col_space = st.columns([0.6, 0.4])
-with col_header:
-    st.title("📘 PDF Assistant")
-    st.markdown("**Created by [Abhishek Saxena](https://www.linkedin.com/in/abhishek-iitr/)**")
+# ---------------- CENTERED TITLE ----------------
+st.markdown("""
+<div class="title-container">
+    <div class="title-text">📘 PDF Assistant</div>
+    <div class="creator-text">
+        Created by <a href="https://www.linkedin.com/in/abhishek-iitr/" target="_blank">Abhishek Saxena</a>
+    </div>
+</div>
+""", unsafe_allow_html=True)
 
-# Sidebar Controls
+# ---------------- SIDEBAR CONTROLS ----------------
 with st.sidebar:
     st.header("Controls")
-    st.button("🗑️ Clear Chat History", on_click=clear_chat_history, use_container_width=True)
-    st.button("🔥 Clear PDF Memory", on_click=clear_memory, use_container_width=True)
     
-    st.markdown("---")
-    if st.session_state.uploaded_file_name:
-        st.success(f"✅ **Active PDF:**\n `{st.session_state.uploaded_file_name}`")
-    else:
-        st.warning("⬆️ Upload a PDF to start chatting!")
-
-# File Upload
-uploaded = st.file_uploader(
-    "Upload your PDF",
-    type=["pdf"],
-    key=st.session_state.uploader_key
-)
+    # Buttons
+    col_clear_chat, col_clear_mem = st.columns(2)
+    with col_clear_chat:
+        if st.button("🗑️ Chat", use_container_width=True):
+            clear_chat_history()
+    with col_clear_mem:
+        if st.button("🔥 Memory", on_click=clear_memory, use_container_width=True):
+             st.success("Cleared!")
 
-if uploaded and uploaded.name != st.session_state.uploaded_file_name:
-    st.session_state.uploaded_file_name = None
-    st.session_state.chat = []
+    st.markdown("---")
     
-    with st.spinner(f"Processing '{uploaded.name}'..."):
-        chunks_count = process_pdf(uploaded)
-        
-        if chunks_count is not None:
-            st.success(f"✅ PDF processed successfully! {chunks_count} chunks created.")
-            st.session_state.uploaded_file_name = uploaded.name
-        else:
-            st.error("❌ Failed to process PDF")
+    # File Upload
+    uploaded = st.file_uploader(
+        "Upload your PDF",
+        type=["pdf"],
+        key=st.session_state.uploader_key
+    )
+
+    # Status Message directly below uploader
+    if uploaded:
+        if uploaded.name != st.session_state.uploaded_file_name:
+            # Processing logic
             st.session_state.uploaded_file_name = None
-    
-    st.rerun()
+            st.session_state.chat = []
+            
+            with st.spinner(f"Processing '{uploaded.name}'..."):
+                chunks_count = process_pdf(uploaded)
+                
+                if chunks_count is not None:
+                    st.session_state.uploaded_file_name = uploaded.name
+                    st.success(f"✅ Processed! {chunks_count} chunks.")
+                else:
+                    st.error("❌ Failed.")
+        else:
+            # File is already processed and active
+            st.success(f"✅ **Active:** `{uploaded.name}`")
+    else:
+        # No file uploaded yet
+        st.warning("⬆️ Upload a PDF to start chatting!")
 
 # ---------------- INPUT AREA ----------------
-# ✅ Problem #3: "Enter" key submission + Arrow button using st.form
-# We use a form so that pressing "Enter" on the keyboard submits the question.
-
 disabled_input = st.session_state.uploaded_file_name is None or client is None
 
-# Create a form container
 with st.form(key='chat_form', clear_on_submit=True):
-    # Use columns to put the input box and the button side-by-side
     col_input, col_btn = st.columns([0.85, 0.15], gap="small")
     
     with col_input:
@@ -287,41 +292,29 @@ with st.form(key='chat_form', clear_on_submit=True):
         )
         
     with col_btn:
-        # The arrow button acts as the submit button for the form
         submit_btn = st.form_submit_button("➤", disabled=disabled_input, use_container_width=True)
 
-# Processing the Input (Run only when form is submitted)
-if submit_btn or (user_question and not disabled_input): 
-    # Note: Streamlit forms usually require the submit button click check, 
-    # but clear_on_submit=True handles the reset.
+# Processing the Input
+if submit_btn and user_question:
+    st.session_state.chat.append(("user", user_question))
     
-    if user_question:
-        # Add user query to chat history
-        st.session_state.chat.append(("user", user_question))
-        
-        # Get answer
-        with st.spinner("Thinking..."):
-            answer, sources, error = ask_question(user_question)
-            
-            if answer:
-                bot_message = f"{answer}<div class='sources'>Context Chunks Used: {sources}</div>"
-                st.session_state.chat.append(("bot", bot_message))
-            else:
-                st.session_state.chat.append(("bot", f"🔴 **Error:** {error}"))
+    with st.spinner("Thinking..."):
+        answer, sources, error = ask_question(user_question)
         
-        # Rerun to update the chat history display below
-        st.rerun()
+        if answer:
+            bot_message = f"{answer}<div class='sources'>Context Chunks Used: {sources}</div>"
+            st.session_state.chat.append(("bot", bot_message))
+        else:
+            st.session_state.chat.append(("bot", f"🔴 **Error:** {error}"))
+    
+    # No st.rerun() needed here; the submit interaction refreshes the page automatically.
 
 # ---------------- CHAT HISTORY DISPLAY ----------------
 st.markdown("---")
-st.markdown("## Chat History")
-
-# Display in reverse order (newest top) or standard (newest bottom)? 
-# Standard chat is newest bottom.
+# Display newest messages at the top? (Standard is bottom, but customizable)
+# Currently displaying standard order (Top = Oldest, Bottom = Newest)
 for role, msg in 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)
-
-# (Footer removed as requested, info moved to top)
+        st.markdown(f"<div class='chat-bot'>{msg}</div>", unsafe_allow_html=True)