Update app.py

app.py

@@ -4,25 +4,64 @@ from langchain.document_loaders import PyPDFDirectoryLoader
 from dotenv import load_dotenv
 import os
 
-# Load API keys from .env
+# Load API keys
 load_dotenv()
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
-# HF_API_KEY = os.getenv("HF_API_KEY")  # Not used directly here, but loaded if needed
 
 # Page configuration
-st.set_page_config(page_title="Chat with PDFs", page_icon="📚")
+st.set_page_config(
+    page_title="📚 Chat with PDFs",
+    page_icon="📄",
+    layout="wide"
+)
 
+# Custom CSS styling
+st.markdown("""
+    <style>
+        .main {
+            background-color: #f7f9fc;
+            padding: 20px;
+        }
+        .stChatMessage {
+            background-color: #ffffff;
+            border-radius: 1rem;
+            padding: 15px;
+            margin: 10px 0;
+            box-shadow: 0 2px 8px rgba(0,0,0,0.05);
+        }
+        .user-message {
+            background-color: #e0f7fa;
+            border-left: 5px solid #00796b;
+        }
+        .assistant-message {
+            background-color: #fff8e1;
+            border-left: 5px solid #fbc02d;
+        }
+        .sidebar .stButton button {
+            width: 100%;
+        }
+        .uploaded-files {
+            font-size: 0.9rem;
+            color: #666;
+        }
+    </style>
+""", unsafe_allow_html=True)
+
+
+# Initialize session variables
 def initialize_session_state():
-    session_vars = {
+    defaults = {
         "messages": [],
         "loaded_files": False,
         "pdf_content": None,
         "chat": None
     }
-    for key, value in session_vars.items():
+    for key, val in defaults.items():
         if key not in st.session_state:
-            st.session_state[key] = value
+            st.session_state[key] = val
+
 
+# Load PDFs
 def load_pdfs(folder_path):
     if not os.path.exists(folder_path):
         os.makedirs(folder_path)
@@ -30,64 +69,78 @@ def load_pdfs(folder_path):
     documents = loader.load()
     return "\n\n".join(doc.page_content for doc in documents)
 
+
+# Initialize Gemini chat
 def initialize_chat(pdf_content):
     genai.configure(api_key=GOOGLE_API_KEY)
-    generation_config = {
+    config = {
         "temperature": 0.7,
         "top_p": 0.95,
         "top_k": 40,
         "max_output_tokens": 8192,
     }
 
-    model = genai.GenerativeModel(
-        model_name="gemini-1.5-pro",
-        generation_config=generation_config,
-    )
+    model = genai.GenerativeModel("gemini-1.5-pro", generation_config=config)
 
-    context_prompt = f"""You are a helpful assistant. Use the following PDF content to answer user questions:
+    prompt = f"""You are a helpful assistant. Use the following PDF content to answer questions:
 {pdf_content}
-If the answer isn't in the content, reply accordingly."""
+If the answer is not in the content, kindly let the user know."""
     
     chat = model.start_chat(history=[])
-    chat.send_message(context_prompt)
+    chat.send_message(prompt)
     return chat
 
+
+# Main application
 def main():
     initialize_session_state()
-    st.title("💬 Chat with PDFs")
+
+    st.title("📘 Chat with Your PDF Files")
+    st.caption("Built with 💡 Gemini AI & LangChain")
 
     with st.sidebar:
-        st.header("Upload PDFs")
-        uploaded_files = st.file_uploader("Upload PDF files", type=["pdf"], accept_multiple_files=True)
+        st.header("📁 Upload PDFs")
+        uploaded_files = st.file_uploader(
+            "Drag and drop PDFs here",
+            type=["pdf"],
+            accept_multiple_files=True,
+            help="You can upload multiple PDFs"
+        )
 
         if uploaded_files and not st.session_state.loaded_files:
-            if not os.path.exists("pdfs"):
-                os.makedirs("pdfs")
-            for file in os.listdir("pdfs"):
-                os.remove(os.path.join("pdfs", file))
+            os.makedirs("pdfs", exist_ok=True)
+            for f in os.listdir("pdfs"):
+                os.remove(os.path.join("pdfs", f))
             for file in uploaded_files:
-                with open(f"pdfs/{file.name}", "wb") as f:
-                    f.write(file.getvalue())
+                with open(f"pdfs/{file.name}", "wb") as f_out:
+                    f_out.write(file.getvalue())
 
-            with st.spinner("Processing PDFs..."):
+            with st.spinner("🔍 Processing PDFs..."):
                 try:
                     pdf_text = load_pdfs("pdfs")
                     st.session_state.pdf_content = pdf_text
-                    st.session_state.loaded_files = True
                     st.session_state.chat = initialize_chat(pdf_text)
+                    st.session_state.loaded_files = True
+                    st.success("PDFs processed and ready to chat!")
                 except Exception as e:
-                    st.error(f"Error loading PDFs: {str(e)}")
-                    return
+                    st.error(f"Failed to process PDFs: {str(e)}")
+
+        if st.session_state.loaded_files:
+            st.subheader("📄 Files Uploaded")
+            for file in uploaded_files:
+                st.markdown(f"- {file.name}", unsafe_allow_html=True)
 
+    # Chat section
     if st.session_state.loaded_files:
         for msg in st.session_state.messages:
+            css_class = "user-message" if msg["role"] == "user" else "assistant-message"
             with st.chat_message(msg["role"]):
-                st.markdown(msg["content"])
+                st.markdown(f"<div class='stChatMessage {css_class}'>{msg['content']}</div>", unsafe_allow_html=True)
 
-        if prompt := st.chat_input("Ask a question about your PDFs:"):
+        if prompt := st.chat_input("Ask something about the PDFs..."):
             st.session_state.messages.append({"role": "user", "content": prompt})
             with st.chat_message("user"):
-                st.markdown(prompt)
+                st.markdown(f"<div class='stChatMessage user-message'>{prompt}</div>", unsafe_allow_html=True)
 
             with st.chat_message("assistant"):
                 placeholder = st.empty()
@@ -95,13 +148,13 @@ def main():
                     if not st.session_state.chat:
                         st.session_state.chat = initialize_chat(st.session_state.pdf_content)
                     response = st.session_state.chat.send_message(prompt)
-                    response_text = response.text
-                    placeholder.markdown(response_text)
-                    st.session_state.messages.append({"role": "assistant", "content": response_text})
+                    reply = response.text
+                    placeholder.markdown(f"<div class='stChatMessage assistant-message'>{reply}</div>", unsafe_allow_html=True)
+                    st.session_state.messages.append({"role": "assistant", "content": reply})
                 except Exception as e:
-                    placeholder.error(f"Error generating response: {str(e)}")
+                    placeholder.error(f"⚠️ Error generating response: {str(e)}")
     else:
-        st.info("Please upload PDFs to begin.")
+        st.info("📤 Upload your PDF documents from the sidebar to begin chatting.")
 
 if __name__ == "__main__":
     main()