AamirAli123's picture
Upload project files
e31d3f6
"""
main module blog generator.
"""
import gradio as gr
from module.generator import (generate_blog)
###################### Front-end ########################
with gr.Blocks(theme = gr.themes.Monochrome()) as demo:
gr.Markdown("""
<h1><center>Writer.ai</center></h1>
<h3><center>Blog and essay writer ai tool</center></h3>
""")
with gr.Column():
output_block = gr.Markdown(line_breaks = True)
with gr.Row():
topic = gr.Text(label = "Topic",
placeholder = "essay or blog topic")
content_type = gr.Text(label = "Type",
placeholder = "essay, blog post")
tone = gr.Text(label = "Tone",
placeholder = "soft, professional")
link = gr.Text(label = "link",
placeholder = "links with comma seperated")
length = gr.Text(label = "Length",
placeholder = "length of generated contents")
with gr.Row():
stop_button = gr.Button("Stop")
resume_button = gr.Button("Resume")
start_button = gr.Button("Start")
start_button.click(fn = generate_blog, inputs = [topic,
content_type, link,
tone, length],
outputs = output_block)
demo.queue(max_size = 1)
demo.launch(debug = True)