Update app.py

app.py

@@ -4,20 +4,50 @@ import io
 import random
 import os
 import time
-from PIL import Image
 import json
+from PIL import Image
 
 # Project by Nymbo
 
-# Base API URL for Hugging Face inference
-API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
-# Retrieve the API token from environment variables
-API_TOKEN = os.getenv("HF_READ_TOKEN")
-headers = {"Authorization": f"Bearer {API_TOKEN}"}
+# Retrieve the default API token from environment variables
+HF_TOKEN = os.getenv("HF_READ_TOKEN")
 # Timeout for requests
 timeout = 100
 
-def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024, custom_api_key=""):
+# Provider-specific API handlers
+def query_hf_inference(prompt, model_path, headers, payload, timeout):
+    """Handler for Hugging Face Inference API requests"""
+    api_url = f"https://api-inference.huggingface.co/models/{model_path}"
+    print(f"Making request to HF Inference API: {api_url}")
+    return requests.post(api_url, headers=headers, json=payload, timeout=timeout)
+
+def query_fal_ai(prompt, model_path, headers, payload, timeout):
+    """Handler for Fal AI API requests"""
+    api_url = f"https://live.fal.ai/hugging-face/{model_path}"
+    print(f"Making request to Fal AI: {api_url}")
+    return requests.post(api_url, headers=headers, json=payload, timeout=timeout)
+
+def query_nebius(prompt, model_path, headers, payload, timeout):
+    """Handler for Nebius API requests"""
+    api_url = f"https://router.huggingface.co/nebius/v1/text-to-image/{model_path}"
+    print(f"Making request to Nebius: {api_url}")
+    return requests.post(api_url, headers=headers, json=payload, timeout=timeout)
+
+def query_replicate(prompt, model_path, headers, payload, timeout):
+    """Handler for Replicate API requests"""
+    api_url = f"https://router.huggingface.co/replicate/v1/text-to-image/{model_path}"
+    print(f"Making request to Replicate: {api_url}")
+    return requests.post(api_url, headers=headers, json=payload, timeout=timeout)
+
+def query_together(prompt, model_path, headers, payload, timeout):
+    """Handler for Together API requests"""
+    api_url = f"https://router.huggingface.co/together/v1/text-to-image/{model_path}"
+    print(f"Making request to Together: {api_url}")
+    return requests.post(api_url, headers=headers, json=payload, timeout=timeout)
+
+# Main query function
+def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7, sampler="DPM++ 2M Karras", 
+          seed=-1, strength=0.7, width=1024, height=1024, custom_api_key="", provider="hf-inference"):
     # Debug log to indicate function start
     print("Starting query function...")
     # Print the parameters for debugging purposes
@@ -25,6 +55,7 @@ def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7,
     print(f"Model: {model}")
     print(f"Custom LoRA: {custom_lora}")
     print(f"Parameters - Steps: {steps}, CFG Scale: {cfg_scale}, Seed: {seed}, Strength: {strength}, Width: {width}, Height: {height}")
+    print(f"Provider: {provider}")
     print(f"Custom API Key provided: {bool(custom_api_key.strip())}")  # Log whether a custom key was provided without printing the key
 
     # Check if the prompt is empty or None
@@ -41,9 +72,16 @@ def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7,
         print("USING CUSTOM API KEY: BYOK token provided by user is being used for authentication")
         API_TOKEN = custom_api_key.strip()
     else:
-        # Randomly select an API token from available options to distribute the load
-        API_TOKEN = random.choice([os.getenv("HF_READ_TOKEN"), os.getenv("HF_READ_TOKEN_2"), os.getenv("HF_READ_TOKEN_3"), os.getenv("HF_READ_TOKEN_4"), os.getenv("HF_READ_TOKEN_5")])
-        print("USING DEFAULT API KEY: Random environment variable token is being used for authentication")
+        if provider == "hf-inference":
+            # Randomly select an API token from available options to distribute the load for HF Inference
+            API_TOKEN = random.choice([os.getenv("HF_READ_TOKEN"), os.getenv("HF_READ_TOKEN_2"), 
+                                   os.getenv("HF_READ_TOKEN_3"), os.getenv("HF_READ_TOKEN_4"), 
+                                   os.getenv("HF_READ_TOKEN_5")])
+            print("USING DEFAULT API KEY: Random environment variable token is being used for authentication")
+        else:
+            # For other providers, use the default HF token
+            API_TOKEN = HF_TOKEN
+            print("USING DEFAULT HF TOKEN: Environment variable HF_TOKEN is being used for authentication")
     
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
@@ -51,290 +89,200 @@ def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7,
     prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
     print(f'Generation {key}: {prompt}')  # Debug log
 
-    # Set the API URL based on the selected model or custom LoRA
+    # Set the model path based on the selected model or custom LoRA
     if custom_lora.strip() != "":
-        API_URL = f"https://api-inference.huggingface.co/models/{custom_lora.strip()}"
+        model_path = custom_lora.strip()
     else:
-        if model == 'Stable Diffusion XL':
-            API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
-        if model == 'FLUX.1 [Dev]':
-            API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
-        if model == 'FLUX.1 [Schnell]':
-            API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
-        if model == 'HiDream-I1-Full':
-            API_URL = "https://api-inference.huggingface.co/models/HiDream-ai/HiDream-I1-Full"
-        if model == 'HiDream-I1-Dev':
-            API_URL = "https://api-inference.huggingface.co/models/HiDream-ai/HiDream-I1-Dev"
-        if model == 'HiDream-I1-Fast':
-            API_URL = "https://api-inference.huggingface.co/models/HiDream-ai/HiDream-I1-Fast"
-        if model == 'Animagine 4.0':
-            API_URL = "https://api-inference.huggingface.co/models/cagliostrolab/animagine-xl-4.0"
-            prompt = f"masterpiece, high score, great score, absurdres, {prompt}"
-        if model == 'Flux Icon Kit':
-            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Icon-Kit-LoRA"
-            prompt = f"Icon Kit, {prompt}"
-        if model == 'Pixel Background':
-            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Pixel-Background-LoRA"
-            prompt = f"Pixel Background, {prompt}"
-        if model == 'Meme XD':
-            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Flux-Meme-Xd-LoRA"
-            prompt = f"meme, {prompt}"
-        if model == 'Chill Guy':
-            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Flux-Chill-Guy-Zone"
-            prompt = f"chill guy, {prompt}"
-        if model == 'Pepe':
-            API_URL = "https://api-inference.huggingface.co/models/openfree/pepe"
-            prompt = f"pepe, {prompt}"
-        if model == 'NSFWmodel':
-            API_URL = "https://api-inference.huggingface.co/models/lexa862/NSFWmodel"
-            prompt = f"nude, {prompt}"
-        if model == 'Claude Art':
-            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Claude-Art"
-            prompt = f"claude art, {prompt}"
-        if model == 'Open Genmoji':
-            API_URL = "https://api-inference.huggingface.co/models/EvanZhouDev/open-genmoji"
-            prompt = f"emoji, {prompt}"
-        if model == 'EBook Creative Cover':
-            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/EBook-Creative-Cover-Flux-LoRA"
-            prompt = f"EBook Cover, {prompt}"
-        if model == 'Flux Logo Design 2':
-            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Logo-Design-Flux-LoRA"
-            prompt = f"Logo Design, {prompt}"
-        if model == 'Isometric 3D':
-            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Isometric-3D-LoRA"
-            prompt = f"Isometric 3D, {prompt}"
-        if model == 'Flux Condensation':
-            API_URL = "https://api-inference.huggingface.co/models/fofr/flux-condensation"
-            prompt = f"CONDENSATION, {prompt}"
-        if model == 'Flux Handwriting':
-            API_URL = "https://api-inference.huggingface.co/models/fofr/flux-handwriting"
-            prompt = f"HWRIT handwriting, {prompt}"
-        if model == 'Shou Xin':
-            API_URL = "https://api-inference.huggingface.co/models/Datou1111/shou_xin"
-            prompt = f"shou_xin, pencil sketch, {prompt}"
-        if model == 'Sketch Smudge':
-            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Sketch-Smudge-LoRA"
-            prompt = f"Sketch Smudge, {prompt}"
-        if model == '80s Cyberpunk':
-            API_URL = "https://api-inference.huggingface.co/models/fofr/flux-80s-cyberpunk"
-            prompt = f"80s cyberpunk, {prompt}"
-        if model == 'Coloring Book Flux':
-            API_URL = "https://api-inference.huggingface.co/models/renderartist/coloringbookflux"
-            prompt = f"c0l0ringb00k, coloring book, coloring book page, {prompt}"
-        if model == 'Flux Miniature LoRA':
-            API_URL = "https://api-inference.huggingface.co/models/gokaygokay/Flux-Miniature-LoRA"
-            prompt = f"MNTR, miniature drawing, {prompt}"
-        if model == 'Sketch Paint':
-            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Sketch-Paint"
-            prompt = f"Sketch paint, {prompt}"
-        if model == 'Flux UltraRealism 2.0':
-            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0"
-            prompt = f"Ultra realistic, {prompt}"
-        if model == 'Midjourney Mix':
-            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Midjourney-Mix-LoRA"
-            prompt = f"midjourney mix, {prompt}"
-        if model == 'Midjourney Mix 2':
-            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Midjourney-Mix2-LoRA"
-            prompt = f"MJ v6, {prompt}"
-        if model == 'Flux Logo Design':
-            API_URL = "https://api-inference.huggingface.co/models/Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design"
-            prompt = f"wablogo, logo, Minimalist, {prompt}"
-        if model == 'Flux Uncensored':
-            API_URL = "https://api-inference.huggingface.co/models/enhanceaiteam/Flux-uncensored"
-        if model == 'Flux Uncensored V2':
-            API_URL = "https://api-inference.huggingface.co/models/enhanceaiteam/Flux-Uncensored-V2"
-        if model == 'Flux Tarot Cards':
-            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Ton618-Tarot-Cards-Flux-LoRA"
-            prompt = f"Tarot card, {prompt}"
-        if model == 'Pixel Art Sprites':
-            API_URL = "https://api-inference.huggingface.co/models/sWizad/pokemon-trainer-sprites-pixelart-flux"
-            prompt = f"a pixel image, {prompt}"
-        if model == '3D Sketchfab':
-            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Castor-3D-Sketchfab-Flux-LoRA"
-            prompt = f"3D Sketchfab, {prompt}"
-        if model == 'Retro Comic Flux':
-            API_URL = "https://api-inference.huggingface.co/models/renderartist/retrocomicflux"
-            prompt = f"c0m1c, comic book panel, {prompt}"
-        if model == 'Caricature':
-            API_URL = "https://api-inference.huggingface.co/models/TheAwakenOne/caricature"
-            prompt = f"CCTUR3, {prompt}"
-        if model == 'Huggieverse':
-            API_URL = "https://api-inference.huggingface.co/models/Chunte/flux-lora-Huggieverse"
-            prompt = f"HGGRE, {prompt}"
-        if model == 'Propaganda Poster':
-            API_URL = "https://api-inference.huggingface.co/models/AlekseyCalvin/Propaganda_Poster_Schnell_by_doctor_diffusion"
-            prompt = f"propaganda poster, {prompt}"
-        if model == 'Flux Game Assets V2':
-            API_URL = "https://api-inference.huggingface.co/models/gokaygokay/Flux-Game-Assets-LoRA-v2"
-            prompt = f"wbgmsst, white background, {prompt}"
-        if model == 'SDXL HS Card Style':
-            API_URL = "https://api-inference.huggingface.co/models/Norod78/sdxl-hearthstone-card-style-lora"
-            prompt = f"Hearthstone Card, {prompt}"
-        if model == 'SLDR FLUX NSFW v2 Studio':
-            API_URL = "https://api-inference.huggingface.co/models/xey/sldr_flux_nsfw_v2-studio"
-        if model == 'SoftPasty Flux':
-            API_URL = "https://api-inference.huggingface.co/models/alvdansen/softpasty-flux-dev"
-            prompt = f"araminta_illus illustration style, {prompt}"
-        if model == 'Flux Stickers':
-            API_URL = "https://api-inference.huggingface.co/models/diabolic6045/Flux_Sticker_Lora"
-            prompt = f"5t1cker 5ty1e, {prompt}"
-        if model == 'Flux Animex V2':
-            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Animex-v2-LoRA"
-            prompt = f"Animex, {prompt}"
-        if model == 'Flux Animeo V1':
-            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Animeo-v1-LoRA"
-            prompt = f"Animeo, {prompt}"
-        if model == 'Movie Board':
-            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Flux.1-Dev-Movie-Boards-LoRA"
-            prompt = f"movieboard, {prompt}"
-        if model == 'Purple Dreamy':
-            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Purple-Dreamy-Flux-LoRA"
-            prompt = f"Purple Dreamy, {prompt}"
-        if model == 'PS1 Style Flux':
-            API_URL = "https://api-inference.huggingface.co/models/veryVANYA/ps1-style-flux"
-            prompt = f"ps1 game screenshot, {prompt}"
-        if model == 'Softserve Anime':
-            API_URL = "https://api-inference.huggingface.co/models/alvdansen/softserve_anime"
-            prompt = f"sftsrv style illustration, {prompt}"
-        if model == 'Flux Tarot v1':
-            API_URL = "https://api-inference.huggingface.co/models/multimodalart/flux-tarot-v1"
-            prompt = f"in the style of TOK a trtcrd tarot style, {prompt}"
-        if model == 'Half Illustration':
-            API_URL = "https://api-inference.huggingface.co/models/davisbro/half_illustration"
-            prompt = f"in the style of TOK, {prompt}"
-        if model == 'OpenDalle v1.1':
-            API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/OpenDalleV1.1"
-        if model == 'Flux Ghibsky Illustration':
-            API_URL = "https://api-inference.huggingface.co/models/aleksa-codes/flux-ghibsky-illustration"
-            prompt = f"GHIBSKY style, {prompt}"
-        if model == 'Flux Koda':
-            API_URL = "https://api-inference.huggingface.co/models/alvdansen/flux-koda"
-            prompt = f"flmft style, {prompt}"
-        if model == 'Soviet Diffusion XL':
-            API_URL = "https://api-inference.huggingface.co/models/openskyml/soviet-diffusion-xl"
-            prompt = f"soviet poster, {prompt}"
-        if model == 'Flux Realism LoRA':
-            API_URL = "https://api-inference.huggingface.co/models/XLabs-AI/flux-RealismLora"
-        if model == 'Frosting Lane Flux':
-            API_URL = "https://api-inference.huggingface.co/models/alvdansen/frosting_lane_flux"
-            prompt = f"frstingln illustration, {prompt}"
-        if model == 'Phantasma Anime':
-            API_URL = "https://api-inference.huggingface.co/models/alvdansen/phantasma-anime"
-        if model == 'Boreal':
-            API_URL = "https://api-inference.huggingface.co/models/kudzueye/Boreal"
-            prompt = f"photo, {prompt}"
-        if model == 'How2Draw':
-            API_URL = "https://api-inference.huggingface.co/models/glif/how2draw"
-            prompt = f"How2Draw, {prompt}"
-        if model == 'Flux AestheticAnime':
-            API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/FLUX-AestheticAnime"
-        if model == 'Fashion Hut Modeling LoRA':
-            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Fashion-Hut-Modeling-LoRA"
-            prompt = f"Modeling of, {prompt}"
-        if model == 'Flux SyntheticAnime':
-            API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/FLUX-SyntheticAnime"
-            prompt = f"1980s anime screengrab, VHS quality, syntheticanime, {prompt}"
-        if model == 'Flux Midjourney Anime':
-            API_URL = "https://api-inference.huggingface.co/models/brushpenbob/flux-midjourney-anime"
-            prompt = f"egmid, {prompt}"
-        if model == 'Coloring Book Generator':
-            API_URL = "https://api-inference.huggingface.co/models/robert123231/coloringbookgenerator"
-        if model == 'Collage Flux':
-            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Castor-Collage-Dim-Flux-LoRA"
-            prompt = f"collage, {prompt}"
-        if model == 'Flux Product Ad Backdrop':
-            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Flux-Product-Ad-Backdrop"
-            prompt = f"Product Ad, {prompt}"
-        if model == 'Product Design':
-            API_URL = "https://api-inference.huggingface.co/models/multimodalart/product-design"
-            prompt = f"product designed by prdsgn, {prompt}"
-        if model == '90s Anime Art':
-            API_URL = "https://api-inference.huggingface.co/models/glif/90s-anime-art"
-        if model == 'Brain Melt Acid Art':
-            API_URL = "https://api-inference.huggingface.co/models/glif/Brain-Melt-Acid-Art"
-            prompt = f"maximalism, in an acid surrealism style, {prompt}"
-        if model == 'Lustly Flux Uncensored v1':
-            API_URL = "https://api-inference.huggingface.co/models/lustlyai/Flux_Lustly.ai_Uncensored_nsfw_v1"
-        if model == 'NSFW Master Flux':
-            API_URL = "https://api-inference.huggingface.co/models/Keltezaa/NSFW_MASTER_FLUX"
-            prompt = f"NSFW, {prompt}"
-        if model == 'Flux Outfit Generator':
-            API_URL = "https://api-inference.huggingface.co/models/tryonlabs/FLUX.1-dev-LoRA-Outfit-Generator"
-        if model == 'Midjourney':
-            API_URL = "https://api-inference.huggingface.co/models/Jovie/Midjourney"
-        if model == 'DreamPhotoGASM':
-            API_URL = "https://api-inference.huggingface.co/models/Yntec/DreamPhotoGASM"
-        if model == 'Flux Super Realism LoRA':
-            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Super-Realism-LoRA"
-        if model == 'Stable Diffusion 2-1':
-            API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1-base"
-        if model == 'Stable Diffusion 3.5 Large':
-            API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large"
-        if model == 'Stable Diffusion 3.5 Large Turbo':
-            API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large-turbo"
-        if model == 'Stable Diffusion 3 Medium':
-            API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3-medium-diffusers"
-            prompt = f"A, {prompt}"
-        if model == 'Duchaiten Real3D NSFW XL':
-            API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/duchaiten-real3d-nsfw-xl"
-        if model == 'Pixel Art XL':
-            API_URL = "https://api-inference.huggingface.co/models/nerijs/pixel-art-xl"
-            prompt = f"pixel art, {prompt}"
-        if model == 'Character Design':
-            API_URL = "https://api-inference.huggingface.co/models/KappaNeuro/character-design"
-            prompt = f"Character Design, {prompt}"
-        if model == 'Sketched Out Manga':
-            API_URL = "https://api-inference.huggingface.co/models/alvdansen/sketchedoutmanga"
-            prompt = f"daiton, {prompt}"
-        if model == 'Archfey Anime':
-            API_URL = "https://api-inference.huggingface.co/models/alvdansen/archfey_anime"
-        if model == 'Lofi Cuties':
-            API_URL = "https://api-inference.huggingface.co/models/alvdansen/lofi-cuties"
-        if model == 'YiffyMix':
-            API_URL = "https://api-inference.huggingface.co/models/Yntec/YiffyMix"
-        if model == 'Analog Madness Realistic v7':
-            API_URL = "https://api-inference.huggingface.co/models/digiplay/AnalogMadness-realistic-model-v7"
-        if model == 'Selfie Photography':
-            API_URL = "https://api-inference.huggingface.co/models/artificialguybr/selfiephotographyredmond-selfie-photography-lora-for-sdxl"
-            prompt = f"instagram model, discord profile picture, {prompt}"
-        if model == 'Filmgrain':
-            API_URL = "https://api-inference.huggingface.co/models/artificialguybr/filmgrain-redmond-filmgrain-lora-for-sdxl"
-            prompt = f"Film Grain, FilmGrainAF, {prompt}"
-        if model == 'Leonardo AI Style Illustration':
-            API_URL = "https://api-inference.huggingface.co/models/goofyai/Leonardo_Ai_Style_Illustration"
-            prompt = f"leonardo style, illustration, vector art, {prompt}"
-        if model == 'Cyborg Style XL':
-            API_URL = "https://api-inference.huggingface.co/models/goofyai/cyborg_style_xl"
-            prompt = f"cyborg style, {prompt}"
-        if model == 'Little Tinies':
-            API_URL = "https://api-inference.huggingface.co/models/alvdansen/littletinies"
-        if model == 'NSFW XL':
-            API_URL = "https://api-inference.huggingface.co/models/Dremmar/nsfw-xl"
-        if model == 'Analog Redmond':
-            API_URL = "https://api-inference.huggingface.co/models/artificialguybr/analogredmond"
-            prompt = f"timeless style, {prompt}"
-        if model == 'Pixel Art Redmond':
-            API_URL = "https://api-inference.huggingface.co/models/artificialguybr/PixelArtRedmond"
-            prompt = f"Pixel Art, {prompt}"
-        if model == 'Ascii Art':
-            API_URL = "https://api-inference.huggingface.co/models/CiroN2022/ascii-art"
-            prompt = f"ascii art, {prompt}"
-        if model == 'Analog':
-            API_URL = "https://api-inference.huggingface.co/models/Yntec/Analog"
-        if model == 'Maple Syrup':
-            API_URL = "https://api-inference.huggingface.co/models/Yntec/MapleSyrup"
-        if model == 'Perfect Lewd Fantasy':
-            API_URL = "https://api-inference.huggingface.co/models/digiplay/perfectLewdFantasy_v1.01"
-        if model == 'AbsoluteReality 1.8.1':
-            API_URL = "https://api-inference.huggingface.co/models/digiplay/AbsoluteReality_v1.8.1"
-        if model == 'Disney':
-            API_URL = "https://api-inference.huggingface.co/models/goofyai/disney_style_xl"
-            prompt = f"Disney style, {prompt}"
-        if model == 'Redmond SDXL':
-            API_URL = "https://api-inference.huggingface.co/models/artificialguybr/LogoRedmond-LogoLoraForSDXL-V2"
-        if model == 'epiCPhotoGasm':
-            API_URL = "https://api-inference.huggingface.co/models/Yntec/epiCPhotoGasm"
-    print(f"API URL set to: {API_URL}")  # Debug log
+        # Map to select the appropriate model path
+        model_mapping = {
+            'Stable Diffusion XL': "stabilityai/stable-diffusion-xl-base-1.0",
+            'FLUX.1 [Dev]': "black-forest-labs/FLUX.1-dev",
+            'FLUX.1 [Schnell]': "black-forest-labs/FLUX.1-schnell",
+            'HiDream-I1-Full': "HiDream-ai/HiDream-I1-Full",
+            'HiDream-I1-Dev': "HiDream-ai/HiDream-I1-Dev",
+            'HiDream-I1-Fast': "HiDream-ai/HiDream-I1-Fast",
+            'Animagine 4.0': "cagliostrolab/animagine-xl-4.0",
+            'Flux Icon Kit': "strangerzonehf/Flux-Icon-Kit-LoRA",
+            'Pixel Background': "strangerzonehf/Flux-Pixel-Background-LoRA",
+            'Meme XD': "prithivMLmods/Flux-Meme-Xd-LoRA",
+            'Chill Guy': "prithivMLmods/Flux-Chill-Guy-Zone",
+            'Pepe': "openfree/pepe",
+            'NSFWmodel': "lexa862/NSFWmodel",
+            'Claude Art': "strangerzonehf/Flux-Claude-Art",
+            'Open Genmoji': "EvanZhouDev/open-genmoji",
+            'EBook Creative Cover': "prithivMLmods/EBook-Creative-Cover-Flux-LoRA",
+            'Flux Logo Design 2': "prithivMLmods/Logo-Design-Flux-LoRA",
+            'Isometric 3D': "strangerzonehf/Flux-Isometric-3D-LoRA",
+            'Flux Condensation': "fofr/flux-condensation",
+            'Flux Handwriting': "fofr/flux-handwriting",
+            'Shou Xin': "Datou1111/shou_xin",
+            'Sketch Smudge': "strangerzonehf/Flux-Sketch-Smudge-LoRA",
+            '80s Cyberpunk': "fofr/flux-80s-cyberpunk",
+            'Coloring Book Flux': "renderartist/coloringbookflux",
+            'Flux Miniature LoRA': "gokaygokay/Flux-Miniature-LoRA",
+            'Sketch Paint': "strangerzonehf/Sketch-Paint",
+            'Flux UltraRealism 2.0': "prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0",
+            'Midjourney Mix': "strangerzonehf/Flux-Midjourney-Mix-LoRA",
+            'Midjourney Mix 2': "strangerzonehf/Flux-Midjourney-Mix2-LoRA",
+            'Flux Logo Design': "Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design",
+            'Flux Uncensored': "enhanceaiteam/Flux-uncensored",
+            'Flux Uncensored V2': "enhanceaiteam/Flux-Uncensored-V2",
+            'Flux Tarot Cards': "prithivMLmods/Ton618-Tarot-Cards-Flux-LoRA",
+            'Pixel Art Sprites': "sWizad/pokemon-trainer-sprites-pixelart-flux",
+            '3D Sketchfab': "prithivMLmods/Castor-3D-Sketchfab-Flux-LoRA",
+            'Retro Comic Flux': "renderartist/retrocomicflux",
+            'Caricature': "TheAwakenOne/caricature",
+            'Huggieverse': "Chunte/flux-lora-Huggieverse",
+            'Propaganda Poster': "AlekseyCalvin/Propaganda_Poster_Schnell_by_doctor_diffusion",
+            'Flux Game Assets V2': "gokaygokay/Flux-Game-Assets-LoRA-v2",
+            'SDXL HS Card Style': "Norod78/sdxl-hearthstone-card-style-lora",
+            'SLDR FLUX NSFW v2 Studio': "xey/sldr_flux_nsfw_v2-studio",
+            'SoftPasty Flux': "alvdansen/softpasty-flux-dev",
+            'Flux Stickers': "diabolic6045/Flux_Sticker_Lora",
+            'Flux Animex V2': "strangerzonehf/Flux-Animex-v2-LoRA",
+            'Flux Animeo V1': "strangerzonehf/Flux-Animeo-v1-LoRA",
+            'Movie Board': "prithivMLmods/Flux.1-Dev-Movie-Boards-LoRA",
+            'Purple Dreamy': "prithivMLmods/Purple-Dreamy-Flux-LoRA",
+            'PS1 Style Flux': "veryVANYA/ps1-style-flux",
+            'Softserve Anime': "alvdansen/softserve_anime",
+            'Flux Tarot v1': "multimodalart/flux-tarot-v1",
+            'Half Illustration': "davisbro/half_illustration",
+            'OpenDalle v1.1': "dataautogpt3/OpenDalleV1.1",
+            'Flux Ghibsky Illustration': "aleksa-codes/flux-ghibsky-illustration",
+            'Flux Koda': "alvdansen/flux-koda",
+            'Soviet Diffusion XL': "openskyml/soviet-diffusion-xl",
+            'Flux Realism LoRA': "XLabs-AI/flux-RealismLora",
+            'Frosting Lane Flux': "alvdansen/frosting_lane_flux",
+            'Phantasma Anime': "alvdansen/phantasma-anime",
+            'Boreal': "kudzueye/Boreal",
+            'How2Draw': "glif/how2draw",
+            'Flux AestheticAnime': "dataautogpt3/FLUX-AestheticAnime",
+            'Fashion Hut Modeling LoRA': "prithivMLmods/Fashion-Hut-Modeling-LoRA",
+            'Flux SyntheticAnime': "dataautogpt3/FLUX-SyntheticAnime",
+            'Flux Midjourney Anime': "brushpenbob/flux-midjourney-anime",
+            'Coloring Book Generator': "robert123231/coloringbookgenerator",
+            'Collage Flux': "prithivMLmods/Castor-Collage-Dim-Flux-LoRA",
+            'Flux Product Ad Backdrop': "prithivMLmods/Flux-Product-Ad-Backdrop",
+            'Product Design': "multimodalart/product-design",
+            '90s Anime Art': "glif/90s-anime-art",
+            'Brain Melt Acid Art': "glif/Brain-Melt-Acid-Art",
+            'Lustly Flux Uncensored v1': "lustlyai/Flux_Lustly.ai_Uncensored_nsfw_v1",
+            'NSFW Master Flux': "Keltezaa/NSFW_MASTER_FLUX",
+            'Flux Outfit Generator': "tryonlabs/FLUX.1-dev-LoRA-Outfit-Generator",
+            'Midjourney': "Jovie/Midjourney",
+            'DreamPhotoGASM': "Yntec/DreamPhotoGASM",
+            'Flux Super Realism LoRA': "strangerzonehf/Flux-Super-Realism-LoRA",
+            'Stable Diffusion 2-1': "stabilityai/stable-diffusion-2-1-base",
+            'Stable Diffusion 3.5 Large': "stabilityai/stable-diffusion-3.5-large",
+            'Stable Diffusion 3.5 Large Turbo': "stabilityai/stable-diffusion-3.5-large-turbo",
+            'Stable Diffusion 3 Medium': "stabilityai/stable-diffusion-3-medium-diffusers",
+            'Duchaiten Real3D NSFW XL': "stablediffusionapi/duchaiten-real3d-nsfw-xl",
+            'Pixel Art XL': "nerijs/pixel-art-xl",
+            'Character Design': "KappaNeuro/character-design",
+            'Sketched Out Manga': "alvdansen/sketchedoutmanga",
+            'Archfey Anime': "alvdansen/archfey_anime",
+            'Lofi Cuties': "alvdansen/lofi-cuties",
+            'YiffyMix': "Yntec/YiffyMix",
+            'Analog Madness Realistic v7': "digiplay/AnalogMadness-realistic-model-v7",
+            'Selfie Photography': "artificialguybr/selfiephotographyredmond-selfie-photography-lora-for-sdxl",
+            'Filmgrain': "artificialguybr/filmgrain-redmond-filmgrain-lora-for-sdxl",
+            'Leonardo AI Style Illustration': "goofyai/Leonardo_Ai_Style_Illustration",
+            'Cyborg Style XL': "goofyai/cyborg_style_xl",
+            'Little Tinies': "alvdansen/littletinies",
+            'NSFW XL': "Dremmar/nsfw-xl",
+            'Analog Redmond': "artificialguybr/analogredmond",
+            'Pixel Art Redmond': "artificialguybr/PixelArtRedmond",
+            'Ascii Art': "CiroN2022/ascii-art",
+            'Analog': "Yntec/Analog",
+            'Maple Syrup': "Yntec/MapleSyrup",
+            'Perfect Lewd Fantasy': "digiplay/perfectLewdFantasy_v1.01",
+            'AbsoluteReality 1.8.1': "digiplay/AbsoluteReality_v1.8.1",
+            'Disney': "goofyai/disney_style_xl",
+            'Redmond SDXL': "artificialguybr/LogoRedmond-LogoLoraForSDXL-V2",
+            'epiCPhotoGasm': "Yntec/epiCPhotoGasm"
+        }
+        
+        # Add special prompt prefixes for certain models
+        prompt_prefixes = {
+            'Animagine 4.0': "masterpiece, high score, great score, absurdres, ",
+            'Flux Icon Kit': "Icon Kit, ",
+            'Pixel Background': "Pixel Background, ",
+            'Meme XD': "meme, ",
+            'Chill Guy': "chill guy, ",
+            'Pepe': "pepe, ",
+            'NSFWmodel': "nude, ",
+            'Claude Art': "claude art, ",
+            'Open Genmoji': "emoji, ",
+            'EBook Creative Cover': "EBook Cover, ",
+            'Flux Logo Design 2': "Logo Design, ",
+            'Isometric 3D': "Isometric 3D, ",
+            'Flux Condensation': "CONDENSATION, ",
+            'Flux Handwriting': "HWRIT handwriting, ",
+            'Shou Xin': "shou_xin, pencil sketch, ",
+            'Sketch Smudge': "Sketch Smudge, ",
+            '80s Cyberpunk': "80s cyberpunk, ",
+            'Coloring Book Flux': "c0l0ringb00k, coloring book, coloring book page, ",
+            'Flux Miniature LoRA': "MNTR, miniature drawing, ",
+            'Sketch Paint': "Sketch paint, ",
+            'Flux UltraRealism 2.0': "Ultra realistic, ",
+            'Midjourney Mix': "midjourney mix, ",
+            'Midjourney Mix 2': "MJ v6, ",
+            'Flux Logo Design': "wablogo, logo, Minimalist, ",
+            'Flux Tarot Cards': "Tarot card, ",
+            'Pixel Art Sprites': "a pixel image, ",
+            '3D Sketchfab': "3D Sketchfab, ",
+            'Retro Comic Flux': "c0m1c, comic book panel, ",
+            'Caricature': "CCTUR3, ",
+            'Huggieverse': "HGGRE, ",
+            'Propaganda Poster': "propaganda poster, ",
+            'Flux Game Assets V2': "wbgmsst, white background, ",
+            'SDXL HS Card Style': "Hearthstone Card, ",
+            'SoftPasty Flux': "araminta_illus illustration style, ",
+            'Flux Stickers': "5t1cker 5ty1e, ",
+            'Flux Animex V2': "Animex, ",
+            'Flux Animeo V1': "Animeo, ",
+            'Movie Board': "movieboard, ",
+            'Purple Dreamy': "Purple Dreamy, ",
+            'PS1 Style Flux': "ps1 game screenshot, ",
+            'Softserve Anime': "sftsrv style illustration, ",
+            'Flux Tarot v1': "in the style of TOK a trtcrd tarot style, ",
+            'Half Illustration': "in the style of TOK, ",
+            'Flux Ghibsky Illustration': "GHIBSKY style, ",
+            'Flux Koda': "flmft style, ",
+            'Soviet Diffusion XL': "soviet poster, ",
+            'Frosting Lane Flux': "frstingln illustration, ",
+            'Boreal': "photo, ",
+            'How2Draw': "How2Draw, ",
+            'Fashion Hut Modeling LoRA': "Modeling of, ",
+            'Flux SyntheticAnime': "1980s anime screengrab, VHS quality, syntheticanime, ",
+            'Flux Midjourney Anime': "egmid, ",
+            'Collage Flux': "collage, ",
+            'Flux Product Ad Backdrop': "Product Ad, ",
+            'Product Design': "product designed by prdsgn, ",
+            'Brain Melt Acid Art': "maximalism, in an acid surrealism style, ",
+            'NSFW Master Flux': "NSFW, ",
+            'Stable Diffusion 3 Medium': "A, ",
+            'Pixel Art XL': "pixel art, ",
+            'Character Design': "Character Design, ",
+            'Sketched Out Manga': "daiton, ",
+            'Selfie Photography': "instagram model, discord profile picture, ",
+            'Filmgrain': "Film Grain, FilmGrainAF, ",
+            'Leonardo AI Style Illustration': "leonardo style, illustration, vector art, ",
+            'Cyborg Style XL': "cyborg style, ",
+            'Analog Redmond': "timeless style, ",
+            'Pixel Art Redmond': "Pixel Art, ",
+            'Ascii Art': "ascii art, ",
+            'Disney': "Disney style, "
+        }
+        
+        # Get the model path from the mapping
+        model_path = model_mapping.get(model, "black-forest-labs/FLUX.1-schnell")  # Default to FLUX.1 Schnell if not found
+        
+        # Add model-specific prefixes to prompt if needed
+        if model in prompt_prefixes:
+            prompt = f"{prompt_prefixes[model]}{prompt}"
+    
+    print(f"Model path: {model_path}")  # Debug log
 
     # Define the payload for the request
     payload = {
@@ -351,9 +299,21 @@ def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7,
     }
     print(f"Payload: {json.dumps(payload, indent=2)}")  # Debug log
 
