waseoke commited on
Commit
08e8f76
·
verified ·
1 Parent(s): cddaebd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +31 -14
app.py CHANGED
@@ -78,7 +78,8 @@ def embed_user_data(user_data):
78
 
79
  # MongoDB Atlas에서 데이터 가져오기
80
  product_data = product_collection.find_one({"product_id": 1}) # 특정 상품 ID
81
- user_data = user_collection.find_one({'user_id': 2}) # 특정 사용자 ID
 
82
 
83
  # 상품 임베딩 수행
84
  if product_data:
@@ -95,17 +96,33 @@ if product_data:
95
  else:
96
  print("Product not found.")
97
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
98
  # 사용자 임베딩 수행
99
- if user_data:
100
- user_embedding = embed_user_data(user_data)
101
- print("User Embedding:", user_embedding)
102
-
103
- # MongoDB Atlas의 user_embeddings 컬렉션에 임베딩 저장
104
- user_embedding_collection.update_one(
105
- {"user_id": user_data["user_id"]}, # user_id 기준으로 찾기
106
- {"$set": {"embedding": user_embedding.tolist()}}, # 벡터를 리스트 형태로 저장
107
- upsert=True # 기존 항목이 없으면 새로 삽입
108
- )
109
- print("Embedding saved to MongoDB Atlas based on user_id.")
110
- else:
111
- print("User not found.")
 
 
78
 
79
  # MongoDB Atlas에서 데이터 가져오기
80
  product_data = product_collection.find_one({"product_id": 1}) # 특정 상품 ID
81
+ # user_data = user_collection.find_one({'user_id': 2}) # 특정 사용자 ID
82
+ all_users = user_collection.find() # 모든 사용자 데이터 가져오기
83
 
84
  # 상품 임베딩 수행
85
  if product_data:
 
96
  else:
97
  print("Product not found.")
98
 
99
+ # # 사용자 임베딩 수행
100
+ # if user_data:
101
+ # user_embedding = embed_user_data(user_data)
102
+ # print("User Embedding:", user_embedding)
103
+
104
+ # # MongoDB Atlas의 user_embeddings 컬렉션에 임베딩 저장
105
+ # user_embedding_collection.update_one(
106
+ # {"user_id": user_data["user_id"]}, # user_id 기준으로 찾기
107
+ # {"$set": {"embedding": user_embedding.tolist()}}, # 벡터를 리스트 형태로 저장
108
+ # upsert=True # 기존 항목이 없으면 새로 삽입
109
+ # )
110
+ # print("Embedding saved to MongoDB Atlas based on user_id.")
111
+ # else:
112
+ # print("User not found.")
113
+
114
  # 사용자 임베딩 수행
115
+ for user_data in all_users:
116
+ try:
117
+ user_embedding = embed_user_data(user_data)
118
+ print(f"User ID {user_data['user_id']} Embedding:", user_embedding)
119
+
120
+ # MongoDB Atlas의 user_embeddings 컬렉션에 임베딩 저장
121
+ user_embedding_collection.update_one(
122
+ {"user_id": user_data["user_id"]}, # user_id 기준으로 찾기
123
+ {"$set": {"embedding": user_embedding.tolist()}}, # 벡터를 리스트 형태로 저장
124
+ upsert=True # 기존 항목이 없으면 새로 삽입
125
+ )
126
+ print(f"Embedding saved to MongoDB Atlas for user_id {user_data['user_id']}.")
127
+ except ValueError as e:
128
+ print(f"Skipping user_id {user_data['user_id']} due to error: {e}")