Update app.py

app.py

@@ -7,6 +7,7 @@ from discord.ext import commands
 from dotenv import load_dotenv
 import logging
 import io
+import time
 from PIL import Image
 
 # Load environment variables from the .env file
@@ -23,14 +24,19 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Function to query the Hugging Face model with a structured prompt
-def query(payload):
-    try:
-        response = requests.post(API_URL, headers=headers, json=payload)
-        response.raise_for_status()
-        return response.content
-    except requests.exceptions.RequestException as e:
-        logger.error(f"Error querying the API: {e}")
-        return None
+def query(payload, max_retries=5, delay=5):
+    retries = 0
+    while retries < max_retries:
+        try:
+            response = requests.post(API_URL, headers=headers, json=payload)
+            response.raise_for_status()
+            return response.content
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Error querying the API: {e}. Retrying in {delay} seconds...")
+            time.sleep(delay)
+            retries += 1
+    logger.error(f"Failed to query the API after {max_retries} attempts.")
+    return None
 
 # Initialize the Discord bot with a default prefix
 intents = discord.Intents.default()
@@ -97,3 +103,4 @@ async def help_custom(ctx):
 # Run the bot
 bot.run(DISCORD_TOKEN)
 
+