File size: 3,907 Bytes
fac9a75 4b0678e c982cf8 5f1077a 4b0678e fac9a75 579432a fac9a75 c982cf8 39f86d4 5f1077a c982cf8 0bce450 5f1077a 0bce450 5f1077a 0bce450 5f1077a c982cf8 5f1077a 579432a fac9a75 579432a fac9a75 5f1077a 579432a 5f1077a fac9a75 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 |
from pptx import Presentation
import gradio as gr
from pdf2image import convert_from_path
import pdfplumber
from docx import Document
import subprocess
import os
def extract_text_from_pptx(file_path):
prs = Presentation(file_path)
text_content = []
for slide in prs.slides:
slide_text = []
for shape in slide.shapes:
if hasattr(shape, "text"):
slide_text.append(shape.text)
text_content.append("\n".join(slide_text))
return "\n\n".join(text_content)
def extract_text_from_ppt(file_path):
try:
# Convert PPT to PPTX using unoconv
pptx_file_path = os.path.splitext(file_path)[0] + '.pptx'
subprocess.run(['unoconv', '-f', 'pptx', file_path], check=True)
# Extract text from PPTX
presentation = Presentation(pptx_file_path)
text_content = []
for slide in presentation.slides:
slide_text = []
for shape in slide.shapes:
if hasattr(shape, "text"):
slide_text.append(shape.text)
text_content.append("\n".join(slide_text))
# Remove the converted PPTX file
os.remove(pptx_file_path)
return "\n\n".join(text_content)
except Exception as e:
print(f"Error extracting text from PPT file: {e}")
return "Error extracting text from PPT file"
def extract_text_from_ppt_or_pptx(file_path):
if file_path.endswith('.pptx'):
return extract_text_from_pptx(file_path)
elif file_path.endswith('.ppt'):
return extract_text_from_ppt(file_path)
else:
return "Unsupported file type. Please provide a .ppt or .pptx file."
def convert_pdf_to_image(file):
images = convert_from_path(file)
return images
def extract_text_from_pdf(file):
text = ""
with pdfplumber.open(file) as pdf:
for page in pdf.pages:
text += page.extract_text() + "\n"
return text
def extract_text_from_docx(file):
text = ""
doc = Document(file.name)
for paragraph in doc.paragraphs:
text += paragraph.text + "\n"
return text
def convert_doc_to_text(doc_path):
try:
subprocess.run(
["unoconv", "--format", "txt", doc_path],
capture_output=True,
text=True,
check=True,
)
txt_file_path = doc_path.replace(".doc", ".txt")
with open(txt_file_path, "r") as f:
text = f.read()
text = text.lstrip("\ufeff")
os.remove(txt_file_path)
return text
except subprocess.CalledProcessError as e:
print(f"Error converting {doc_path} to text: {e}")
return ""
def extract_text_from_doc_or_docx(file):
if file.name.endswith(".docx"):
return extract_text_from_docx(file)
elif file.name.endswith(".doc"):
return convert_doc_to_text(file.name)
else:
return "Unsupported file type. Please upload a .doc or .docx file."
pdf_to_img = gr.Interface(
convert_pdf_to_image, gr.File(), gr.Gallery(), api_name="pdf_to_img"
)
pdf_to_text = gr.Interface(
extract_text_from_pdf,
gr.File(),
gr.Textbox(placeholder="Extracted text will appear here"),
api_name="pdf_to_text",
)
doc_or_docx_to_text = gr.Interface(
extract_text_from_doc_or_docx,
gr.File(),
gr.Textbox(placeholder="Extracted text from DOC or DOCX will appear here"),
api_name="doc_or_docx_to_text",
)
pptx_or_ppt_to_text = gr.Interface(
extract_text_from_ppt_or_pptx,
gr.File(),
gr.Textbox(placeholder="Extracted text from PPTX will appear here"),
api_name="pptx_or_ppt_to_text",
)
demo = gr.TabbedInterface(
[pdf_to_img, pdf_to_text, doc_or_docx_to_text, pptx_or_ppt_to_text],
["PDF to Image", "Extract PDF Text", "Extract DOC/DOCX Text", "Extract PPTX/PPT Text"],
)
demo.launch(server_name="0.0.0.0.", server_port=7860, debug=True)
|