Update app.py

app.py

@@ -22,10 +22,11 @@ warnings.filterwarnings("ignore")
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-# --- 1. 資料讀取 (主要資料庫) ---
-df = pd.DataFrame() 
+# --- 1. 資料讀取 (主要資料庫 - 使用 restaurants.csv) ---
+df = pd.DataFrame()
 try:
-    df = pd.read_csv('google_maps_rag_data.csv', engine='python')
+    # 修改：讀取 restaurants.csv
+    df = pd.read_csv('restaurants.csv', engine='python')
     df.columns = df.columns.str.strip() # 清理欄位空白
 
     def classify_mood(row):
@@ -48,29 +49,33 @@ try:
         
     if 'Mood_Tags' not in df.columns:
         df['Mood_Tags'] = df.apply(classify_mood, axis=1)
+    
+    print("✅ 成功讀取 restaurants.csv 並建立 Mood_Tags")
 except Exception as e:
-    print(f"⚠️ 讀取 google_maps_rag_data.csv 失敗: {e}")
+    print(f"⚠️ 讀取 restaurants.csv 失敗: {e}")
 
-# --- 2. 資料讀取 (Prompt 資料庫) ---
+# --- 2. 資料讀取 (Prompt 資料庫 - 使用 restaurants.csv) ---
 df_prompts = pd.DataFrame()
-PROMPT_COL_NAME = 'Image_Prompt' 
+# 修改：您的 CSV 中 Prompt 欄位名稱為 Visual_prompt
+PROMPT_COL_NAME = 'Visual_prompt' 
 
 try:
-    df_prompts = pd.read_csv('restaurants_prompts_v6.csv', engine='python')
-    df_prompts.columns = df_prompts.columns.str.strip()
-    
-    if 'Name' in df_prompts.columns:
-        df_prompts.set_index('Name', inplace=True)
-    else:
-        print(f"⚠️ Prompts CSV 警告: 找不到 'Name' 欄位")
+    # 直接使用已讀取的 df 來建立 Prompt 索引
+    if not df.empty:
+        df_prompts = df.copy()
+        
+        if 'Name' in df_prompts.columns:
+            df_prompts.set_index('Name', inplace=True)
+        else:
+            print(f"⚠️ Prompts 處理警告: 找不到 'Name' 欄位")
 
-    if PROMPT_COL_NAME not in df_prompts.columns:
-        print(f"⚠️ Prompts CSV 警告: 找不到 '{PROMPT_COL_NAME}' 欄位")
+        if PROMPT_COL_NAME not in df_prompts.columns:
+            print(f"⚠️ Prompts 處理警告: 找不到 '{PROMPT_COL_NAME}' 欄位")
+    else:
+        print("⚠️ 主資料庫為空，無法建立 Prompt 索引")
 
-except FileNotFoundError:
-    print("⚠️ 找不到 restaurants_prompts_v6.csv")
 except Exception as e:
-    print(f"⚠️ 讀取 Prompt CSV 發生錯誤: {e}")
+    print(f"⚠️ 處理 Prompt 資料發生錯誤: {e}")
 
 
 # --- RAG 建置 ---
@@ -227,7 +232,8 @@ def mood_agent_logic(score_input, food_input, diary_input, quiz_state, debug_mod
                 
                 if pd.notna(csv_prompt) and str(csv_prompt).strip() != "":
                     img_prompt = str(csv_prompt)
-                    prompt_source = "✅ CSV 檔案 (restaurants_prompts_v6.csv)"
+                    # 修改：更新來源顯示為 restaurants.csv
+                    prompt_source = "✅ CSV 檔案 (restaurants.csv)"
                 else:
                     prompt_source = "⚠️ CSV 有此餐廳但 Prompt 空白"
             except Exception as e: