年級：{grade}

app.py

@@ -8,13 +8,15 @@ import json
 OPEN_AI_KEY = os.getenv("OPEN_AI_KEY")
 OPEN_AI_CLIENT = OpenAI(api_key=OPEN_AI_KEY)
 
-def generate_article(reference_text, lesson_title, seed_words, derivative_words, prompt_text):    
+def generate_article(reference_text, lesson_title, seed_words, derivative_words, grade, word_count, prompt_text):    
     # 結合用戶輸入來形成最終的prompt
     reference_text = f"""
         文章參考：{reference_text}
         文章標題：{lesson_title}
         關鍵詞：{seed_words}
         衍生詞：{derivative_words}
+        年級：{grade}
+        文章字數限制：{word_count}
     """
 
 
@@ -48,6 +50,9 @@ with gr.Blocks() as demo:
         lesson_title = gr.Textbox(label="課程標題", placeholder="輸入課程標題...")
         seed_words = gr.Textbox(label="文章生字", placeholder="貼上文章生字...", lines=2)
         derivative_words = gr.Textbox(label="衍生生字", placeholder="貼上衍生生字...", lines=2)
+    with gr.Row():
+        grade = gr.Dropdown(label="年級", choices=["一年級", "二年級", "三年級", "四年級", "五年級", "六年級"], default="一年級")
+        word_count = gr.Textbox(label="文章字數", placeholder="輸入文章字數...", type="number")
         prompt_text = gr.Textbox(label="prompt的文字", placeholder="輸入或修改prompt的文字...")
         generate_article_btn = gr.Button(value="生成文章")
     
@@ -56,7 +61,7 @@ with gr.Blocks() as demo:
 
     generate_article_btn.click(
         generate_article, 
-        inputs =[reference_text, lesson_title, seed_words, derivative_words, prompt_text],
+        inputs =[reference_text, lesson_title, seed_words, derivative_words, grade, word_count, prompt_text],
         outputs=new_article
     )