Upload app.py

app.py

@@ -3,7 +3,7 @@ import streamlit as st
 import fitz
 from PIL import Image
 import tempfile
-from langchain_community.document_loaders import PyPDFLoader, TextLoader, Docx2txtLoader
+from langchain_community.document_loaders import PyPDFLoader, Docx2txtLoader
 from langchain.chains.question_answering import load_qa_chain
 from docx import Document
 import io
@@ -27,9 +27,9 @@ def process_document(uploaded_file, query):
     with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
         tmp_file.write(uploaded_file.read())
         file_extension = os.path.splitext(uploaded_file.name)[1].lower()
+        document_text = None
         if file_extension == ".pdf":
             loader = PyPDFLoader(tmp_file.name)
-            document_text = None
         elif file_extension == ".docx":
             loader = Docx2txtLoader(tmp_file.name)
             document = Document(tmp_file.name)
@@ -69,7 +69,7 @@ def main():
     if uploaded_file is not None:
         st.title("Document Content")
         file_extension = os.path.splitext(uploaded_file.name)[1].lower()
-        if file_extension in [".docx"]:
+        if file_extension == ".docx":
             response, document_text = process_document(uploaded_file, "")
             if document_text is not None:
                 st.text_area("Document Text", value=document_text, height=300)