Update app.py

app.py

@@ -22,45 +22,35 @@ sd_decode = os.environ.get('sd_decode')
 
 def freeimagehost(prompt, img, seed):
     try:
-        # Replace with the path to your proxy.txt file
-        proxy_file = "proxies.txt"
-
-        # Read the proxies from the file and store them in a list
-        with open(proxy_file, "r") as f:
-            proxies = f.read().splitlines()
+        rp = RegisteredProviders()
+        rp.parse_providers()
+        # get one proxy
+        proxy = rp.get_random_proxy()
+        proxy = proxy.get_proxy()
+        # convert proxy to dictionary
+        proxy = proxy.split(':')
+        # convert list to dictionary
+        proxy = {'http': 'http://' + proxy[0] + ':' + proxy[1]}
+    except:
+        print("Error getting proxy")
+        proxy = {}
 
-        # Read the image file and encode it as base64
+    try:
         buffered = BytesIO()
-        img.save(buffered, format="PNG")
-        encoded_image = base64.b64encode(buffered.getvalue()).decode("utf-8")
-
-        uploaded = False
-        # Iterate over the proxies and make a post request with each one
-        for proxy in proxies:
-            if not uploaded:
-                try:
-                    # Create a payload with the base64-encoded image and API key
-                    key = "6d207e02198a847aa98d0a2a901485a5"
-                    payload = {"key": key, "image": encoded_image, "format": "json", "expiration": "3600"}
-
-                    # Make a post request with the proxy to upload the image
-                    response = requests.post("https://freeimage.host/api/1/upload", data=payload, proxies={"http": proxy})
-                    link = loglink = response.json()['image']['url']
-                    link = '<a style="color:blue;" href="https://arinteli.com/app/?' + link + '+' + prompt + '+' + seed + '" target=”_blank”>'+ link +'</a>'
-                    uploaded = True
-
-                except:
-                    # If there's an error, assume the proxy is not working
-                    print(f"Error uploading file with proxy {proxy}")
-
-        if not uploaded:
-            # If the image was not uploaded with any of the proxies, fallback to imagebb
+        img.save(buffered, format="JPEG")
+        img_str = base64.b64encode(buffered.getvalue())
+        key = "6d207e02198a847aa98d0a2a901485a5"
+        payload = {'key': key, 'image': img_str, 'format': 'json', 'expiration': '3600'}
+        r = requests.post("https://freeimage.host/api/1/upload", data=payload, proxies=proxy)
+        r = json.loads(r.text)
+        try:
+            link = loglink = r['image']['url']
+            link = '<a style="color:blue;" href="https://arinteli.com/app/?' + link + '+' + prompt + '+' + seed + '" target=”_blank”>'+ link +'</a>'
+        except:
+            print(r)
             return imagebb(prompt, img, seed)
-
-    except:
-        # If there's an error, fallback to imagebb
+    except: 
         return imagebb(prompt, img, seed)
-
     return link, loglink
 
 bbtries = 0