Update app.py

app.py

@@ -1,67 +1,27 @@
 import streamlit as st
 import google.generativeai as genai
-from langchain.document_loaders import PyPDFDirectoryLoader
+from langchain_community.document_loaders import PyPDFDirectoryLoader
 from dotenv import load_dotenv
 import os
 
-# Load API keys
+# Load environment variables
 load_dotenv()
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 
-# Page configuration
-st.set_page_config(
-    page_title="📚 Chat with PDFs",
-    page_icon="📄",
-    layout="wide"
-)
+# Page config
+st.set_page_config(page_title="📚 PDF Chat Assistant", 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():
-    defaults = {
+    default_values = {
         "messages": [],
         "loaded_files": False,
         "pdf_content": None,
         "chat": None
     }
-    for key, val in defaults.items():
+    for key, value in default_values.items():
         if key not in st.session_state:
-            st.session_state[key] = val
-
+            st.session_state[key] = value
 
-# Load PDFs
 def load_pdfs(folder_path):
     if not os.path.exists(folder_path):
         os.makedirs(folder_path)
@@ -69,78 +29,58 @@ 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)
-    config = {
-        "temperature": 0.7,
-        "top_p": 0.95,
-        "top_k": 40,
-        "max_output_tokens": 8192,
-    }
-
-    model = genai.GenerativeModel("gemini-1.5-pro", generation_config=config)
-
-    prompt = f"""You are a helpful assistant. Use the following PDF content to answer questions:
-{pdf_content}
-If the answer is not in the content, kindly let the user know."""
-    
+    model = genai.GenerativeModel(
+        model_name="gemini-1.5-pro",
+        generation_config={
+            "temperature": 0.7,
+            "top_p": 0.95,
+            "top_k": 40,
+            "max_output_tokens": 8192,
+        }
+    )
+    prompt = f"""You are a helpful assistant. Use the following PDF content to answer questions:\n{pdf_content}\nIf the answer isn't present, say so."""
     chat = model.start_chat(history=[])
     chat.send_message(prompt)
     return chat
 
-
-# Main application
 def main():
     initialize_session_state()
-
-    st.title("📘 Chat with Your PDF Files")
-    st.caption("Built with 💡 Gemini AI & LangChain")
+    st.markdown("<h1 style='text-align: center;'>💬 PDF Chat Assistant</h1>", unsafe_allow_html=True)
 
     with st.sidebar:
         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"
-        )
+        uploaded_files = st.file_uploader("Choose PDF files", type=["pdf"], accept_multiple_files=True)
 
         if uploaded_files and not st.session_state.loaded_files:
-            os.makedirs("pdfs", exist_ok=True)
+            if not os.path.exists("pdfs"):
+                os.makedirs("pdfs")
             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_out:
                     f_out.write(file.getvalue())
 
-            with st.spinner("🔍 Processing PDFs..."):
+            with st.spinner("🔄 Reading PDFs..."):
                 try:
-                    pdf_text = load_pdfs("pdfs")
-                    st.session_state.pdf_content = pdf_text
-                    st.session_state.chat = initialize_chat(pdf_text)
+                    content = load_pdfs("pdfs")
+                    st.session_state.pdf_content = content
                     st.session_state.loaded_files = True
-                    st.success("PDFs processed and ready to chat!")
+                    st.session_state.chat = initialize_chat(content)
+                    st.success("✅ PDFs processed successfully!")
                 except Exception as e:
-                    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)
+                    st.error(f"❌ Error loading PDFs: {str(e)}")
 
-    # 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(f"<div class='stChatMessage {css_class}'>{msg['content']}</div>", unsafe_allow_html=True)
+        for message in st.session_state.messages:
+            with st.chat_message(message["role"]):
+                st.markdown(message["content"])
 
         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(f"<div class='stChatMessage user-message'>{prompt}</div>", unsafe_allow_html=True)
+                st.markdown(prompt)
 
             with st.chat_message("assistant"):
                 placeholder = st.empty()
@@ -149,12 +89,12 @@ def main():
                         st.session_state.chat = initialize_chat(st.session_state.pdf_content)
                     response = st.session_state.chat.send_message(prompt)
                     reply = response.text
-                    placeholder.markdown(f"<div class='stChatMessage assistant-message'>{reply}</div>", unsafe_allow_html=True)
+                    placeholder.markdown(reply)
                     st.session_state.messages.append({"role": "assistant", "content": reply})
                 except Exception as e:
-                    placeholder.error(f"⚠️ Error generating response: {str(e)}")
+                    placeholder.error(f"❌ Failed to respond: {str(e)}")
     else:
-        st.info("📤 Upload your PDF documents from the sidebar to begin chatting.")
+        st.info("Upload some PDF files to begin chatting with them!")
 
 if __name__ == "__main__":
     main()