Update app.py

app.py

@@ -10,7 +10,9 @@ import gradio as gr  # Import Gradio
 DISCORD_BOT_TOKEN = os.getenv("DISCORD_BOT_TOKEN")
 GLIF_API_TOKEN = os.getenv("GLIF_API_TOKEN")
 GLIF_API_URL = "https://simple-api.glif.app"
-
+glif_token_id = 0
+glif_tokens_tried = 0
+no_of_accounts = 7
 if not DISCORD_BOT_TOKEN or not GLIF_API_TOKEN:
     raise ValueError("Both DISCORD_BOT_TOKEN and GLIF_API_TOKEN must be set.")
 
@@ -34,7 +36,14 @@ async def generate_image_async(prompt, aspect_ratio):
             ) as response:
                 response.raise_for_status()
                 response_data = await response.json()
-                if "output" in response_data:
+                if "error" in response_data:
+                    if 'error 429' in response_data['error']:
+                        if glif_tokens_tried<no_of_accounts:
+                            global glif_token_id = (glif_token_id+1)%no_of_accounts
+                            global GLIF_API_TOKEN = os.getenv(f"GLIF_API_TOKEN{glif_token_id}")
+                            response_data = await generate_image_async(prompt, aspect_ratio)
+                            glif_tokens_tried = 0
+                        response_data = "No credits available"
                     return response_data
                 elif "error" in response_data:
                     return response_data['error']