Update app.py

app.py

@@ -77,36 +77,31 @@ custom_css = """
 .vertical-radio label { width: 100% !important; margin: 0 !important; display: flex !important; }
 """
 
-# 設定圖片路徑 (包含新增的 Q3, Q4, Q5)
+# 設定圖片路徑
 img_path_1 = "images/image_0.png"
 img_path_2 = "images/image_1.png"
 img_path_3 = "images/image_2.png" # Q3
-img_path_4 = "images/image_3.png" # Q4
-img_path_5 = "images/image_4.png" # Q5
+img_path_4 = "images/image_4.png" # Q4
+img_path_5 = "images/image_3.png" # Q5
 
 if not os.path.exists("images"): os.makedirs("images")
-# 建立預設圖片以防報錯
-for p in [img_path_1, img_path_2, img_path_3, img_path_4, img_path_5]:
+for p in [img_path_0, img_path_1, img_path_2, img_path_4, img_path_3]:
     if not os.path.exists(p): 
         Image.new('RGB', (400, 300), color='lightgray').save(p)
 
-# --- Q1 選項 ---
+# --- 選項設定 ---
 q1_options_map = {
     "不​要​再​問白癡​問題​了​我​只​想​吃飯​": "現實主義者 (只想吃飯)",
     "去​過隱​居​生活": "隱士 (嚮往平靜)",
     "加入​禁衛軍，​保衛​這個​世界": "守護者 (充滿正義感)",
     "神​羅​天​征，毀滅這個​世界​": "破壞神 (心情可能很差或很中二)"
 }
-
-# --- Q2 選項 ---
 q2_options_map = {
     "不​要​再​問白癡​問題​了​我​只​想​吃飯​": "無魔法 (飢餓度MAX)",
     "麵​包形狀​的​魔法​炸彈": "爆炸魔法 (喜歡刺激/重口味)",
     "毒氣​的​生化​魔法": "毒氣魔法 (可能想吃臭豆腐或特殊風味)",
     "領域​展開:​無量​空處​": "領域展開 (思緒混亂或想放空)"
 }
