Update embed_data.py

embed_data.py

@@ -68,7 +68,7 @@ def initialize_pinecone():
 def load_ebikes_data(file_path="ebikes_data.json"):
     with open(file_path, 'r') as f:
         data = json.load(f)
-    return data.get('ebikes', [])
+    return data.get('pogo-cycles-data', [])
 
 # Create embeddings and upload to Pinecone
 def create_and_upload_embeddings(ebikes_data, encoder, pinecone_index):
@@ -81,10 +81,11 @@ def create_and_upload_embeddings(ebikes_data, encoder, pinecone_index):
         ids.append(bike['id'])
         descriptions.append(bike['description'])
         metadata.append({
-            'id': bike['id'],
-            'name': bike['name'],
-            'type': bike['type'],
-            'description': bike['description'][:100] + "..."  # Adding truncated description to metadata
+            "id": bike["id"],
+            "name": bike["name"],
+            "product_type": bike["type"],       # or "escooter"
+            "category": bike["category"],        # mountain / folding / cargo ...
+            "description": bike["description"]
         })
     
     # Create embeddings