Update app.py

app.py

@@ -46,11 +46,12 @@ if product_data:
     product_embedding = embed_product_data(product_data)
     print("Product Embedding:", product_embedding)
 
-    # product_id 기준으로 임베딩을 MongoDB에 저장
-    product_collection.update_one(
-        {"product_id": "1"},
-        {"$set": {"embedding": product_embedding.tolist()}}
+    # MongoDB Atlas의 product_embeddings 컬렉션에 임베딩 저장
+    embedding_collection.update_one(
+        {"product_id": product_data["product_id"]},  # product_id 기준으로 찾기
+        {"$set": {"embedding": product_embedding.tolist()}},  # 벡터를 리스트 형태로 저장
+        upsert=True  # 기존 항목이 없으면 새로 삽입
     )
-    print("Embedding saved to MongoDB based on product_id.")
+    print("Embedding saved to MongoDB Atlas based on product_id.")
 else:
     print("Product not found.")