Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
100 |
-
|
101 |
-
|
102 |
-
|
103 |
-
|
104 |
-
|
105 |
-
|
106 |
-
|
107 |
-
|
108 |
-
|
109 |
-
|
110 |
-
|
111 |
-
|
|
|
|
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}")
|