Spaces:
Runtime error
Runtime error
import gradio as gr | |
import wikipedia | |
import numpy as np | |
import pandas as pd | |
from os import path | |
from PIL import Image | |
from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator | |
import matplotlib.pyplot as plt | |
def wikipediaScrap(article_name, wikipedia_language="en"): | |
if wikipedia_language: | |
wikipedia.set_lang(wikipedia_language) | |
et_page = wikipedia.page(article_name.replace(" ","")) | |
title = et_page.title | |
content = et_page.content | |
page_url = et_page.url | |
linked_pages = et_page.links | |
text = content | |
# Create and generate a word cloud image: | |
wordcloud = WordCloud(font_path="HelveticaWorld-Regular.ttf").generate(text) | |
# Display the generated image: | |
plt.imshow(wordcloud, interpolation='bilinear') | |
plt.axis("off") | |
return content | |
with gr.Blocks(css="footer {visibility: hidden} #dsd_button {background: purple, color: white}") as demo: | |
with gr.Row(): | |
inp = gr.Textbox(placeholder="Enter the name of wikipedia article", label="Wikipedia article name") | |
btn = gr.Button("Start Scraping", elem_id="dsd_button") | |
with gr.Row(): | |
content = gr.Textbox(label="Content") | |
with gr.Row(): | |
gr.Examples( | |
examples=[["Eiffel Tower"], ["Eiffel tower"]], fn=wikipediaScrap, inputs=[inp], | |
outputs=[content], cache_examples=True) | |
interface = gr.Interface(fn=wikipediaScrap, inputs=[inp], outputs=[content]) | |
interface.launch(inline=False) |