-    # Make a request to the API to generate the image
+    # Select the appropriate API handler based on the provider
+    provider_handlers = {
+        "hf-inference": query_hf_inference,
+        "fal-ai": query_fal_ai,
+        "nebius": query_nebius,
+        "replicate": query_replicate,
+        "together": query_together
+    }
+    
+    # Get the appropriate handler for the selected provider
+    handler = provider_handlers.get(provider, query_hf_inference)
+    
+    # Make a request to the API to generate the image using the selected handler
     try:
-        response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout)
+        response = handler(prompt, model_path, headers, payload, timeout)
         print(f"Response status code: {response.status_code}")  # Debug log
     except requests.exceptions.RequestException as e:
         # Log any request exceptions and raise an error for the user
@@ -535,18 +495,26 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme_5') as dalle:
     # Tab for advanced settings
     with gr.Tab("Advanced Settings"):
         with gr.Row():
-            # Textbox for specifying elements to exclude from the image
-            negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="What should not be in the image", value="(deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, misspellings, typos", lines=3, elem_id="negative-prompt-text-input")
+            # Provider selection
+            provider_radio = gr.Radio(
+                choices=["hf-inference", "fal-ai", "nebius", "replicate", "together"],
+                value="hf-inference",
+                label="Inference Provider",
+                info="Select the provider to use for image generation"
+            )
         with gr.Row():
