init

app.py

@@ -79,36 +79,35 @@ def load_csv(file):
         return first_row['lesson_words'], first_row['original_article'], first_row['original_word_count'], first_row['base_chars']
     return "", "", 0, ""
 
-def main():
-    with gr.Blocks() as demo:
-        gr.Markdown("Generate New Article")
-
-        with gr.Tab("Manual Input"):
-            lesson_words_input = gr.Textbox(label="Lesson Words (以空格分隔)")
-            original_article_input = gr.Textarea(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.Tab("CSV Input"):
-            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")
-
-        output_text = gr.Textbox(label="Generated Article")
-
-        generate_button.click(
-            generate_new_article,
-            inputs=[lesson_words_input, original_article_input, original_word_count_input, base_chars_input],
-            outputs=output_text
-        )
 
-        load_button.click(
-            load_csv,
-            inputs=[csv_file_input],
-            outputs=[lesson_words_input, original_article_input, original_word_count_input, base_chars_input]
-        )
+with gr.Blocks() as demo:
+    gr.Markdown("Generate New Article")
+
+    with gr.Tab("Manual Input"):
+        lesson_words_input = gr.Textbox(label="Lesson Words (以空格分隔)")
+        original_article_input = gr.Textarea(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.Tab("CSV Input"):
+        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")
+
+    output_text = gr.Textbox(label="Generated Article")
+
+    generate_button.click(
+        generate_new_article,
+        inputs=[lesson_words_input, original_article_input, original_word_count_input, base_chars_input],
+        outputs=output_text
+    )
+
+    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()
 
-    demo.launch()
 
-if __name__ == "__main__":
-    main()