Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| import pdfplumber | |
| import docx | |
| import pylatexenc | |
| class docLoader(): | |
| def __init__(self): | |
| pass | |
| def load(self, uploaded_file): | |
| if uploaded_file is not None: | |
| st.toast("File uploaded successfully!", icon="✅") | |
| file_extension = uploaded_file.name.split(".")[-1] | |
| load_functions = { | |
| "pdf": self.load_pdf, | |
| "txt": self.load_txt, | |
| "docx": self.load_docx, | |
| "tex": self.load_tex | |
| } | |
| if file_extension in load_functions: | |
| text = load_functions[file_extension](uploaded_file) | |
| else: | |
| st.write("Unsupported file format") | |
| else: | |
| text = '' | |
| return text | |
| def load_pdf(self, uploaded_file): | |
| with pdfplumber.open(uploaded_file) as pdf: | |
| pages = pdf.pages | |
| text = "" | |
| for page in pages: | |
| text += page.extract_text() | |
| return text | |
| def load_txt(self, uploaded_file): | |
| return uploaded_file.getvalue().decode("utf-8") | |
| def load_docx(self, uploaded_file): | |
| docx_text = docx.Document(uploaded_file) | |
| full_text = [para.text for para in docx_text.paragraphs] | |
| return "\n".join(full_text) | |
| def load_tex(self, uploaded_file): | |
| with open(uploaded_file.name, 'r') as tex_file: | |
| tex_content = tex_file.read() | |
| return pylatexenc.latex2text(tex_content) | |
| def load_doc(): | |
| uploaded_file = st.file_uploader("Choose a document file", type=["pdf", "txt", "docx"]) | |
| loader = docLoader() | |
| return loader.load(uploaded_file) | |
| if __name__ == "__main__": | |
| load_doc() |