Update app.py

app.py

@@ -5,17 +5,12 @@ from PIL import Image
 import os
 # HARDCODED_URL = ""  # Replace with your desired URL
 
-GLIF_API_TOKEN = os.getenv("GLIF_API_TOKEN1")
-GLIF_API_URL = "https://simple-api.glif.app"
-glif_token_id = 1
-glif_tokens_tried = 0
+TOKEN = os.getenv("TOKEN1")
+API_URL = os.getenv("API_URL")
+token_id = 1
+tokens_tried = 0
 no_of_accounts = 3
-
-import gradio as gr
-from PIL import Image
-from io import BytesIO
-import requests
-import os
+model_id = os.getenv("cm3ugmzv2002gnckiosrwk6xi")
 
 def get_image_from_url(url):
     """
@@ -32,27 +27,27 @@ def get_image_from_url(url):
         return f"Error processing image: {e}"
 
 def generate_image(prompt, aspect_ratio, realism):
-    global glif_token_id
-    global GLIF_API_TOKEN
-    global glif_tokens_tried
+    global token_id
+    global TOKEN
+    global tokens_tried
     global no_of_accounts
 
     payload = {
-        "id": "cm3ugmzv2002gnckiosrwk6xi",  # Replace with appropriate ID if necessary
+        "id": model_id,  # Replace with appropriate ID if necessary
         "inputs": [prompt, aspect_ratio, str(realism).lower()],
     }
-    headers = {"Authorization": f"Bearer {GLIF_API_TOKEN}"}
+    headers = {"Authorization": f"Bearer {TOKEN}"}
 
     try:
-        response_data = requests.post(GLIF_API_URL, json=payload, headers=headers).json()
+        response_data = requests.post(API_URL, json=payload, headers=headers).json()
         if "error" in response_data:
             if 'error 429' in response_data['error']:
-                if glif_tokens_tried < no_of_accounts:
-                    glif_token_id = (glif_token_id + 1) % no_of_accounts
-                    glif_tokens_tried += 1
-                    GLIF_API_TOKEN = os.getenv(f"GLIF_API_TOKEN{glif_token_id}")
+                if tokens_tried < no_of_accounts:
+                    token_id = (token_id + 1) % no_of_accounts
+                    tokens_tried += 1
+                    TOKEN = os.getenv(f"TOKEN{token_id}")
                     response_data = generate_image(prompt, aspect_ratio, realism)
-                    glif_tokens_tried = 0
+                    tokens_tried = 0
                     return response_data
                 return "No credits available"
             return response_data
@@ -62,7 +57,7 @@ def generate_image(prompt, aspect_ratio, realism):
         else:
             return "Error: Unexpected response from server"
     except Exception as e:
-        return f"Error: {e}"
+        return f"Error"
 
 # Define the Gradio interface
 interface = gr.Interface(