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)