๊ฐ•๋ฏผ๊ท  commited on
Commit
c95db27
ยท
1 Parent(s): e18fae2

Fix: Absolute path for data loading and add import os

Browse files
Files changed (1) hide show
  1. backend/src/logic.py +10 -5
backend/src/logic.py CHANGED
@@ -8,6 +8,7 @@ import pickle
8
  from datetime import datetime, timedelta, timezone
9
  from supabase import create_client
10
  import re
 
11
  from collections import Counter
12
 
13
  # ==========================================
@@ -243,18 +244,22 @@ def load_resources():
243
  global method_map, recipes_by_ingredient, ing_method_counts, ing_cat_counts, total_method_counts, total_cat_counts, TOTAL_RECIPES
244
 
245
  print("Loading resources... (This may take a while)")
 
 
 
 
246
  # mmap='r' ์˜ต์…˜์œผ๋กœ ๋ฉ”๋ชจ๋ฆฌ ์‚ฌ์šฉ๋Ÿ‰ ์ตœ์†Œํ™” (๋””์Šคํฌ์—์„œ ์ง์ ‘ ์ฝ์Œ)
247
- w2v_model = Word2Vec.load("models/w2v.model", mmap='r')
248
- d2v_model = Doc2Vec.load("models/d2v.model", mmap='r')
249
- df_temp = pd.read_csv("data/final_recipe_data.csv")
250
  df_temp['์žฌ๋ฃŒํ† ํฐ'] = df_temp['์žฌ๋ฃŒํ† ํฐ'].apply(literal_eval)
251
  df = df_temp # Assign to global
252
 
253
- with open("data/stats.pkl", "rb") as f:
254
  stats = pickle.load(f)
255
 
256
  try:
257
- price_df = pd.read_csv("data/price_rank.csv", encoding='utf-8-sig')
258
  price_df.columns = price_df.columns.str.strip()
259
  price_map = dict(zip(price_df['ingredient'], price_df['rank']))
260
  except:
 
8
  from datetime import datetime, timedelta, timezone
9
  from supabase import create_client
10
  import re
11
+ import os
12
  from collections import Counter
13
 
14
  # ==========================================
 
244
  global method_map, recipes_by_ingredient, ing_method_counts, ing_cat_counts, total_method_counts, total_cat_counts, TOTAL_RECIPES
245
 
246
  print("Loading resources... (This may take a while)")
247
+
248
+ # ๊ธฐ์ค€ ๊ฒฝ๋กœ ์„ค์ • (ํ˜„์žฌ ํŒŒ์ผ ์œ„์น˜ logic.py ๊ธฐ์ค€ ์ƒ์œ„ ํด๋”)
249
+ base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
250
+
251
  # mmap='r' ์˜ต์…˜์œผ๋กœ ๋ฉ”๋ชจ๋ฆฌ ์‚ฌ์šฉ๋Ÿ‰ ์ตœ์†Œํ™” (๋””์Šคํฌ์—์„œ ์ง์ ‘ ์ฝ์Œ)
252
+ w2v_model = Word2Vec.load(os.path.join(base_dir, "models/w2v.model"), mmap='r')
253
+ d2v_model = Doc2Vec.load(os.path.join(base_dir, "models/d2v.model"), mmap='r')
254
+ df_temp = pd.read_csv(os.path.join(base_dir, "data/final_recipe_data.csv"))
255
  df_temp['์žฌ๋ฃŒํ† ํฐ'] = df_temp['์žฌ๋ฃŒํ† ํฐ'].apply(literal_eval)
256
  df = df_temp # Assign to global
257
 
258
+ with open(os.path.join(base_dir, "data/stats.pkl"), "rb") as f:
259
  stats = pickle.load(f)
260
 
261
  try:
262
+ price_df = pd.read_csv(os.path.join(base_dir, "data/price_rank.csv"), encoding='utf-8-sig')
263
  price_df.columns = price_df.columns.str.strip()
264
  price_map = dict(zip(price_df['ingredient'], price_df['rank']))
265
  except: