Update main.py

main.py

@@ -5,8 +5,8 @@ from datasets import load_dataset
 app = FastAPI()
 
 # Load the datasets from the Hugging Face Hub
-customers_dataset = load_dataset("dwb2023/customers")["train"]
-orders_dataset = load_dataset("dwb2023/orders")["train"]
+customers_dataset = load_dataset("dwb2023/blackbird-customers", split="train")
+orders_dataset = load_dataset("dwb2023/blackbird-orders", split="train")
 
 class GetUserInput(BaseModel):
     key: str
@@ -43,7 +43,7 @@ def update_user(input: UpdateUserInput):
     global customers_dataset
     customers_data = customers_dataset.to_list()
     user_found = False
-    
+
     for customer in customers_data:
         if customer["id"] == input.user_id:
             if input.email:
@@ -52,10 +52,10 @@ def update_user(input: UpdateUserInput):
                 customer["phone"] = input.phone
             user_found = True
             break
-    
+
     if user_found:
         updated_dataset = Dataset.from_list(customers_data)
-        updated_dataset.push_to_hub("dwb2023/customers", split="train")
+        updated_dataset.push_to_hub("dwb2023/blackbird-customers", split="train")
         customers_dataset = updated_dataset  # Update the global dataset
         return {"message": "User information updated successfully"}
     else:
@@ -79,7 +79,7 @@ def cancel_order(input: CancelOrderInput):
     global orders_dataset
     orders_data = orders_dataset.to_list()
     order_found = False
-    
+
     for order in orders_data:
         if order["id"] == input.order_id:
             if order["status"] == "Processing":
@@ -88,10 +88,10 @@ def cancel_order(input: CancelOrderInput):
                 break
             else:
                 raise HTTPException(status_code=400, detail="Order has already shipped. Can't cancel it.")
-    
+
     if order_found:
         updated_orders_dataset = Dataset.from_list(orders_data)
-        updated_orders_dataset.push_to_hub("dwb2023/orders", split="train")
+        updated_orders_dataset.push_to_hub("dwb2023/blackbird-orders", split="train")
         orders_dataset = updated_orders_dataset  # Update the global dataset
         return {"status": "Cancelled"}
     else: