Update app.py

app.py

@@ -6,7 +6,7 @@ import base64
 import asyncio
 from datetime import datetime, timedelta
 from bs4 import BeautifulSoup
-from sqlalchemy import select
+from sqlalchemy import select, text  # make sure to import text for SQL commands
 from pydantic import BaseModel
 
 from fastapi import FastAPI, Request, HTTPException, BackgroundTasks, UploadFile, File, Form
@@ -25,10 +25,10 @@ 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")  # Example using SQLite
+DATABASE_URL = os.getenv("DATABASE_URL", "postgresql+asyncpg://user:password@localhost/dbname")  # Update as needed
 NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY", "default_fallback_value")
 openai.api_key = os.getenv("OPENAI_API_KEY", "default_fallback_value")
-GOOGLE_MAPS_API_KEY = os.getenv("GOOGLE_MAPS_API_KEY", "default_fallback_value")  # Add Google Maps API key
+GOOGLE_MAPS_API_KEY = os.getenv("GOOGLE_MAPS_API_KEY", "default_fallback_value")
 
 # WhatsApp Business API credentials (Cloud API)
 WHATSAPP_PHONE_NUMBER_ID = os.getenv("WHATSAPP_PHONE_NUMBER_ID", "default_value")
@@ -100,9 +100,13 @@ class OrderTracking(Base):
 engine = create_async_engine(DATABASE_URL, echo=True)
 async_session = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
 
+# --- Migration Functions ---
 async def add_order_ids_column():
     async with engine.connect() as conn:
-        await conn.execute(text("ALTER TABLE user_profiles ADD COLUMN order_ids TEXT DEFAULT ''"))
+        # You can add an existence check if needed.
+        await conn.execute(
+            text("ALTER TABLE user_profiles ADD COLUMN IF NOT EXISTS order_ids TEXT DEFAULT ''")
+        )
         await conn.commit()
     print("Column 'order_ids' added successfully.")
 
@@ -111,21 +115,16 @@ async def init_db():
         await conn.run_sync(Base.metadata.create_all)
     print("Database initialized.")
 
-# The main function serves as the entry point for our async operations.
-async def main():
-    # Await each coroutine in order
+# --- FastAPI Application Setup ---
+app = FastAPI()
+
+# Run migrations at startup
+@app.on_event("startup")
+async def startup_event():
     await add_order_ids_column()
     await init_db()
 
-if __name__ == "__main__":
-    # asyncio.run() creates a new event loop and runs the main() coroutine inside it.
-    asyncio.run(main())
-    
-async def init_db():
-    async with engine.begin() as conn:
-        await conn.run_sync(Base.metadata.create_all)
-
-# --- Global In-Memory Stores ---
+# --- Global In-Memory Stores and Other Code ---
 user_state = {}       # e.g., { user_id: ConversationState }
 conversation_context = {}  # { user_id: [ { "timestamp": ..., "role": "user"/"bot", "message": ... }, ... ] }
 proactive_timer = {}
@@ -137,7 +136,6 @@ menu_items = [
     {"name": "Egusi Soup", "description": "A rich and hearty soup made with melon seeds", "price": 1000, "nutrition": "Calories: 250 kcal, Carbs: 15g, Protein: 8g, Fat: 10g"}
 ]
 
-# --- Conversation State Management ---
 SESSION_TIMEOUT = timedelta(minutes=5)
 
 class ConversationState: