Update app.py

app.py

@@ -8,11 +8,11 @@ def on_first_button_click():
     return gr.update(visible=True)
 
 def ai_article_generator(
-    topic, text_type, length, ai_model
+    topic, text_type, length, ai_model, api = None
 ):
     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_model)
+    ai_text = generate(f"Write an article about the topic: {text}", ai_model, api)
     ai_text = ai_text.choices[0].message.content
     return ai_text
 
@@ -156,7 +156,8 @@ with gr.Blocks() as demo:
                 input_topic, 
                 input_text_type, 
                 input_length,
-                ai_generator
+                ai_generator,
+                input_api
             ],
             outputs=[ai_label],
         )