init

app.py

@@ -0,0 +1,114 @@
+import gradio as gr
+import os
+import pandas as pd
+from openai import OpenAI
+
+# 設置 OpenAI API 客戶端
+openai_api_key = os.getenv("OPENAI_API_KEY")
+openai_client = OpenAI(api_key=openai_api_key)
+
+def extract_article_from_content(article_text):
+    start_markers = ["新文章：", "New Article:", "Here it is:"]
+    end_marker = "\nThank you"
+    for start_marker in start_markers:
+        start_index = article_text.find(start_marker)
+        if start_index != -1:
+            start_index += len(start_marker)
+            while article_text[start_index] in "\n":
+                start_index += 1
+            end_index = article_text.find(end_marker, start_index)
+            if end_index != -1:
+                return article_text[start_index:end_index].rstrip()
+            return article_text[start_index:].rstrip()
+    return article_text
+
+def validate_article(generated_article, lesson_words, base_chars, original_word_count):
+    clean_article = "".join(char for char in generated_article if char not in "，。！？；：「」『』（）《》【】'\n'")
+    not_every_new_word_is_used = not all(word in clean_article for word in lesson_words.split())
+    word_out_of_range = not set(clean_article).issubset(set(lesson_words + base_chars))
+    new_word_count = len(clean_article)
+    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):
+    attempt = 0
+    max_attempts = 3
+    generated_article = ""
+
+    while attempt < max_attempts:
+        attempt += 1
+        prompt = f"""
+            Please write a new and original Chinese article tailored for first-grade students. Here's a summary of the key points that you should follow:
+
+            Use Traditional Chinese Characters: The article should be written in Traditional Chinese, not Simplified Chinese.
+
+            Adherence to the Original Article: The new creation should closely follow the spirit, style, and rhythmic pattern of the provided original article. The number of words, excluding punctuation marks, should be similar to that of the original, approximately {original_word_count} words.
+
+            Incorporate "New Words": Every word listed under "new words" must be used in the article. These words are: {lesson_words}.
+
+            Utilize the "Word Library": Additional words required for the article can be selected from the provided "word library," which includes: {base_chars}.
+
+            Restriction on Vocabulary: Do not use any words outside the "new words" or the "word library".
+
+            Originality: The new article must be unique and original, not a copy of the original work.
+
+            "Original Article" for Reference: The example provided is {original_article}。This article serves as a model for the spirit, style, and rhythmic pattern to be emulated.
+        """
+        
+        response = openai_client.Completion.create(
+            model="text-davinci-003",
+            prompt=prompt,
+            max_tokens=1000
+        )
+
+        generated_text = response.choices[0].text.strip()
+        generated_article = extract_article_from_content(generated_text)
+
+        not_every_new_word_is_used, word_out_of_range, word_count_error = validate_article(
+            generated_article, lesson_words, base_chars, original_word_count)
+
+        if not not_every_new_word_is_used and not word_out_of_range and not word_count_error:
+            break
+
+    return generated_article
+
+def load_csv(file):
+    df = pd.read_csv(file)
+    if not df.empty:
+        first_row = df.iloc[0]
+        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]
+        )
+
+    demo.launch()
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+pandas
+openai>=1.0.0