Update app.py

app.py

@@ -8,11 +8,11 @@ def on_first_button_click():
     return gr.update(visible=True)
 
 def ai_article_generator(
-    topic, type, length
+    topic, text_type, length, ai_model
 ):
-    prompt = f"Write a {type} about the topic in about {length} words: {topic}"
+    prompt = f"Write a {text_type} about the topic in about {length} words: {topic}"
     print(prompt)
-    ai_text = generate(f"Write an article about the topic: {text}")
+    ai_text = generate(f"Write an article about the topic: {text}", ai_model)
     ai_text = ai_text.choices[0].message.content
     return ai_text
 
@@ -77,10 +77,10 @@ with gr.Blocks() as demo:
             input_topic = gr.Textbox(label="Topic")
             with gr.Row():
                 with gr.Column():
-                    input_type = gr.Dropdown(
+                    input_text_type = gr.Dropdown(
                         choices = ['article', 'essay', 'blog', 'report', 'letter'],
                         value = 'article',
-                        label="Choose type of text:"
+                        label="Choose text_type of text:"
                     )
                 with gr.Column():
                     input_length = gr.Number(value=200, label="Choose length of the text:")
@@ -154,8 +154,9 @@ with gr.Blocks() as demo:
             fn=ai_article_generator,
             inputs=[
                 input_topic, 
-                input_type, 
-                input_length
+                input_text_type, 
+                input_length,
+                ai_generator
             ],
             outputs=[ai_label],
         )