Update app.py

app.py

@@ -30,7 +30,13 @@ else:
     st.warning("⚠️ GROQ_API_KEY not found. Please add it to Hugging Face secrets.")
 
 # ---------------- STREAMLIT UI SETUP ----------------
-st.set_page_config(page_title="PDF Assistant", page_icon="📘", layout="wide")
+# CHANGED: Added initial_sidebar_state="expanded" to ensure it starts open
+st.set_page_config(
+    page_title="PDF Assistant", 
+    page_icon="📘", 
+    layout="wide", 
+    initial_sidebar_state="expanded"
+)
 
 # ---------------- CSS ----------------
 st.markdown("""
@@ -41,13 +47,26 @@ html, body {
     height: 100%;
     margin: 0;
 }
-/* 2. HIDE DEFAULT STREAMLIT ELEMENTS */
+
+/* 2. HIDE DEFAULT STREAMLIT ELEMENTS & SIDEBAR TOGGLES */
 header[data-testid="stHeader"] {
     display: none;
 }
 footer {
     display: none;
 }
+
+/* --- NEW CODE START: HIDE SIDEBAR TOGGLE BUTTONS --- */
+/* Hide the 'Close Sidebar' (<<) button inside the sidebar */
+section[data-testid="stSidebar"] > div > div:first-child {
+    display: none;
+}
+/* Hide the 'Open Sidebar' (>) button on main screen */
+[data-testid="collapsedControl"] {
+    display: none;
+}
+/* --- NEW CODE END --- */
+
 /* 3. SIDEBAR STYLING (INDEPENDENT LEFT PANEL SCROLL) */
 [data-testid="stSidebar"] {
     position: fixed;
@@ -65,6 +84,7 @@ footer {
     background: #2d3748;
     border-radius: 3px;
 }
+
 /* 4. FIXED HEADER STYLING */
 .fixed-header {
     position: fixed;
@@ -80,6 +100,7 @@ footer {
     align-items: center;
     border-bottom: 1px solid rgba(255, 255, 255, 0.1);
 }
+
 /* 5. MAIN CONTENT SCROLLING (INDEPENDENT RIGHT PANEL SCROLL) */
 .main .block-container {
     margin-top: 6rem;
@@ -95,6 +116,7 @@ footer {
     background: #2d3748;
     border-radius: 4px;
 }
+
 /* 6. SIDEBAR BUTTON STYLING */
 [data-testid="stSidebar"] .stButton button {
     width: 100%;
@@ -102,6 +124,7 @@ footer {
     font-weight: 600;
     margin-bottom: 6px;
 }
+
 /* 7. HIDE UPLOADED FILE LIST & NAME */
 [data-testid='stFileUploaderFile'] {
     display: none;
@@ -112,6 +135,7 @@ section[data-testid="stFileUploader"] ul {
 section[data-testid="stFileUploader"] small {
     display: none;
 }
+
 /* 8. CHAT BUBBLES */
 .chat-user {
     background: #2d3748;
@@ -131,10 +155,10 @@ section[data-testid="stFileUploader"] small {
     text-align: left;
     color: #ffffff;
 }
-/* Sources CSS removed/hidden as it is no longer used */
 .sources {
     display: none; 
 }
+
 /* 9. TITLE TEXT */
 .title-text {
     font-size: 2.5rem;
@@ -151,6 +175,7 @@ section[data-testid="stFileUploader"] small {
     color: #4da6ff;
     text-decoration: none;
 }
+
 /* 10. INPUT FORM STYLING */
 [data-testid="stForm"] {
     border: none;
@@ -169,20 +194,15 @@ st.markdown("""
 </div>
 """, unsafe_allow_html=True)
 
-
 # ---------------- 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
 
@@ -241,13 +261,14 @@ def ask_question(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 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}
+        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...
-"""
+
+Answer on your behalf, don't say based on the context..."""
         
         response = client.chat.completions.create(
             model=GROQ_MODEL,
@@ -266,16 +287,17 @@ 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!")
-
+    
+    if st.button("🔥 Clear PDF Memory", on_click=clear_memory, use_container_width=True): 
+        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"
     )
-
+    
     if uploaded:
         if uploaded.name != st.session_state.uploaded_file_name:
             st.session_state.uploaded_file_name = None
@@ -313,7 +335,6 @@ if submit_btn and user_question:
     with st.spinner("Thinking..."):
         answer, sources, error = ask_question(user_question)
         if answer:
-            # CHANGED: Removed the 'Context Chunks Used' HTML from the message
             bot_msg = answer
             st.session_state.chat.append(("bot", bot_msg))
         else: