PDF_Translate / app.py
MUSCAT41's picture
Create app.py
5e695ae
raw
history blame
748 Bytes
import gradio as gr
from textblob import TextBlob
import PyPDF2
def extract_text_from_pdf(file_path):
pdf_file_obj = open(file_path, 'rb')
pdf_reader = PyPDF2.PdfFileReader(pdf_file_obj)
text = ''
for page_num in range(pdf_reader.numPages):
page_obj = pdf_reader.getPage(page_num)
text += page_obj.extractText()
pdf_file_obj.close()
return text
def translate_text(text, dest_lang='en'):
blob = TextBlob(text)
translation = blob.translate(to=dest_lang)
return str(translation)
def translate_pdf(file):
text = extract_text_from_pdf(file.name)
translation = translate_text(text)
return translation
iface = gr.Interface(fn=translate_pdf, inputs='file', outputs='text')
iface.launch()