Update app.py

app.py

@@ -71,6 +71,12 @@ class Order(Base):
     payment_reference = Column(String(100), nullable=True)
     last_location = Column(Text, nullable=True)  # Stored as a JSON string
 
+    # New fields for delivery details:
+    package_description = Column(Text, nullable=False)
+    pickup_address = Column(Text, nullable=False)
+    delivery_address = Column(Text, nullable=False)
+    contact_number = Column(String(20), nullable=False)
+
 class OrderTracking(Base):
     __tablename__ = "order_tracking"
     id = Column(Integer, primary_key=True, index=True)
@@ -80,6 +86,14 @@ class OrderTracking(Base):
     timestamp = Column(DateTime, default=datetime.utcnow)
     location = Column(Text, nullable=True)  # Stored as a JSON string
 
+    
+class DeliveryOrderRequest(BaseModel):
+    user_id: str
+    package_description: str
+    pickup_address: str
+    delivery_address: str
+    contact_number: str
+
 # --------------------
 # INITIALIZATION
 # --------------------
@@ -503,6 +517,29 @@ async def send_proactive_update(user_id: str, update_type: str):
     
     return response
 
+@app.post("/delivery")
+async def create_delivery_order(order_req: DeliveryOrderRequest):
+    async with async_session() as session:
+        try:
+            # Generate a unique order ID using UUID
+            order_id = str(uuid.uuid4())
+            new_order = Order(
+                order_id=order_id,
+                user_id=order_req.user_id,
+                status="pending",
+                package_description=order_req.package_description,
+                pickup_address=order_req.pickup_address,
+                delivery_address=order_req.delivery_address,
+                contact_number=order_req.contact_number,
+            )
+            session.add(new_order)
+            await session.commit()
+            await session.refresh(new_order)
+            return {"delivery_id": new_order.order_id}
+        except SQLAlchemyError as e:
+            await session.rollback()
+            raise HTTPException(status_code=500, detail="Error creating delivery order.") from e
+
 # --------------------
 # ERROR HANDLING
 # --------------------