Update app.py

app.py

@@ -93,7 +93,7 @@ vectorstore = FAISS.load_local(
     embeddings=embedding_model,
     allow_dangerous_deserialization=True
 )
-retriever = vectorstore.as_retriever(search_kwargs={"k": 4})
+retriever = vectorstore.as_retriever(search_kwargs={"k": 8})
 print("✅ RAG 系統初始化完成。")
 
 
@@ -219,11 +219,16 @@ def math_solver_process(user_question):
         return error_msg, "Error", "Error", "Error", "Error"
 
 
-# =========================================================
+    # =========================================================
 # 4. Gradio 介面配置 (作為測試介面，確保能跑)
 # =========================================================
 
-# 6. Gradio 介面
+# 定義 LaTeX 渲染模式（統一所有 Markdown 元件）
+LATEX_DELIMITERS = [
+    { "left": "$$", "right": "$$", "display": True },
+    { "left": "$", "right": "$", "display": False }
+]
+
 with gr.Blocks(title="實變函數論 AI Agent") as demo:
     gr.Markdown("# 📐 實變函數論 AI 助教 (Hugging Face Spaces RAG + CoT)")
     gr.Markdown("輸入你的數學問題，AI 將查詢課本、多角度思考、並經由同儕審查機制產出嚴謹證明。")
@@ -232,31 +237,47 @@ with gr.Blocks(title="實變函數論 AI Agent") as demo:
         input_box = gr.Textbox(label="請輸入問題", placeholder="例如：請解釋 Lebesgue 積分與 Riemann 積分在極限交換時的差異")
         run_btn = gr.Button("開始解題", variant="primary")
 
-    with gr.Accordion("📚 步驟一：檢索資料 (RAG)", open=False):
-        out_rag = gr.TextArea(label="RAG Context", interactive=False)
+    # 步驟一：檢索資料 (RAG) - 使用 Markdown 渲染 LaTeX
+    with gr.Accordion("📚 步驟一：檢索資料 (RAG Context)", open=False):
+        out_rag = gr.Markdown(
+            label="RAG Context", 
+            latex_delimiters=LATEX_DELIMITERS
+        )
 
-    with gr.Accordion("🧠 步驟二：思考與選擇 (CoT)", open=False):
-        out_thoughts = gr.TextArea(label="5 種解題思路", interactive=False)
+    # 步驟二：思考與選擇 (CoT) - 使用 Markdown 渲染 LaTeX
+    with gr.Accordion("🧠 步驟二：思考與選擇 (CoT Planner)", open=False):
+        out_thoughts = gr.Markdown(
+            label="5 種解題思路", 
+            latex_delimiters=LATEX_DELIMITERS
+        )
 
-    with gr.Row():
-        out_draft = gr.TextArea(label="📝 初稿 (Writer)", interactive=False)
-        out_review = gr.TextArea(label="🧐 審查意見 (Reviewer)", interactive=False)
+    # 步驟三、四：初稿與審查 - 包裹在一個新的可收合區塊
+    with gr.Accordion("📝 步驟三/四：初稿與審查 (Writer / Reviewer)", open=False):
+        with gr.Row():
+            # 初稿 (Writer) - 使用 Markdown 渲染 LaTeX
+            out_draft = gr.Markdown(
+                label="初稿 (Writer)", 
+                latex_delimiters=LATEX_DELIMITERS
+            )
+            # 審查意見 (Reviewer) - 使用 Markdown 渲染 LaTeX
+            out_review = gr.Markdown(
+                label="審查意見 (Reviewer)", 
+                latex_delimiters=LATEX_DELIMITERS
+            )
 
     gr.Markdown("### ✨ 最終解答 (Final Answer)")
 
+    # 最終解答 (Final Answer) - 維持不變，確保在最顯眼的位置
     out_final = gr.Markdown(
         label="最終解答",
-        latex_delimiters=[
-            { "left": "$$", "right": "$$", "display": True },
-            { "left": "$", "right": "$", "display": False }
-        ]
+        latex_delimiters=LATEX_DELIMITERS
     )
 
-    # 按鈕事件綁定
+    # 按鈕事件綁定 (注意：math_solver_process 必須回傳 5 個輸出)
     run_btn.click(
         math_solver_process,
         inputs=[input_box],
-        outputs=[out_rag, out_thoughts, out_draft, out_review, out_final]
+        outputs=[out_rag, out_thoughts, out_draft, out_review, out_final] # 輸出數量與函數回傳匹配
     )
 
     # 在 Hugging Face 上執行時，只需呼叫 demo.launch()