init commit

app.py

@@ -0,0 +1,73 @@
+import gradio as gr
+import stylecloud
+from PIL import Image
+import os
+from utils import data_to_text
+from reddit import RedditContent
+
+
+def create_stylecloud(subreddit, icon):
+    ## Get content
+    rdt_content = RedditContent(
+        thread_limit=10,
+        include_nsfw=False,
+        max_comment_length=2000,
+        min_comment_length=50,
+        min_comment_score=10,
+    )
+    rdt_content.process(search_query=subreddit, exact_search=True)
+
+    # Load the content
+    text = data_to_text(rdt_content.contents)
+
+    # Initiate params
+    output_file = 'stylecloud.png'
+    icon = icon_dict[icon]
+
+    # Generate the word cloud
+    stylecloud.gen_stylecloud(
+        text=text,
+        icon_name=icon,
+        size=500,
+        output_name=output_file,
+    )
+
+    # Load the generated image
+    image = Image.open(output_file)
+    image = image.resize((300, 300))  # Resize the image to 300x300 pixels
+
+    # Return the image
+    return image
+
+
+icon_dict = {
+    "Car" : "fas fa-car",
+    "Star and Crescent" : "fas fa-star-and-crescent",
+    "Trophy" : "fas fa-trophy",
+    "Heart" : "fas fa-heart",
+    "Flag" : "fas fa-flag"
+}
+
+
+with gr.Blocks() as demo:
+    gr.Markdown('Word Cloud for Reddit Comments')
+
+    with gr.Row():
+        subreddits = gr.Textbox(lines=1,
+                                label="Write subreddit or search query here",
+                                placeholder="Enter the subreddits you want to get content from")
+
+    with gr.Row():
+        icon = gr.Dropdown(choices=list(icon_dict.keys()),
+                           label='Select Icon', value="Heart")
+    with gr.Row():
+        create_button = gr.Button('Submit')
+        output_image = gr.Image(label='Word Cloud')
+
+    create_button.click(
+        create_stylecloud,
+        inputs=[subreddits, icon],
+        outputs=[output_image]
+    )
+
+demo.launch(share=True)