Update app.py

app.py

@@ -63,6 +63,12 @@ def ai_check(text, option):
     else :
         return ai_generated_test(text)
 
+def update_visibility_api(model):
+    if model in ['OpenAI GPT 3.5', 'OpenAI GPT 4']:
+        return gr.update(visible=True)
+    else:
+        return gr.update(visible=False)
+
 with gr.Blocks() as demo:
     gr.Markdown("# Polygraf Article Writer")
     with gr.Row():
@@ -78,6 +84,14 @@ with gr.Blocks() as demo:
                     )
                 with gr.Column():
                     input_length = gr.Number(value=200, label="Choose length of the text:")
+            with gr.Row():
+                ai_generator = gr.Dropdown(
+                    choices = ['GPT-Neo', 'Llama 3', 'OpenAI GPT 3.5', 'OpenAI GPT 4'],
+                    value = 'GPT-Neo',
+                    label="Choose AI Model"
+                )
+                input_api = gr.Textbox(label="API Key", visible=False)
+                ai_generator.change(update_visibility, ai_generator, input_api)
             ai_article_btn = gr.Button("Generate", visible=True)
             ai_generated = gr.Markdown("### Generated article:", visible=False)
             ai_label = gr.HTML(label="AI", visible=False)