Update app.py

app.py

@@ -36,13 +36,18 @@ if not GOOGLE_API_KEY:
 # Configure the API key
 genai.configure(api_key=GOOGLE_API_KEY)
 
-def get_pdf_text(pdf_docs):
-    """Extract text from uploaded PDF documents."""
+def get_pdf_text(pdf_path):
+    """Extract text from a PDF document."""
     text = ""
-    for pdf in pdf_docs:
-        pdf_reader = PdfReader(pdf)
-        for page in pdf_reader.pages:
-            text += page.extract_text()
+    try:
+        with open(pdf_path, "rb") as pdf_file:
+            pdf_reader = PdfReader(pdf_file)
+            for page in pdf_reader.pages:
+                extracted_text = page.extract_text()
+                if extracted_text:
+                    text += extracted_text
+    except Exception as e:
+        st.error(f"Error reading PDF file: {e}")
     return text
 
 def get_text_chunks(text):
@@ -119,7 +124,10 @@ def main():
     st.set_page_config(page_title="Robi AI Customer Service", page_icon="📱")
     
     # Display Logo
-    st.image("robi_logo.png", width=200)  # Ensure you have a 'robi_logo.png' in your project directory
+    if os.path.exists("robi_logo.png"):
+        st.image("robi_logo.png", width=200)  # Ensure you have a 'robi_logo.png' in your project directory
+    else:
+        st.warning("Logo image 'robi_logo.png' not found. Please add it to the project directory.")
     
     # Title and Description
     st.title("📶 Robi AI Customer Service & Recommendation System")
@@ -197,16 +205,20 @@ def main():
         st.header("📄 Documents:")
         # Check if the FAISS index exists
         if not os.path.exists("faiss_index/index.faiss"):
-            with st.spinner("Processing documents..."):
-                # Process the provided PDFs
-                uploaded_files = st.file_uploader("Upload Robi Information PDF", type=["pdf"], accept_multiple_files=True)
-                if uploaded_files:
-                    raw_text = get_pdf_text(uploaded_files)
-                    text_chunks = get_text_chunks(raw_text)
-                    get_vector_store(text_chunks)             
-                    st.success("Documents processed and FAISS index created.")
+            pdf_path = "customerservicebot.pdf"  # Default PDF file
+            if os.path.exists(pdf_path):
+                with st.spinner(f"Processing '{pdf_path}'..."):
+                    raw_text = get_pdf_text(pdf_path)
+                    if raw_text:
+                        text_chunks = get_text_chunks(raw_text)
+                        get_vector_store(text_chunks)             
+                        st.success(f"'{pdf_path}' processed and FAISS index created.")
+                    else:
+                        st.error(f"No text extracted from '{pdf_path}'. Please check the PDF content.")
+            else:
+                st.error(f"Default PDF '{pdf_path}' not found. Please add it to the project directory.")
         else:
             st.info("FAISS index loaded from 'faiss_index'.")
 
 if __name__ == "__main__":
-    main()
+    main()