tubifier / app.py
wldmr's picture
app file
837fdb6
raw history blame
No virus
1.32 kB
import gradio as gr
from PIL import Image
import os
import summarizer as su
import nltk
def image_mod(rpunkt_switch, link):
if len(link)==0:
return 'Error: No link provided', None
nltk_file = 'nltk_data/tokenizers/punkt.zip'
home_pc = '/Users/hujo/'
home_hf = '/home/user/'
if os.path.exists(home_pc+nltk_file) or os.path.exists(home_hf+nltk_file):
print('nltk punkt file exists in ', nltk_file)
else:
nltk.download('punkt')
#link = 'https://www.youtube.com/watch?v=lCnHfTHkhbE'
lexrank_switch = True
html = ''
images = []
html, images = su.getSummary(link, lexrank_switch, rpunkt_switch)
#images = su.getSummaryImage(link, lexrank_switch, rpunkt_switch)
print(html)
files = os.listdir('workdir/')
print('local files: ',files)
#image_path = 'workdir/lion.jpg'
#im = Image.open(image_path)
#images.append(im)
#with Image.open(open(image_path,'rb')) as im:
# images.append(im)
#images.append(im.rotate(90))
#images[0].save("newlion.png")
print('images',images)
return html, images
demo = gr.Interface(image_mod,
[gr.Checkbox(label='Restore runctuation'), "text"] , ["html", gr.Gallery()],
allow_flagging="never")
if __name__ == "__main__":
demo.launch()