スクレイピング機能の追加

app.py

@@ -2,12 +2,12 @@ from janome.tokenizer import Tokenizer
 from wordcloud import WordCloud
 import gradio as gr
 
-def generate_cloud(text, width, height):
-    
-    # default setting
-    width = width if width is not None else 1024
-    height = height if height is not None else 768
+from urllib.request import urlopen
+from bs4 import BeautifulSoup
+
 
+
+def generate_cloud(text, width, height):
     t = Tokenizer()
 
     tokens = t.tokenize(text)
@@ -24,13 +24,54 @@ def generate_cloud(text, width, height):
     return wc.to_array()
 
 
-gr.Interface(
-    fn=generate_cloud, 
+def generate_cloud_from_text(text, width, height):
+    
+    # default setting
+    width = width if width is not None else 1024
+    height = height if height is not None else 768
+
+    return generate_cloud(text, width, height)
+
+
+def generate_cloud_from_url(url, width, height):
+    
+    # URLからHTMLを取得
+    html = urlopen(url)
+
+    # HTMLからbodyのテキストを抽出
+    soup = BeautifulSoup(html, "html.parser")
+    body_text = soup.body.get_text()
+    # default setting
+    width = width if width is not None else 1024
+    height = height if height is not None else 768
+
+    return generate_cloud(body_text, width, height)
+
+
+from_text = gr.Interface(
+    fn=generate_cloud_from_text, 
     inputs=[
         gr.Textbox(label="入力テキスト"),
-        gr.Number(label="横幅（デフォルト値：1024）"),
-        gr.Number(label="高さ（デフォルト値：768）")
+        gr.Number(value=1024, label="横幅（デフォルト値：1024）"),
+        gr.Number(value=768, label="高さ（デフォルト値：768）")
+        ], 
+    outputs=gr.Image(type="pil"),
+    title="☁️にほんご　わーどくらうど☁"
+    )
+
+
+from_url = gr.Interface(
+    fn=generate_cloud_from_url, 
+    inputs=[
+        gr.Textbox(label="URL"),
+        gr.Number(value=1024, label="横幅（デフォルト値：1024）"),
+        gr.Number(value=768, label="高さ（デフォルト値：768）")
         ], 
     outputs=gr.Image(type="pil"),
     title="☁️にほんご　わーどくらうど☁"
-    ).launch()
+    )
+
+
+demo = gr.TabbedInterface([from_text, from_url], ["from_text", "from_url"])
+
+demo.launch()