Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
 import os
 from threading import Thread
 
-import fitz
+import pymupdf
 import docx
 from pptx import Presentation
 
@@ -56,7 +56,7 @@ def extract_text(path):
     return open(path, 'r').read()
 
 def extract_pdf(path):
-    doc = fitz.open(path)
+    doc = pymupdf.open(path)
     text = ""
     for page in doc:
         text += page.get_text()
@@ -82,12 +82,13 @@ def extract_pptx(path):
 def mode_load(path):
     choice = ""
     file_type = path.split(".")[-1]
+    print(file_type)
     if file_type in ["pdf", "txt", "py", "docx", "pptx", "json", "cpp", "md"]:
-        if file_type.endswith(".pdf"):
+        if file_type.endswith("pdf"):
             content = extract_pdf(path)
-        elif file_type.endswith(".docx"):
+        elif file_type.endswith("docx"):
             content = extract_docx(path)
-        elif file_type.endswith(".pptx"):
+        elif file_type.endswith("pptx"):
             content = extract_pptx(path)
         else:
             content = extract_text(path)