-            # New: BYOK (Bring Your Own Key) textbox
+            # BYOK (Bring Your Own Key) textbox
             byok_textbox = gr.Textbox(
                 value="",
                 label="BYOK (Bring Your Own Key)",
-                info="Enter a custom Hugging Face API key here. When provided, this key will be used instead of the default keys.",
-                placeholder="Enter your Hugging Face API token",
+                info="Enter a custom API key here. When provided, this key will be used instead of the default keys.",
+                placeholder="Enter your API token",
                 type="password",  # Hide the API key for security
                 elem_id="byok-input"
             )
+        with gr.Row():
+            # Textbox for specifying elements to exclude from the image
+            negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="What should not be in the image", value="(deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, misspellings, typos", lines=3, elem_id="negative-prompt-text-input")
         with gr.Row():
             # Slider for selecting the image width
             width = gr.Slider(label="Width", value=1024, minimum=64, maximum=1216, step=32)
@@ -568,6 +536,23 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme_5') as dalle:
             # Radio buttons for selecting the sampling method
             method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "Euler", "Euler a", "Heun", "DDIM"])
 
+        # Function to validate provider selection
+        def validate_provider(api_key, provider):
+            # If custom key is provided, any provider can be used
+            if api_key.strip():
+                return gr.update(value=provider, interactive=True)
+            # If no custom key is provided, only HF Inference is allowed
+            elif provider != "hf-inference":
+                return gr.update(value="hf-inference", interactive=True)
+            return gr.update(value=provider, interactive=True)
+
+        # Connect the BYOK textbox to validate provider selection
+        byok_textbox.change(
+            fn=validate_provider,
+            inputs=[byok_textbox, provider_radio],
+            outputs=provider_radio
+        )
+
     # Tab for image editing options
     with gr.Tab("Image Editor"):
         # Function to simulate a delay for processing
