Update app.py

app.py

@@ -122,7 +122,7 @@ def process_pdf(uploaded_file):
         st.error(f"PDF error: {str(e)}")
         return None, None, "", ""
 
-# Improved summarization
+# Improved summarization: More chunks, better concatenation
 def summarize_pdf(text):
     logger.info("Generating summary")
     try:
@@ -144,7 +144,7 @@ def summarize_pdf(text):
         logger.error(f"Summary error: {str(e)}")
         return f"Oops, something went wrong summarizing: {str(e)}"
 
-# FIXED answer_question function
+# Improved Q&A: Better context, code handling
 def answer_question(text_vector_store, code_vector_store, query):
     logger.info(f"Processing query: {query}")
     try:
@@ -161,13 +161,13 @@ def answer_question(text_vector_store, code_vector_store, query):
         if not vector_store:
             return "No relevant content found for your query."
         
-        docs = vector_store.similarity_search(query, k=4)
+        docs = vector_store.similarity_search(query, k=4)  # Increased for better context
         context = "\n\n".join(doc.page_content for doc in docs)
         prompt = f"Based on this context: {context}\n\nQuestion: {query}\n\nProvide a friendly, concise answer like a helpful assistant:"
         response = qa_pipeline(prompt)[0]['generated_text'].strip()
         
         if is_code_query:
-            response = f"Here's the relevant code extracted from the PDF:\n```python\n{context}\n```"
+            response = f"Here's the relevant code extracted from the PDF:\n```python\n{response}\n```"
         
         logger.info("Answer generated")
         return f"Got it! Here's the answer:\n\n{response}"
@@ -179,7 +179,7 @@ def answer_question(text_vector_store, code_vector_store, query):
 try:
     st.set_page_config(page_title="Smart PDF Q&A", page_icon="📄", layout="wide", initial_sidebar_state="expanded")
     
-    # Enhanced CSS with fixed chatbox colors
+    # Enhanced CSS for modern, dark-theme friendly design (matching the screenshot's dark mode)
     st.markdown("""
         <style>
         /* General styling */
@@ -192,24 +192,9 @@ try:
         .stSuccess, .stError { border-radius: 6px; padding: 10px; }
         
         /* Chat messages */
-        .stChatMessage {
-            background-color: #000000 !important;
-            color: #ffffff !important;
-            border-radius: 10px;
-            padding: 12px;
-            margin: 8px 0;
-        }
-        .stChatMessage p, 
-        .stChatMessage span, 
-        .stChatMessage div {
-            color: #ffffff !important;
-        }
-        .stChatMessage.user {
-            background-color: #111111 !important;
-        }
-        .stChatMessage.assistant {
-            background-color: #1a1a1a !important;
-        }
+        .stChatMessage { border-radius: 12px; padding: 12px; margin: 8px 0; box-shadow: 0 2px 4px rgba(0,0,0,0.2); }
+        .stChatMessage.user { background-color: #3a3a3a; color: #ffffff; }
+        .stChatMessage.assistant { background-color: #2a2a2a; color: #ffffff; }
         
         /* Code blocks */
         pre { background-color: #252525; color: #d4d4d4; padding: 12px; border-radius: 6px; overflow: auto; font-family: 'Consolas', 'Monaco', monospace; }
@@ -288,4 +273,4 @@ try:
 
 except Exception as e:
     logger.error(f"App initialization failed: {str(e)}")
-    st.error(f"App failed to start: {str(e)}. Please check the logs or contact support.")
+    st.error(f"App failed to start: {str(e)}. Please check the logs or contact support.")