update

app.py

@@ -6,6 +6,8 @@ from docx import Document
 import os
 from openai import OpenAI
 from groq import Groq
+from docx import Document
+from gtts import gTTS
 
 from youtube_transcript_api import YouTubeTranscriptApi
 from youtube_transcript_api._errors import NoTranscriptFound
@@ -713,6 +715,12 @@ def generate_reading_passage(df_string):
 
     return reading_passage
 
+def text_to_speech(video_id, text):
+    tts = gTTS(text, lang='en')
+    filename = f'{video_id}_reading_passage.mp3'
+    tts.save(filename)
+    return filename
+
 def get_mind_map(video_id, df_string, source):
     if source == "gcs":
         print("===get_mind_map on gcs===")
@@ -1025,6 +1033,18 @@ 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 create_word(content):
+    unique_filename = str(uuid.uuid4())
+    word_file_path = f"/tmp/{unique_filename}.docx"
+    doc = Document()
+    doc.add_paragraph(content)
+    doc.save(word_file_path)
+    return word_file_path
+
+def download_exam_result(content):
+    word_path = create_word(content)
+    return word_path
+
 # ---- Chatbot ----
 def respond(password, user_message, data, chat_history, socratic_mode=False):
     verify_password(password)
@@ -1296,7 +1316,7 @@ def chat_with_groq(password, user_message, data, chat_history, socratic_mode=Fal
     # 返回聊天历史和空字符串清空输入框
     return "", chat_history
 
-def chat_with_youtube_transcript(password, youtube_id, thread_id, trascript, user_message, chat_history, socratic_mode=False):
+def chat_with_opan_ai_assistant(password, youtube_id, thread_id, trascript, user_message, chat_history, socratic_mode=False):
     verify_password(password)
 
     # 先計算 user_message 是否超過 500 個字
@@ -1304,6 +1324,11 @@ def chat_with_youtube_transcript(password, youtube_id, thread_id, trascript, use
         error_msg = "你的訊息太長了，請縮短訊息長度至五百字以內"
         raise gr.Error(error_msg)
     
+    # 如果 chat_history 超過 10 則訊息，直接 return "對話超過上限"
+    if chat_history is not None and len(chat_history) > 10:
+        error_msg = "此次對話超過上限"
+        raise gr.Error(error_msg)
+
     try:
         assistant_id = "asst_kmvZLNkDUYaNkMNtZEAYxyPq"
         client = OPEN_AI_CLIENT
@@ -1346,18 +1371,24 @@ def chat_with_youtube_transcript(password, youtube_id, thread_id, trascript, use
             你是一個擅長資料分析跟影片教學的老師，user 為學生
             如果是影片類型，不用解釋逐字稿格式，直接回答學生問題
             socratic_mode = {socratic_mode}
-            如果 socratic_mode = True，
+            if socratic_mode is True，
             - 請用蘇格拉底式的提問方式，引導學生思考，並且給予學生一些提示
+            - 一次只問一個問題，字數在100字以內
             - 不要直接給予答案，讓學生自己思考
             - 但可以給予一些提示跟引導，例如給予影片的時間軸，讓學生自己去找答案
             - 在你回答的開頭標註【蘇格拉底助教：{youtube_id} 】
-            如果 socratic_mode = False，
-            - 直接回答學生問題
+            
+            if socratic_mode is False，
+            - 直接回答學生問題，字數在100字以內
             - 在你回答的開頭標註【一般學習精靈：{youtube_id} 】
+
             如果學生問了一些問題你無法判斷，請告訴學生你無法判斷，並建議學生可以問其他問題
-            或者你可以反問學生一些問題，幫助學生更好的理解資料
-            如果學生的問題與資料文本無關，請告訴學生你無法回答超出範圍的問題
-            最後只要是參考逐字稿資料，請在回答的最後標註【參考資料：（分）：（秒）】
+            或者你可以反問學生一些問題，幫助學生更好的理解資料，字數在100字以內
+            如果學生的問題與資料文本無關，請告訴學生你「無法回答超出範圍的問題」，並告訴他可以怎麼問什麼樣的問題（一個就好）
+            只要是參考逐字稿資料，請在回答的最後標註【參考資料：（分）：（秒）】
+            請一定要用繁體中文回答 zh-TW，並用台灣人的口語表達
+            回答範圍一定要在逐字稿資料內，不要引用其他資料，請嚴格執行
+            並在重複問句後給予學生鼓勵，讓學生有學習的動力
         """
 
         # 创建线程
@@ -1371,7 +1402,7 @@ def chat_with_youtube_transcript(password, youtube_id, thread_id, trascript, use
         client.beta.threads.messages.create(
             thread_id=thread.id,
             role="user",
-            content=user_message
+            content=user_message + "(請一定要用繁體中文回答 zh-TW，並用台灣人的口語表達，回答時不要特別說明這是台灣人的語氣)"
         )
 
         # 运行助手，生成响应
@@ -1580,6 +1611,8 @@ with gr.Blocks() as demo:
             with gr.Column(scale=3):
                 with gr.Tab("文章模式"):
                     reading_passage = gr.Textbox(label="Reading Passage", lines=40)
+                    reading_passage_speak_button = gr.Button("Speak", visible=False)
+                    reading_passage_audio_output = gr.Audio(label="Audio Output", visible=False)
                 with gr.Tab("重點"):
                     df_summarise = gr.Textbox(container=True, show_copy_button=True, lines=40)    
                 with gr.Tab("問題"):
@@ -1675,33 +1708,17 @@ with gr.Blocks() as demo:
             with gr.Column(scale=2):
                 # 生成對應不同模式的結果
                 exam_result_prompt = gr.Textbox(visible=False)
-                exam_result = gr.Textbox(label="初次生成結果", show_copy_button=True)
+                exam_result = gr.Textbox(label="初次生成結果", show_copy_button=True, interactive=True)
                 exam_result_fine_tune_prompt = gr.Textbox(label="根據結果，輸入你想更改的想法")
                 exam_result_fine_tune_btn = gr.Button("微調結果")
                 exam_result_fine_result = gr.Textbox(label="微調結果",show_copy_button=True)
- 
-    # 傳統模式
-    # send_button.click(
-    #     respond, 
-    #     inputs=[msg, df_string_output, chatbot, socratic_mode_btn], 
-    #     outputs=[msg, chatbot]
-    # )
-    # # 连接按钮点击事件
-    # btn_1.click(respond, inputs=[btn_1, df_string_output, chatbot, socratic_mode_btn], outputs=[msg, chatbot])
-    # btn_2.click(respond, inputs=[btn_2, df_string_output, chatbot, socratic_mode_btn], outputs=[msg, chatbot])
-    # btn_3.click(respond, inputs=[btn_3, df_string_output, chatbot, socratic_mode_btn], outputs=[msg, chatbot])
-
-    # chat_with_youtube_transcript
-    # send_button.click(
-    #     chat_with_youtube_transcript,
-    #     inputs=[password, video_id, thread_id, df_string_output, msg, chatbot, socratic_mode_btn],
-    #     outputs=[msg, chatbot, thread_id]
-    # )
-    # chat_with_youtube_transcript
-                
+
+                download_exam_result_button = gr.Button("下载结果")
+                exam_result_word_link = gr.File(label="Download Word")
+             
     # OPENAI 模式
     send_button.click(
-        chat_with_youtube_transcript,
+        chat_with_opan_ai_assistant,
         inputs=[password, video_id, thread_id, df_string_output, msg, chatbot, socratic_mode_btn],
         outputs=[msg, chatbot, thread_id]
     )
@@ -1710,8 +1727,6 @@ with gr.Blocks() as demo:
         inputs=[password, openai_chatbot_audio_input],
         outputs=[msg]
     )
-
-
     # GROQ 模式
     groq_send_button.click(
         chat_with_groq,
@@ -1725,20 +1740,19 @@ with gr.Blocks() as demo:
         outputs=[jutor_msg, jutor_chatbot]
     )
 
-
     # 连接按钮点击事件
     btn_1.click(
-        chat_with_youtube_transcript,
+        chat_with_opan_ai_assistant,
         inputs=[password, video_id, thread_id, df_string_output, btn_1, chatbot, socratic_mode_btn],
         outputs=[msg, chatbot, thread_id]
     )
     btn_2.click(
-        chat_with_youtube_transcript,
+        chat_with_opan_ai_assistant,
         inputs=[password, video_id, thread_id, df_string_output, btn_2, chatbot, socratic_mode_btn],
         outputs=[msg, chatbot, thread_id]
     )
     btn_3.click(
-        chat_with_youtube_transcript,
+        chat_with_opan_ai_assistant,
         inputs=[password, video_id, thread_id, df_string_output, btn_3, chatbot, socratic_mode_btn],
         outputs=[msg, chatbot, thread_id]
     )
@@ -1792,6 +1806,13 @@ with gr.Blocks() as demo:
     # 当输入网页链接时触发
     # web_link.change(process_web_link, inputs=web_link, outputs=[btn_1, btn_2, btn_3, df_summarise, df_string_output])
 
+    # 文章模式
+    reading_passage_speak_button.click(
+        text_to_speech, 
+        inputs=[video_id, reading_passage], 
+        outputs=reading_passage_audio_output
+    )
+
     # 教師版 學習單
     worksheet_content_btn.click(
         on_generate_ai_content,
@@ -1815,6 +1836,12 @@ with gr.Blocks() as demo:
         inputs=[password, exam_result_prompt, df_string_output, exam_result, exam_result_fine_tune_prompt],
         outputs=[exam_result_fine_result]
     )
+    download_exam_result_button.click(
+        download_exam_result,
+        inputs=[exam_result],
+        outputs=[exam_result_word_link]
+    )
+
 
     demo.load(
         init_params,