Update app.py

app.py

@@ -10,6 +10,7 @@ def on_first_button_click():
 def ai_article_generator(
     text
 ):
+    prompt = f"Write a {text[1]} about the topic in about {text[2]} words: {text[0]}"
     ai_text = generate(f"Write an article about the topic: {text}")
     ai_text = ai_text.choices[0].message.content
     return ai_text
@@ -67,6 +68,14 @@ with gr.Blocks() as demo:
         with gr.Column(scale=0.7):
             gr.Markdown("## Enter a topic to write an article about:")
             input_topic = gr.Textbox(label="Topic")
+            with gr.Row():
+                gr.Markdown("## Choose type of text:")
+                input_type = gr.Dropdown(
+                    choices = ['article', 'essay', 'blog', 'report', 'letter'],
+                    value = 'article'
+                )
+                gr.Markdown("## Enter length of the text:")
+                input_length = gr.Number(value=200)
             ai_article_btn = gr.Button("Generate", visible=True)
             ai_generated = gr.Markdown("### Generated article:", visible=False)
             ai_label = gr.HTML(label="AI", visible=False)
@@ -128,7 +137,7 @@ with gr.Blocks() as demo:
         ai_article_btn.click(
             fn=ai_article_generator,
             inputs=[
-                input_topic
+                (input_topic, input_type, input_length)
             ],
             outputs=[ai_label],
         )