Update app.py

app.py

@@ -120,7 +120,12 @@ def recommend_products_for_user(user_id, top_n=1):
             print(f"User ID {user_id} embedding not found.")
             return []
 
-        user_embedding = np.array(user_embedding_data["embedding"]).reshape(1, -1)
+        # user_embedding = np.array(user_embedding_data["embedding"]).reshape(1, -1)
+        user_embedding = np.array(user_embedding_data["embedding"])
+
+        # 차원 확인 및 조정
+        if user_embedding.ndim == 1:  # 1D 배열인 경우 2D로 변환
+            user_embedding = user_embedding.reshape(1, -1)
 
         # 모든 상품 임베딩 가져오기
         all_product_embeddings = list(product_embedding_collection.find())