-
-# --- [新增] Q3 選項 ---
 q3_options_map = {
     "不​要​在​問白癡​問題​了​我​只​想​吃飯​": "現實主義者 (只想吃飯)",
     "歐洲": "嚮往歐洲 (浪漫/西式)",
@@ -114,16 +109,12 @@ q3_options_map = {
     "泰國": "嚮往泰國 (熱情/酸辣)",
     "轉身搭機捷回家": "戀家 (只想回家)"
 }
-
-# --- [新增] Q4 選項 ---
 q4_options_map = {
     "轉頭​回家​睡覺​": "獨行俠 (想睡覺)",
     "看​起來​很​會玩​的​帥潮​": "外向 (找帥潮)",
     "感覺​是​動漫宅​的​同好​": "御宅族 (找同好)",
     "有​興趣​的​異性​": "大膽 (找異性)"
 }
-
-# --- [新增] Q5 選項 ---
 q5_options_map = {
     "甚麼​都​不​想​​": "放空 (什麼都不想)",
     "來​自​星星​的​你​": "韓劇迷 (浪漫愛情)",
@@ -173,14 +164,14 @@ def get_restaurant_data(mood_score_str, food_choice):
         
     return result, is_random, rec_categories, mood_reason
 
-def generate_content_with_groq(restaurant_name, restaurant_detail, user_diary, mood_score, quiz_result, mood_guide_reason):
+# [修改] 增加 debug_mode 參數
+def generate_content_with_groq(restaurant_name, restaurant_detail, user_diary, mood_score, quiz_result, mood_guide_reason, debug_mode=False):
     if not GROQ_API_KEY: return "⚠️ 請設定 GROQ_API_KEY"
     
     client = OpenAI(api_key=GROQ_API_KEY, base_url="https://api.groq.com/openai/v1")
     
     system_prompt = "你是一個幽默、懂吃且善解人意的 AI 朋友。請根據使用者的日記、心情以及「心情美食指南」來推薦餐廳。"
     
-    # [修改] 將新增的測驗結果加入 Prompt
     user_msg = f"""
     【狀態】心情分數：{mood_score}，日記：{user_diary}
     
@@ -200,16 +191,29 @@ def generate_content_with_groq(restaurant_name, restaurant_detail, user_diary, m
     
     任務：
     請用繁體中文寫一段溫暖有趣的回覆：
-    1. 綜合回應他的日記與上述 5 個測驗結果（例如如果他選日本，可以提到這家店有沒有日式氛圍；如果選動漫宅，可以用一點動漫梗）。
+    1. 綜合回應他的日記與上述 5 個測驗結果。
     2. 引用「心情美食指南」的原因。
     3. 介紹這家餐廳的特色。
     (只需要回覆文字內容)
     """
+    
     try:
         response = client.chat.completions.create(
             model="llama-3.3-70b-versatile", messages=[{"role": "system", "content": system_prompt}, {"role": "user", "content": user_msg}]
         )
-        return response.choices[0].message.content
+        content = response.choices[0].message.content
+        
+        # [修改] 如果開啟除錯模式，附加 Prompt 資訊
+        if debug_mode:
+            debug_info = f"""
+            \n\n--- 🛠️ [DEBUG] LLM Prompt 檢查 ---
+            \n**System Prompt:**\n{system_prompt}
+            \n**User Message:**\n{user_msg}
+            \n-----------------------------------
+            """
+            return content + debug_info
+            
+        return content
     except Exception as e:
         return f"Groq Error: {str(e)}"
 
@@ -224,7 +228,8 @@ def generate_image_huggingface(prompt):
         print(f"❌ 生圖失敗: {e}")
         return None
 
-def mood_agent_logic(score_input, food_input, diary_input, quiz_state):
+# [修改] 增加 debug_mode 參數
+def mood_agent_logic(score_input, food_input, diary_input, quiz_state, debug_mode):
     restaurant, is_random, rec_categories, mood_reason = get_restaurant_data(score_input, food_input)
     
     if restaurant is None:
@@ -244,8 +249,19 @@ def mood_agent_logic(score_input, food_input, diary_input, quiz_state):
         if docs:
             rag_info = "\n".join([d.page_content for d in docs])
 
-    ai_text = generate_content_with_groq(name, rag_info, diary_input, score_input, quiz_state, mood_reason)
+    # [修改] 傳入 debug_mode
+    ai_text = generate_content_with_groq(name, rag_info, diary_input, score_input, quiz_state, mood_reason, debug_mode)
     
+    # [修改] 如果開啟除錯模式，附加圖片 Prompt 資訊
+    if debug_mode:
+        img_debug_info = f"""
+        \n\n--- 🛠️ [DEBUG] 圖片生成檢查 ---
+        \n**使用的 Visual Prompt:**\n{img_prompt}
+        \n(如果上方圖片為空白，可能是 HF API 忙碌或 Prompt 無效)
+        \n-----------------------------------
+        """
+        ai_text += img_debug_info
+
     final_response = f"### 🍽️ 推薦：{name} {note}\n\n{ai_text}"
     map_html = f'<div style="text-align:center"><a href="{url}" target="_blank" style="background:#4CAF50;color:white;padding:8px 16px;border-radius:20px;text-decoration:none">🗺️ Google Map 導航</a></div>'
     
@@ -264,21 +280,18 @@ def handle_q1_change(selected_label, current_state):
 def handle_q2_change(selected_label, current_state):
     if not selected_label: return current_state, gr.Button(interactive=False)
     current_state["q2"] = q2_options_map[selected_label]
-    return current_state, gr.Button(interactive=True, variant="primary") # 修改：這裡不再是完成，而是下一頁
+    return current_state, gr.Button(interactive=True, variant="primary")
 
-# [新增] Q3 Handler
 def handle_q3_change(selected_label, current_state):
     if not selected_label: return current_state, gr.Button(interactive=False)
     current_state["q3"] = q3_options_map[selected_label]
     return current_state, gr.Button(interactive=True, variant="primary")
 
-# [新增] Q4 Handler
 def handle_q4_change(selected_label, current_state):
     if not selected_label: return current_state, gr.Button(interactive=False)
     current_state["q4"] = q4_options_map[selected_label]
     return current_state, gr.Button(interactive=True, variant="primary")
 
-# [新增] Q5 Handler (最後一題)
 def handle_q5_change(selected_label, current_state):
     if not selected_label: return current_state, gr.Button(interactive=False)
     current_state["q5"] = q5_options_map[selected_label]
@@ -308,9 +321,9 @@ with gr.Blocks(title="AI 心情食堂") as demo:
                 gr.Image(value=img_path_2, type="filepath", label="請觀察圖片", height=300)
                 gr.Markdown("**問題：身為魔導士的你，會選擇哪一個法術？**")
                 radio_q2 = gr.Radio(choices=list(q2_options_map.keys()), label="請選擇", elem_classes="vertical-radio")
-                btn_q2_next = gr.Button("下一頁 ➔", interactive=False) # 修改按鈕名稱
+                btn_q2_next = gr.Button("下一頁 ➔", interactive=False)
 
-        # [新增] Tab 2: Q3
+        # Tab 2: Q3
         with gr.TabItem("Q3", id=2):
             with gr.Column():
                 gr.Markdown("### 🔮 第三題：旅遊直覺")
@@ -319,7 +332,7 @@ with gr.Blocks(title="AI 心情食堂") as demo:
                 radio_q3 = gr.Radio(choices=list(q3_options_map.keys()), label="請選擇", elem_classes="vertical-radio")
                 btn_q3_next = gr.Button("下一頁 ➔", interactive=False)
 
-        # [新增] Tab 3: Q4
+        # Tab 3: Q4
         with gr.TabItem("Q4", id=3):
             with gr.Column():
                 gr.Markdown("### 🔮 第四題：社交場合")
@@ -328,7 +341,7 @@ with gr.Blocks(title="AI 心情食堂") as demo:
                 radio_q4 = gr.Radio(choices=list(q4_options_map.keys()), label="請選擇", elem_classes="vertical-radio")
                 btn_q4_next = gr.Button("下一頁 ➔", interactive=False)
 
-        # [新增] Tab 4: Q5
+        # Tab 4: Q5
         with gr.TabItem("Q5", id=4):
             with gr.Column():
                 gr.Markdown("### 🔮 第五題：追劇時光")
@@ -346,6 +359,10 @@ with gr.Blocks(title="AI 心情食堂") as demo:
                         score_input = gr.Radio(["1 (心情差)", "2 (不太好)", "3 (普通)", "4 (不錯)", "5 (超棒)"], label="1. 心情分數", value="3 (普通)")
                         food_input = gr.Radio(["吃飯", "吃麵", "隨便"], label="2. 想吃什麼", value="隨便")
                         diary_input = gr.Textbox(lines=4, label="3. 心情日記", placeholder="寫下今天發生的事...")
+                        
+                        # [新增] 除錯模式開關
+                        debug_mode_btn = gr.Checkbox(label="🔧 開啟除錯模式 (顯示 Prompt 與圖片資訊)", value=False)
+                        
                         submit_btn = gr.Button("送出給 Agent", variant="primary")
                     with gr.Column(scale=1):
                         agent_output = gr.Markdown(label="AI 回應")
@@ -353,29 +370,25 @@ with gr.Blocks(title="AI 心情食堂") as demo:
                         map_output = gr.HTML(label="地圖導航")
 
     # 事件綁定 (Events)
-    # Q1
     radio_q1.change(fn=handle_q1_change, inputs=[radio_q1, quiz_state], outputs=[quiz_state, btn_q1_next])
-    btn_q1_next.click(fn=lambda: gr.Tabs(selected=1), outputs=tabs) # 跳到 Q2
+    btn_q1_next.click(fn=lambda: gr.Tabs(selected=1), outputs=tabs)
     
-    # Q2
     radio_q2.change(fn=handle_q2_change, inputs=[radio_q2, quiz_state], outputs=[quiz_state, btn_q2_next])
-    btn_q2_next.click(fn=lambda: gr.Tabs(selected=2), outputs=tabs) # 跳到 Q3
+    btn_q2_next.click(fn=lambda: gr.Tabs(selected=2), outputs=tabs)
 
-    # [新增] Q3
     radio_q3.change(fn=handle_q3_change, inputs=[radio_q3, quiz_state], outputs=[quiz_state, btn_q3_next])
-    btn_q3_next.click(fn=lambda: gr.Tabs(selected=3), outputs=tabs) # 跳到 Q4
+    btn_q3_next.click(fn=lambda: gr.Tabs(selected=3), outputs=tabs)
 
-    # [新增] Q4
     radio_q4.change(fn=handle_q4_change, inputs=[radio_q4, quiz_state], outputs=[quiz_state, btn_q4_next])
-    btn_q4_next.click(fn=lambda: gr.Tabs(selected=4), outputs=tabs) # 跳到 Q5
+    btn_q4_next.click(fn=lambda: gr.Tabs(selected=4), outputs=tabs)
 
-    # [新增] Q5
     radio_q5.change(fn=handle_q5_change, inputs=[radio_q5, quiz_state], outputs=[quiz_state, btn_q5_finish])
-    btn_q5_finish.click(fn=lambda: gr.Tabs(selected=5), outputs=tabs) # 跳到 Main
+    btn_q5_finish.click(fn=lambda: gr.Tabs(selected=5), outputs=tabs)
 
+    # [修改] 加入 debug_mode_btn 到輸入
     submit_btn.click(
         fn=mood_agent_logic,
-        inputs=[score_input, food_input, diary_input, quiz_state],
+        inputs=[score_input, food_input, diary_input, quiz_state, debug_mode_btn],
         outputs=[agent_output, image_output, map_output]
     )