Update app.py

app.py

@@ -203,7 +203,9 @@ def extract_mime_type(file):
     else:
         raise TypeError("Input should be a string or a streamlit.UploadedFile object")
 
-from io import BytesIO
+from PyPDF2 import PdfFileReader
+import os
+import re
 
 def extract_file_extension(file):
     # get the file name directly from the UploadedFile object
@@ -229,8 +231,12 @@ def pdf2txt(docs):
 
         # read the file according to its extension
         try:
-            if file_extension.lower() in ['txt', 'html', 'htm', 'py', 'xml', 'json', 'docx']:
-                text += textract.process(temp_file_name).decode("utf-8")
+            if file_extension.lower() == 'py':
+                with open(temp_file_name, 'r') as f:
+                    text += f.read()
+            elif file_extension.lower() in ['txt', 'html', 'htm', 'xml', 'json']:
+                with open(temp_file_name, 'r') as f:
+                    text += f.read()
             elif file_extension.lower() == 'pdf':
                 with open(temp_file_name, "rb") as f:
                     pdf = PdfFileReader(f)
@@ -245,6 +251,7 @@ def pdf2txt(docs):
     return text
 
 
+
 def pdf2txt_old(pdf_docs):
     st.write(pdf_docs)
     for file in pdf_docs: