Update app.py

app.py

@@ -142,48 +142,20 @@ def compute_pdf_hash(pdf_path):
         return hashlib.md5(f.read()).hexdigest()
 
 def get_total_pages(pdf_path):
-    """Get total pages robustly and extract text per page."""
+    """Get total pages and extract text using PyPDF2 for fast processing."""
     reader = PdfReader(pdf_path)
-    # Handle PDFs that are flagged as encrypted but openable without a password
-    try:
-        if getattr(reader, "is_encrypted", False):
-            try:
-                reader.decrypt("")
-            except Exception:
-                pass
-    except Exception:
-        pass
-
-    try:
-        num_pages = len(reader.pages)
-    except Exception:
-        # Fallback: iterate until failure (very rare)
-        num_pages = 0
-        try:
-            while True:
-                _ = reader.pages[num_pages]
-                num_pages += 1
-        except Exception:
-            pass
-
     documents = []
-    for idx in range(num_pages):
-        page_num = idx + 1
-        try:
-            page = reader.pages[idx]
-            text = page.extract_text() or ""
-        except Exception:
-            text = ""
-
-        class Doc:
+    for page_num, page in enumerate(reader.pages, 1):
+        text = page.extract_text() or ""
+        
+        class Doc: 
             def __init__(self, content, page_number):
                 self.content = content
                 self.page_number = page_number
-
-        documents.append(Doc(text, page_num))
-
-    print(f"PDF detected pages: {num_pages} for {os.path.basename(pdf_path)}")
-    return num_pages, documents
+        
+        doc = Doc(text, page_num)
+        documents.append(doc)
+    return len(documents), documents
 
 def get_embeddings(text):
     """Generate embeddings for text."""
@@ -928,4 +900,4 @@ if __name__ == '__main__':
     os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
     os.makedirs(FAISS_INDEX_DIR, exist_ok=True)
     port = int(os.getenv("PORT", "7860"))
-    app.run(host="0.0.0.0", port=port)
+    app.run(host="0.0.0.0", port=port, debug=True)