model_1 = gr.Textbox(label="Model 1", value="gpt-4-0125-preview")

app.py

@@ -39,7 +39,7 @@ def validate_article(generated_article, lesson_words, base_chars, original_word_
     word_count_error = not (0.9 * original_word_count <= new_word_count <= 1.1 * original_word_count)
     return not_every_new_word_is_used, word_out_of_range, word_count_error
 
-def generate_new_article(lesson_words, original_article, original_word_count, base_chars):
+def generate_new_article(lesson_words, original_article, original_word_count, base_chars, model_name):
     attempt = 0
     max_attempts = 3
     generated_article = ""
@@ -70,7 +70,7 @@ def generate_new_article(lesson_words, original_article, original_word_count, ba
         """
         
         response = openai_client.chat.completions.create(
-            model="gpt-4-0125-preview",
+            model=model_name,
             messages=[
                 {"role": "system", "content": "You are a creative writer specialized in Chinese Children book. You will help me write Chinese Articles."},
                 {"role": "user", "content": prompt}
@@ -112,32 +112,47 @@ def load_csv(file):
     return "", "", 0, ""
 
 with gr.Blocks() as demo:
-    gr.Markdown("import CSV file to load data.")
-    csv_file_input = gr.File(label="Upload CSV file (Columns: lesson_words, original_article, original_word_count, base_chars)")
-    load_button = gr.Button("Load from CSV")
-
-    gr.Markdown("validate the article and generate new article.")
-    lesson_words_input = gr.Textbox(label="Lesson Words (以空格分隔)")
-    original_article_input = gr.Textbox(label="Original Article")
-    original_word_count_input = gr.Number(label="Original Word Count")
-    base_chars_input = gr.Textbox(label="Base Characters")
-    generate_button = gr.Button("Generate Article")
+    with gr.Row():
+        gr.Markdown("import CSV file to load data.")
+        csv_file_input = gr.File(label="Upload CSV file (Columns: lesson_words, original_article, original_word_count, base_chars)")
+        load_button = gr.Button("Load from CSV")
+    with gr.Row():
+        lesson_words_input = gr.Textbox(label="Lesson Words")
+        original_article_input = gr.Textbox(label="Original Article")
+        original_word_count_input = gr.Number(label="Original Word Count")
+        base_chars_input = gr.Textbox(label="Base Characters")
     
-    gr.Markdown("Generated Article")
-    output_text = gr.Textbox(label="Generated Article")
+    with gr.Row():    
+        model_1 = gr.Textbox(label="Model 1", value="gpt-4-0125-preview")
+        generate_button1 = gr.Button("Generate Article - gpt-4-0125-preview")
+        output_text1 = gr.Textbox(label="Generated Article - gpt-4-0125-preview")
 
-    generate_button.click(
+        model_2 = gr.Textbox(label="Model 2", value="gpt-3.5-turbo")
+        generate_button2 = gr.Button("Generate Article - gpt-3.5-turbo")
+        output_text2 = gr.Textbox(label="Generated Article - gpt-3.5-turbo")
+        # 為其他模型添加更多輸出文本框
+
+    generate_button1.click(
+        generate_new_article,
+        inputs=[lesson_words_input, original_article_input, original_word_count_input, base_chars_input, model_1],
+        outputs=output_text1
+    )
+
+    generate_button2.click(
         generate_new_article,
-        inputs=[lesson_words_input, original_article_input, original_word_count_input, base_chars_input],
-        outputs=output_text
+        inputs=[lesson_words_input, original_article_input, original_word_count_input, base_chars_input, model_2],
+        outputs=output_text2
     )
 
+    # 為其他模型添加點擊事件
     load_button.click(
         load_csv,
         inputs=[csv_file_input],
         outputs=[lesson_words_input, original_article_input, original_word_count_input, base_chars_input]
     )
 
+
 demo.launch()
 
 
+