Update app.py

app.py

@@ -1,82 +1,63 @@
 import os
+import requests
 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")
-GLIF_API_KEY = os.getenv("GLIF_API_KEY")
+# GLIF Simple API endpoint
 GLIF_API_URL = "https://simple-api.glif.app"
 
-# Discord bot setup
-intents = discord.Intents.default()
-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
+    # Hardcoded aspect ratio (9:16)
+    aspect_ratio = "9:16"
+    
+    # Prepare the payload for the GLIF API
     payload = {
-        "id": "cm3ugmzv2002gnckiosrwk6xi",  # Your GLIF ID
-        "inputs": [prompt, aspect_ratio]
+        "id": "cm3ugmzv2002gnckiosrwk6xi",  # Glif ID
+        "inputs": [prompt, aspect_ratio]  # Passing the prompt and the fixed aspect ratio
+    }
+
+    # Set up headers with the API token
+    headers = {
+        "Authorization": f"Bearer {os.getenv('GLIF_API_KEY')}"
     }
-    headers = {"Authorization": f"Bearer {GLIF_API_KEY}"}
-    try:
-        response = requests.post(GLIF_API_URL, json=payload, headers=headers)
-        if response.status_code == 200:
-            response_data = response.json()
-            if "output" in response_data:
-                return response_data["output"]  # Image URL
-            elif "error" in response_data:
-                return f"Error: {response_data['error']}"
-        else:
-            return f"API request failed with status code: {response.status_code}"
-    except Exception as e:
-        return f"Error: {str(e)}"
 
-# Discord bot commands
+    # Make the request to the GLIF API
+    response = requests.post(GLIF_API_URL, json=payload, headers=headers)
+
+    if response.status_code == 200:
+        # Parse the response
+        response_data = response.json()
+        if "output" in response_data:
+            # Return the image URL
+            return response_data["output"]
+        elif "error" in response_data:
+            # Handle errors (even though the status code is 200)
+            return f"Error: {response_data['error']}"
+    else:
+        return f"API request failed with status code: {response.status_code}"
+
+# Initialize the Discord bot
+intents = discord.Intents.default()
+intents.message_content = True
+bot = commands.Bot(command_prefix="!", intents=intents)
+
+@bot.event
+async def on_ready():
+    print(f'Logged in as {bot.user.name}')
+
 @bot.command(name="generate")
 async def generate(ctx, *, prompt: str):
-    await ctx.send(f"Generating an image for: `{prompt}`...")
+    """Generate an image based on the provided prompt."""
+    await ctx.send(f"Generating image for prompt: {prompt}...")
+    
+    # Call the GLIF API to generate the image
     image_url = generate_image(prompt)
+    
     if image_url.startswith("http"):
-        await ctx.send(image_url)  # Send the image URL
+        await ctx.send(image_url)
     else:
         await ctx.send(f"Failed to generate image: {image_url}")
 
-@bot.event
-async def on_ready():
-    print(f"Logged in as {bot.user}")
-
-# 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")
-
-# Use the lifespan handler in the FastAPI app
-app.lifespan = lifespan
-
-# Health check endpoint
-@app.get("/")
-def health_check():
-    return {"status": "Bot is running!"}
+# Run the bot
+bot.run(os.getenv('DISCORD_BOT_TOKEN'))