Update app.py

app.py

@@ -3,11 +3,12 @@ import requests
 from fpdf import FPDF
 import pdfplumber
 import os
+import time
 
 # Mistral API key (replace with your key or use environment variable)
 api_key = os.getenv("MISTRAL_API_KEY", "gz6lDXokxgR6cLY72oomALWcm7vhjRzQ")
 
-# Function to call Mistral API
+# Function to call Mistral API with rate limit handling
 def call_mistral_api(prompt):
     url = "https://api.mistral.ai/v1/chat/completions"
     headers = {
@@ -25,6 +26,10 @@ def call_mistral_api(prompt):
         response.raise_for_status()  # Raise an error for bad status codes
         return response.json()['choices'][0]['message']['content']
     except requests.exceptions.HTTPError as err:
+        if response.status_code == 429:  # Rate limit exceeded
+            st.warning("Rate limit exceeded. Please wait a few seconds and try again.")
+            time.sleep(5)  # Wait for 5 seconds before retrying
+            return call_mistral_api(prompt)  # Retry the request
         return f"HTTP Error: {err}"
     except Exception as err:
         return f"Error: {err}"