Update app.py

app.py

@@ -21,9 +21,10 @@ from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
 from sqlalchemy.orm import sessionmaker, declarative_base
 from sqlalchemy import Column, Integer, String, DateTime, Text, Float
 
+# --- Environment Variables and API Keys ---
 SPOONACULAR_API_KEY = os.getenv("SPOONACULAR_API_KEY", "default_fallback_value")
 PAYSTACK_SECRET_KEY = os.getenv("PAYSTACK_SECRET_KEY", "default_fallback_value")
-DATABASE_URL = os.getenv("DATABASE_URL", "default_fallback_value")
+DATABASE_URL = os.getenv("DATABASE_URL", "sqlite+aiosqlite:///./test.db")  # Example using SQLite
 NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY", "default_fallback_value")
 openai.api_key = os.getenv("OPENAI_API_KEY", "default_fallback_value")
 
@@ -74,6 +75,16 @@ class SentimentLog(Base):
     sentiment_score = Column(Float)
     message = Column(Text)
 
+# --- New Model: Order Tracking ---
+class OrderTracking(Base):
+    __tablename__ = "order_tracking"
+    id = Column(Integer, primary_key=True, index=True)
+    order_id = Column(String, index=True)
+    status = Column(String)  # e.g., "Order Placed", "Payment Confirmed", etc.
+    message = Column(Text, nullable=True)  # Optional additional details
+    timestamp = Column(DateTime, default=datetime.utcnow)
+
+# --- Create Engine and Session ---
 engine = create_async_engine(DATABASE_URL, echo=True)
 async_session = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
 
@@ -232,6 +243,17 @@ def stream_image_completion(image_b64: str):
         if line:
             yield line.decode("utf-8") + "\n"
 
+# --- Helper Function for Order Tracking ---
+async def log_order_tracking(order_id: str, status: str, message: str = None):
+    async with async_session() as session:
+        tracking_entry = OrderTracking(
+            order_id=order_id,
+            status=status,
+            message=message
+        )
+        session.add(tracking_entry)
+        await session.commit()
+
 # --- Advanced Internal Flow: Order Processing & Payment Integration ---
 def process_order_flow(user_id: str, message: str) -> str:
     """
@@ -384,6 +406,9 @@ def process_order_flow(user_id: str, message: str) -> str:
                         await session.commit()
                 asyncio.create_task(save_order())
 
+                # Record the initial tracking update: Order Placed
+                asyncio.create_task(log_order_tracking(order_id, "Order Placed", "Order placed and awaiting payment."))
+
                 # Notify management of the new order via WhatsApp
                 async def notify_management_order(order_details: dict):
                     message_body = (
@@ -692,6 +717,8 @@ async def payment_callback(request: Request):
                 await session.commit()
             else:
                 raise HTTPException(status_code=404, detail="Order not found.")
+        # Record payment confirmation tracking update
+        await log_order_tracking(order_id, "Payment Confirmed", f"Payment status updated to {status}.")
         # Notify management via WhatsApp about the payment update
         await asyncio.to_thread(send_whatsapp_message, MANAGEMENT_WHATSAPP_NUMBER,
             f"Payment Update:\nOrder ID: {order_id} is now {status}."
@@ -714,6 +741,7 @@ async def payment_callback(request: Request):
             if order:
                 order.status = new_status
                 await session.commit()
+                await log_order_tracking(order_id, "Payment Confirmed", f"Payment status updated to {new_status}.")
                 await asyncio.to_thread(send_whatsapp_message, MANAGEMENT_WHATSAPP_NUMBER,
                     f"Payment Update:\nOrder ID: {order_id} is now {new_status}."
                 )
@@ -721,6 +749,21 @@ async def payment_callback(request: Request):
             else:
                 raise HTTPException(status_code=404, detail="Order not found.")
 
+# --- New Endpoint: Order Tracking ---
+@app.get("/track_order/{order_id}")
+async def track_order(order_id: str):
+    async with async_session() as session:
+        result = await session.execute(
+            OrderTracking.__table__.select()
+            .where(OrderTracking.order_id == order_id)
+            .order_by(OrderTracking.timestamp)
+        )
+        tracking_updates = result.fetchall()
+        if tracking_updates:
+            return [dict(row) for row in tracking_updates]
+        else:
+            raise HTTPException(status_code=404, detail="No tracking information found for this order.")
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8000)