Update app.py

app.py

@@ -78,7 +78,8 @@ def embed_user_data(user_data):
 
 # MongoDB Atlas에서 데이터 가져오기
 product_data = product_collection.find_one({"product_id": 1})  # 특정 상품 ID
-user_data = user_collection.find_one({'user_id': 2})  # 특정 사용자 ID
+# user_data = user_collection.find_one({'user_id': 2})  # 특정 사용자 ID
+all_users = user_collection.find()  # 모든 사용자 데이터 가져오기
 
 # 상품 임베딩 수행
 if product_data:
@@ -95,17 +96,33 @@ if product_data:
 else:
     print("Product not found.")
 
+# # 사용자 임베딩 수행
+# if user_data:
+#     user_embedding = embed_user_data(user_data)
+#     print("User Embedding:", user_embedding)
+
+#     # MongoDB Atlas의 user_embeddings 컬렉션에 임베딩 저장
+#     user_embedding_collection.update_one(
+#         {"user_id": user_data["user_id"]},  # user_id 기준으로 찾기
+#         {"$set": {"embedding": user_embedding.tolist()}},  # 벡터를 리스트 형태로 저장
+#         upsert=True  # 기존 항목이 없으면 새로 삽입
+#     )
+#     print("Embedding saved to MongoDB Atlas based on user_id.")
+# else:
+#     print("User not found.")
+
 # 사용자 임베딩 수행
-if user_data:
-    user_embedding = embed_user_data(user_data)
-    print("User Embedding:", user_embedding)
-
-    # MongoDB Atlas의 user_embeddings 컬렉션에 임베딩 저장
-    user_embedding_collection.update_one(
-        {"user_id": user_data["user_id"]},  # user_id 기준으로 찾기
-        {"$set": {"embedding": user_embedding.tolist()}},  # 벡터를 리스트 형태로 저장
-        upsert=True  # 기존 항목이 없으면 새로 삽입
-    )
-    print("Embedding saved to MongoDB Atlas based on user_id.")
-else:
-    print("User not found.")
+for user_data in all_users:
+    try:
+        user_embedding = embed_user_data(user_data)
+        print(f"User ID {user_data['user_id']} Embedding:", user_embedding)
+
+        # MongoDB Atlas의 user_embeddings 컬렉션에 임베딩 저장
+        user_embedding_collection.update_one(
+            {"user_id": user_data["user_id"]},  # user_id 기준으로 찾기
+            {"$set": {"embedding": user_embedding.tolist()}},  # 벡터를 리스트 형태로 저장
+            upsert=True  # 기존 항목이 없으면 새로 삽입
+        )
+        print(f"Embedding saved to MongoDB Atlas for user_id {user_data['user_id']}.")
+    except ValueError as e:
+        print(f"Skipping user_id {user_data['user_id']} due to error: {e}")