Update app.py

app.py

@@ -7,22 +7,12 @@ from langchain.schema import StrOutputParser
 from docx import Document
 import fitz  # PyMuPDF
 
-def extract_text_from_pdf_or_docx(file):
-    """Extract text from PDF or Word document."""
-    filename = file.name
+def extract_text_from_pdf(file):
+    """Extract text from PDF."""
     text = ""
-    if filename.endswith('.pdf'):
-        # Extract text from PDF
-        with fitz.open(file) as doc:
-            for page in doc:
-                text += page.get_text()
-    elif filename.endswith('.docx'):
-        # Extract text from Word document
-        doc = Document(file)
-        for paragraph in doc.paragraphs:
-            text += paragraph.text + "\n"
-    else:
-        text = "Unsupported file format. Please upload a PDF or Word document."
+    doc = fitz.open(stream=file.read(), filetype="pdf")
+    for page in doc:
+        text += page.get_text()
     return text
 
 def create_multiple_choice_prompt(num_questions, quiz_context, expertise):