Update app.py

app.py

@@ -3,7 +3,9 @@ import discord
 from discord.ext import commands
 import requests
 from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
 import threading
+from contextlib import asynccontextmanager
 
 # Load the bot token and API key from environment variables
 DISCORD_BOT_TOKEN = os.getenv("DISCORD_BOT_TOKEN")
@@ -17,6 +19,15 @@ bot = commands.Bot(command_prefix="!", intents=intents)
 # FastAPI app
 app = FastAPI()
 
+# Enable CORS if necessary
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 # Function to call the GLIF API
 def generate_image(prompt):
     aspect_ratio = "9:16"  # Hardcoded aspect ratio
@@ -52,14 +63,18 @@ async def generate(ctx, *, prompt: str):
 async def on_ready():
     print(f"Logged in as {bot.user}")
 
-# Function to run the bot in a separate thread
-def run_bot():
-    bot.run(DISCORD_BOT_TOKEN)
+# Define an async context manager for the lifespan event
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Start the bot thread during startup
+    thread = threading.Thread(target=bot.run, args=(DISCORD_BOT_TOKEN,), daemon=True)
+    thread.start()
+    yield  # Application lifecycle continues here
+    # Optionally handle cleanup here if needed
+    print("FastAPI application is shutting down")
 
-# Start the bot thread when FastAPI starts
-@app.on_event("startup")
-def startup_event():
-    threading.Thread(target=run_bot, daemon=True).start()
+# Use the lifespan handler in the FastAPI app
+app.lifespan = lifespan
 
 # Health check endpoint
 @app.get("/")