Update app.py

app.py

@@ -94,6 +94,21 @@ class OrderTracking(Base):
     message = Column(Text, nullable=True)
     timestamp = Column(DateTime, default=datetime.utcnow)
 
+class MenuItem(Base):
+    __tablename__ = "menu_items"
+    id = Column(Integer, primary_key=True, index=True)
+    name = Column(String, unique=True, index=True)
+    description = Column(Text)
+    price = Column(Integer)  # Price as an integer (in Naira)
+    nutrition = Column(Text)
+
+class TownShippingCost(Base):
+    __tablename__ = "town_shipping_costs"
+    id = Column(Integer, primary_key=True, index=True)
+    town = Column(String, unique=True, index=True)
+    cost = Column(Integer)
+
+
 engine = create_async_engine(DATABASE_URL, echo=True)
 async_session = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
 
@@ -167,6 +182,37 @@ class ConversationState:
 
 SESSION_TIMEOUT = timedelta(minutes=5)
 
+async def populate_menu_items():
+    async with async_session() as session:
+        for item in menu_items:
+            # Check if this menu item already exists to avoid duplicates.
+            result = await session.execute(select(MenuItem).where(MenuItem.name == item["name"]))
+            existing = result.scalars().first()
+            if not existing:
+                new_item = MenuItem(
+                    name=item["name"],
+                    description=item["description"],
+                    price=item["price"],
+                    nutrition=item["nutrition"]
+                )
+                session.add(new_item)
+        await session.commit()
+
+async def populate_shipping_costs():
+    async with async_session() as session:
+        for town, cost in TOWN_SHIPPING_COSTS.items():
+            # Check if the shipping cost entry for this town already exists.
+            result = await session.execute(select(TownShippingCost).where(TownShippingCost.town == town))
+            existing = result.scalars().first()
+            if not existing:
+                new_cost = TownShippingCost(
+                    town=town,
+                    cost=cost
+                )
+                session.add(new_cost)
+        await session.commit()
+
+
 async def log_chat_to_db(user_id: str, direction: str, message: str):
     async with async_session() as session:
         entry = ChatHistory(user_id=user_id, direction=direction, message=message)
@@ -337,6 +383,22 @@ def is_order_intent(message: str) -> bool:
             return True
     return False
 
+async def get_menu_items():
+    async with async_session() as session:
+        result = await session.execute(select(MenuItem))
+        items = result.scalars().all()
+        # Build a list of dictionaries to send as a response.
+        return [
+            {
+                "name": item.name,
+                "description": item.description,
+                "price": item.price,
+                "nutrition": item.nutrition
+            }
+            for item in items
+        ]
+
+
 async def track_order(user_id: str, order_id: str) -> str:
     async with async_session() as session:
         order_result = await session.execute(
@@ -445,6 +507,29 @@ def get_dish_price(dish: str) -> int:
             return item["price"]
     return 0  # or raise an error if dish not found
 
+# Fetch dish price from the MenuItem table
+async def get_dish_price_two(dish: str) -> int:
+    async with async_session() as session:
+        # Using ilike() for case-insensitive matching
+        result = await session.execute(select(MenuItem).where(MenuItem.name.ilike(dish)))
+        menu_item = result.scalars().first()
+        return menu_item.price if menu_item else 0
+
+# Fetch shipping cost from the TownShippingCost table based on the address.
+async def get_shipping_cost(address: str) -> int:
+    async with async_session() as session:
+        result = await session.execute(select(TownShippingCost))
+        costs = result.scalars().all()
+        address_lower = address.lower()
+        # Look for a matching town in the address.
+        for cost in costs:
+            if cost.town in address_lower:
+                return cost.cost
+        # Fallback: return the default cost.
+        default_cost = next((c for c in costs if c.town == "default"), None)
+        return default_cost.cost if default_cost else 1000
+
+
 def send_email_notification(order_details):
     # Construct the email payload. Adjust values as needed.
     payload = {
@@ -845,6 +930,8 @@ app = FastAPI()
 @app.on_event("startup")
 async def on_startup():
     await init_db()
+    await populate_menu_items()
+    await populate_shipping_costs()
 
 @app.post("/chatbot")
 async def chatbot_response(request: Request, background_tasks: BackgroundTasks):