Update app.py

app.py

@@ -21,6 +21,20 @@ output_dir = "./question_bank"
 # 載入題庫字典(question_bank_dict)，格式為 { "年級_學期": [題庫檔名列表] }
 question_bank_dict = json.load(open(f"{output_dir}/question_bank_dict.json", "r"))
 
+import re
+
+def remove_think_tags(content):
+    """刪除 <think>...</think> 標籤及其內容的正規表達式解法"""
+    # pattern = r'<think>.*?</think>'
+    # cleaned_content = re.sub(pattern, '', content, flags=re.DOTALL)
+
+    # 刪除所有 <think> 標籤及其內容
+    content = re.sub(r'<\s*think[^>]*>.*?<\s*/\s*think\s*>', '', content, flags=re.DOTALL)
+    # 清除殘留空行 (連續兩個以上換行符)
+    content = re.sub(r'\n{3,}', '\n\n', content)
+    
+    return cleaned_content.strip()
+
 def random_questions_with_limit(data, limit=20000):
     """
     隨機從 data 中挑選題目，並將總字串長度限制在 limit 字元數內（至少不小於 5000）。
@@ -147,7 +161,7 @@ def generate_math_questions(grade, term, qtype="Unspecified", num_questions=10):
     )
 
     response = completion.choices[0].message.content
-
+    response = remove_think_tags(response)
     print(response)
 
     # 傳回模型產生的文本