gr.Markdown("### Original Lesson CSV File")

app.py

@@ -137,7 +137,7 @@ def generate_new_article(lesson_words, original_article, original_word_count, ba
 
     return generated_article, validate_article_result
 
-def load_csv(file):
+def load_lesson_csv(file):
     try:
         df = pd.read_csv(file, encoding='utf-8')
     except:
@@ -153,9 +153,26 @@ def load_csv(file):
         return lesson_words, original_article, original_word_count, base_chars
     return "", "", 0, ""
 
+def load_base_chars_csv(file):
+    try:
+        df = pd.read_csv(file, encoding='utf-8')
+    except:
+        df = pd.read_csv(file.name, encoding='utf-8')
+
+    if not df.empty:
+        first_row = df.iloc[0]
+        base_chars = first_row['words']
+        return base_chars
+    return ""
+
 with gr.Blocks() as demo:
-    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():
+        with gr.Column():
+            gr.Markdown("### Original Lesson CSV File")
+            lesson_csv_file_input = gr.File(label="Upload CSV file (Columns: lesson_words, original_article, original_word_count, base_chars)")
+        with gr.Column():
+            gr.Markdown("### Base Characters CSV File")
+            base_chars_csv_file_input = gr.File(label="Upload Base Characters File")
     
     with gr.Row():
         lesson_words_input = gr.Textbox(label="Lesson Words")
@@ -207,19 +224,17 @@ with gr.Blocks() as demo:
     )
 
     # 為其他模型添加點擊事件
-    csv_file_input.change(
-        load_csv,
-        inputs=[csv_file_input],
+    lesson_csv_file_input.change(
+        load_lesson_csv,
+        inputs=[lesson_csv_file_input],
         outputs=[lesson_words_input, original_article_input, original_word_count_input, base_chars_input]
     )
-
-    load_button.click(
-        load_csv,
-        inputs=[csv_file_input],
-        outputs=[lesson_words_input, original_article_input, original_word_count_input, base_chars_input]
+    base_chars_csv_file_input.change(
+        load_base_chars_csv,
+        inputs=[base_chars_csv_file_input],
+        outputs=[base_chars_input]
     )
 
-
 demo.launch()