@@ -625,16 +610,46 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme_5') as dalle:
             """
             )
 
+        # Accordion providing an overview of providers
+        with gr.Accordion("Providers Information", open=False):
+            gr.Markdown(
+            """
+            ## Inference Providers
+            
+            This application supports multiple inference providers for image generation:
+            
+            ### HF Inference
+            ###### The default Hugging Face Inference API. This is a serverless solution that allows you to run model inference on Hugging Face's infrastructure. Free to use with limitations.
+            
+            ### Fal AI
+            ###### Fal AI provides optimized inference for image generation models. May offer better performance for certain model types.
+            
+            ### Nebius
+            ###### Nebius AI is a technology company specializing in AI-centric cloud platforms, offering scalable GPU clusters for inference.
+            
+            ### Replicate
+            ###### Replicate's API makes it easy to run machine learning models in the cloud with a simple API.
+            
+            ### Together
+            ###### Together AI offers high-performance inference for various AI models.
+            
+            When no custom API key is provided, only HF Inference can be used. For other providers, you must bring your own API key.
+            """
+            )
+
         # Accordion providing an overview of advanced settings
         with gr.Accordion("Advanced Settings Overview", open=False):
             gr.Markdown(
             """
+            ## Inference Provider
+            ###### Select which provider to use for image generation. Each provider has different performance characteristics and model availability.
+            
+            ## BYOK (Bring Your Own Key)
+            ###### This allows you to use your own API key instead of the default keys. Enter your key here for direct access to models using your account's permissions and rate limits.
+            
             ## Negative Prompt
             ###### This box is for telling the AI what you don't want in your images. Think of it as a way to avoid certain elements. For instance, if you don't want blurry images or extra limbs showing up, this is where you'd mention it.
 
-            ## BYOK (Bring Your Own Key)
-            ###### This allows you to use your own Hugging Face API key instead of the default keys. Enter your key here for direct access to models using your account's permissions and rate limits.
-
             ## Width & Height
             ###### These sliders allow you to specify the resolution of your image. Default value is 1024x1024, and maximum output is 1216x1216.
 
@@ -664,8 +679,8 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme_5') as dalle:
     with gr.Row():
         image_output = gr.Image(type="pil", label="Image Output", elem_id="gallery")
 
-    # Set up button click event to call the query function with the added BYOK parameter
-    text_button.click(query, inputs=[text_prompt, model, custom_lora, negative_prompt, steps, cfg, method, seed, strength, width, height, byok_textbox], outputs=image_output)
+    # Set up button click event to call the query function with the provider and BYOK parameters
+    text_button.click(query, inputs=[text_prompt, model, custom_lora, negative_prompt, steps, cfg, method, seed, strength, width, height, byok_textbox, provider_radio], outputs=image_output)
 
 print("Launching Gradio interface...")  # Debug log
 # Launch the Gradio interface without showing the API or sharing externally