Update app.py

app.py

@@ -1,25 +1,39 @@
-import os, sys, json, re, time, base64, random, requests, shutil
+import os, sys, json, re, time, base64, random, shutil
 import gradio as gr
+import numpy as np
+import requests
+from requests import Session
+from requests.adapters import HTTPAdapter
+from requests.packages.urllib3.util.retry import Retry
 from datetime import datetime
 import urllib.parse
 from groq import Groq
 from exif import Image
 from PIL import Image as PILImage, ExifTags as PILExifTags
+from io import BytesIO
 import colorsys
 
 # MARK: INIT
-MAX_SEED = 9999
+MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
-GROQ_APIKEY_PROMPTENHANCE = os.environ["GROQ_APIKEY_PROMPTENHANCE"]
+GROQ_APIKEY_PROMPTENHANCE = os.getenv("GROQ_APIKEY_PROMPTENHANCE")
+API_V1 = os.getenv("API_V1")
+API_V1_MODELS = urllib.parse.unquote(API_V1 + "/models")
+API_V1_IMAGE = urllib.parse.unquote(API_V1 + "/imagine2")
+API_OLD = os.getenv("API_OLD")
+
+# print(API_V1_MODELS)
+# print(API_V1_IMAGE)
+
 
-# delete cache folder if exist and create new
 CACHE_DIR = os.path.join(os.path.dirname(__file__), "cache")
 IMAGE_DIR = os.path.join(CACHE_DIR, "images")
-if os.path.exists(CACHE_DIR):
-    shutil.rmtree(CACHE_DIR)
-os.makedirs(CACHE_DIR)
-os.makedirs(IMAGE_DIR)
+if not os.path.exists(CACHE_DIR):
+    os.makedirs(CACHE_DIR)
+    print(f"Created cache dir on path {CACHE_DIR}")
+    os.makedirs(IMAGE_DIR)
+    print(f"Created images dir on path {IMAGE_DIR}")
 
 RES = os.path.join(os.path.dirname(__file__), "_res")
 
@@ -227,18 +241,18 @@ with gr.Blocks(theme=theme, head=custom_head, css=custom_css, js=custom_js, titl
         def calculate_ratio_values(image_ratio_buttons, image_width, image_height):
             ratio_value = image_ratio_buttons.split(":")
             if int(ratio_value[0]) > int(ratio_value[1]):
-                a = 1024
+                a = 2048
                 b = int(a * int(ratio_value[1]) / int(ratio_value[0]))
                 new_width = a
                 new_height = b
             elif int(ratio_value[0]) < int(ratio_value[1]):
-                b = 1024
+                b = 2048
                 a = int(b * int(ratio_value[0]) / int(ratio_value[1]))
                 new_width = a
                 new_height = b
             else:
-                new_width = 1024
-                new_height = 1024
+                new_width = 2048
+                new_height = 2048
 
             return new_width, new_height