worksheet_exam_result_retrun_original.click(

app.py

@@ -1017,7 +1017,7 @@ def on_generate_ai_content(password, df_string, topic, grade, level, specific_fe
         "max_tokens": 4000  # 举例，实际上您可能需要更详细的配置
     }
     ai_content = material.send_ai_request(OPEN_AI_CLIENT, request_payload)
-    return ai_content, prompt, prompt
+    return ai_content, ai_content, prompt, prompt
 
 def generate_exam_fine_tune_result(password, exam_result_prompt , df_string_output, exam_result, exam_result_fine_tune_prompt):
     verify_password(password)
@@ -1033,6 +1033,10 @@ def generate_exam_fine_tune_result(password, exam_result_prompt , df_string_outp
     ai_content = material.send_ai_request(OPEN_AI_CLIENT, request_payload)
     return ai_content
 
+def return_original_exam_result(password, exam_result_original):
+    verify_password(password)
+    return exam_result_original
+
 def create_word(content):
     unique_filename = str(uuid.uuid4())
     word_file_path = f"/tmp/{unique_filename}.docx"
@@ -1680,11 +1684,11 @@ with gr.Blocks() as demo:
                         with gr.Column(scale=2):
                             # 生成對應不同模式的結果
                             worksheet_exam_result_prompt = gr.Textbox(visible=False)
+                            worksheet_exam_result_original = gr.Textbox(visible=False)
                             worksheet_exam_result = gr.Textbox(label="初次生成結果", show_copy_button=True, interactive=True)
                             worksheet_exam_result_fine_tune_prompt = gr.Textbox(label="根據結果，輸入你想更改的想法")
                             worksheet_exam_result_fine_tune_btn = gr.Button("微調結果")
-                            worksheet_exam_result_fine_result = gr.Textbox(label="微調結果",show_copy_button=True)
-
+                            worksheet_exam_result_retrun_original = gr.Button("返回原始結果")
                             worksheet_download_exam_result_button = gr.Button("下载结果")
                             worksheet_exam_result_word_link = gr.File(label="Download Word")
 
@@ -1699,11 +1703,11 @@ with gr.Blocks() as demo:
                         with gr.Column(scale=2):
                             # 生成對應不同模式的結果
                             lesson_plan_exam_result_prompt = gr.Textbox(visible=False)
+                            lesson_plan_exam_result_original = gr.Textbox(visible=False)
                             lesson_plan_exam_result = gr.Textbox(label="初次生成結果", show_copy_button=True, interactive=True)
                             lesson_plan_exam_result_fine_tune_prompt = gr.Textbox(label="根據結果，輸入你想更改的想法")
                             lesson_plan_exam_result_fine_tune_btn = gr.Button("微調結果")
-                            lesson_plan_exam_result_fine_result = gr.Textbox(label="微調結果",show_copy_button=True)
-
+                            lesson_plan_exam_result_retrun_original = gr.Button("返回原始結果")
                             lesson_plan_download_exam_result_button = gr.Button("下载结果")
                             lesson_plan_exam_result_word_link = gr.File(label="Download Word")
                 
@@ -1718,11 +1722,11 @@ with gr.Blocks() as demo:
                         with gr.Column(scale=2):
                             # 生成對應不同模式的結果
                             exit_ticket_exam_result_prompt = gr.Textbox(visible=False)
+                            exit_ticket_exam_result_original = gr.Textbox(visible=False)
                             exit_ticket_exam_result = gr.Textbox(label="初次生成結果", show_copy_button=True, interactive=True)
                             exit_ticket_exam_result_fine_tune_prompt = gr.Textbox(label="根據結果，輸入你想更改的想法")
                             exit_ticket_exam_result_fine_tune_btn = gr.Button("微調結果")
-                            exit_ticket_exam_result_fine_result = gr.Textbox(label="微調結果",show_copy_button=True)
-
+                            exit_ticket_exam_result_retrun_original = gr.Button("返回原始結果")
                             exit_ticket_download_exam_result_button = gr.Button("下载结果")
                             exit_ticket_exam_result_word_link = gr.File(label="Download Word")
 
@@ -1843,30 +1847,66 @@ with gr.Blocks() as demo:
     worksheet_content_btn.click(
         on_generate_ai_content,
         inputs=[password, df_string_output, content_topic, content_grade, content_level, worksheet_algorithm, worksheet_content_type_name],
-        outputs=[worksheet_exam_result, worksheet_prompt, worksheet_exam_result_prompt]
+        outputs=[worksheet_exam_result_original, worksheet_exam_result, worksheet_prompt, worksheet_exam_result_prompt]
     )
     lesson_plan_btn.click(
         on_generate_ai_content,
         inputs=[password, df_string_output, content_topic, content_grade, content_level, lesson_plan_time, lesson_plan_content_type_name],
-        outputs=[lesson_plan_exam_result, lesson_plan_prompt, lesson_plan_exam_result_prompt]
+        outputs=[lesson_plan_exam_result_original, lesson_plan_exam_result, lesson_plan_prompt, lesson_plan_exam_result_prompt]
     )
     exit_ticket_btn.click(
         on_generate_ai_content,
         inputs=[password, df_string_output, content_topic, content_grade, content_level, exit_ticket_time, exit_ticket_content_type_name],
-        outputs=[exit_ticket_exam_result, exit_ticket_prompt, exit_ticket_exam_result_prompt]
+        outputs=[exit_ticket_exam_result_original, exit_ticket_exam_result, exit_ticket_prompt, exit_ticket_exam_result_prompt]
     )
 
     # 生成結果微調
     worksheet_exam_result_fine_tune_btn.click(
         generate_exam_fine_tune_result,
         inputs=[password, worksheet_exam_result_prompt, df_string_output, worksheet_exam_result, worksheet_exam_result_fine_tune_prompt],
-        outputs=[worksheet_exam_result_fine_result]
+        outputs=[worksheet_exam_result]
     )
     worksheet_download_exam_result_button.click(
         download_exam_result,
         inputs=[worksheet_exam_result],
         outputs=[worksheet_exam_result_word_link]
     )
+    worksheet_exam_result_retrun_original.click(
+        return_original_exam_result,
+        inputs=[worksheet_exam_result_original],
+        outputs=[worksheet_exam_result]
+    )
+    lesson_plan_exam_result_fine_tune_btn.click(
+        generate_exam_fine_tune_result,
+        inputs=[password, lesson_plan_exam_result_prompt, df_string_output, lesson_plan_exam_result, lesson_plan_exam_result_fine_tune_prompt],
+        outputs=[lesson_plan_exam_result]
+    )
+    lesson_plan_download_exam_result_button.click(
+        download_exam_result,
+        inputs=[lesson_plan_exam_result],
+        outputs=[lesson_plan_exam_result_word_link]
+    )
+    lesson_plan_exam_result_retrun_original.click(
+        return_original_exam_result,
+        inputs=[lesson_plan_exam_result_original],
+        outputs=[lesson_plan_exam_result]
+    )
+    exit_ticket_exam_result_fine_tune_btn.click(
+        generate_exam_fine_tune_result,
+        inputs=[password, exit_ticket_exam_result_prompt, df_string_output, exit_ticket_exam_result, exit_ticket_exam_result_fine_tune_prompt],
+        outputs=[exit_ticket_exam_result]
+    )
+    exit_ticket_download_exam_result_button.click(
+        download_exam_result,
+        inputs=[exit_ticket_exam_result],
+        outputs=[exit_ticket_exam_result_word_link]
+    )
+    exit_ticket_exam_result_retrun_original.click(
+        return_original_exam_result,
+        inputs=[exit_ticket_exam_result_original],
+        outputs=[exit_ticket_exam_result]
+    )
+
 
 
     demo.load(