Update app.py via AI Editor

app.py

@@ -15,10 +15,9 @@ import base64
 import datetime
 from werkzeug.utils import secure_filename
 import numpy as np
-
 import io
 
-from pdfminer.high_level import extract_text as pdf_extract_text
+import PyPDF2
 import docx
 import openpyxl
 
@@ -341,7 +340,13 @@ def _extract_text_from_upload(filepath, ext):
             return text
         elif ext == ".pdf":
             try:
-                text = pdf_extract_text(filepath)
+                text = ""
+                with open(filepath, "rb") as f:
+                    reader = PyPDF2.PdfReader(f)
+                    for page in reader.pages:
+                        page_text = page.extract_text()
+                        if page_text:
+                            text += page_text + "\n"
                 return text
             except Exception as e:
                 logger.error(f"Error reading PDF {filepath}: {e}")
@@ -575,7 +580,6 @@ def main_callback(session_id, send_clicks, file_contents, new_chat_clicks, strea
                         session_lock = get_session_lock(session_id)
                         with session_lock:
                             load_session_state(session_id)
-                            state = get_session_state(session_id)
                             state["messages"].append({"role": "assistant", "content": reply})
                             state["stream_buffer"] = ""
                             state["streaming"] = False