Update app.py

app.py

@@ -4,58 +4,28 @@ import io
 import random
 import os
 import time
-import json
 from PIL import Image
+import json
 
 # Project by Nymbo
 
-# Retrieve the default API token from environment variables
-HF_TOKEN = os.getenv("HF_READ_TOKEN")
+# 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}"}
 # Timeout for requests
 timeout = 100
 
-# 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"):
+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, provider="hf-inference", custom_api_key=""):
     # Debug log to indicate function start
     print("Starting query function...")
     # Print the parameters for debugging purposes
     print(f"Prompt: {prompt}")
     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"Parameters - Steps: {steps}, CFG Scale: {cfg_scale}, Seed: {seed}, Strength: {strength}, Width: {width}, Height: {height}")
     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
@@ -72,16 +42,14 @@ 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:
-        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")
+        # If no custom key, check if provider is "hf-inference"
+        if provider != "hf-inference":
+            print(f"ERROR: Custom API key is required for {provider} provider")
+            raise gr.Error(f"A custom API key is required when using the {provider} provider. Please enter your key in the BYOK field.")
+        
+        # For hf-inference, 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")
     
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
@@ -89,200 +57,315 @@ 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 model path based on the selected model or custom LoRA
+    # Set the API URL based on the selected model or custom LoRA
     if custom_lora.strip() != "":
-        model_path = custom_lora.strip()
+        API_URL = f"https://api-inference.huggingface.co/models/{custom_lora.strip()}"
     else:
-        # 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}"
+        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}"
+            
+    # Adjust URL for different providers if needed
+    base_url = "https://api-inference.huggingface.co"
+    if provider == "fal-ai":
+        base_url = "https://api.fal.ai/hf"
+    elif provider == "nebius":
+        base_url = "https://api.nebius.ai/hf"
+    elif provider == "replicate":
+        base_url = "https://api.replicate.com/v1/models"
+        # Replicate has a different endpoint structure
+        if "https://api-inference.huggingface.co/models/" in API_URL:
+            model_id = API_URL.replace("https://api-inference.huggingface.co/models/", "")
+            API_URL = f"{base_url}/{model_id}"
+    elif provider == "together":
+        base_url = "https://api.together.xyz/inference"
+        if "https://api-inference.huggingface.co/models/" in API_URL:
+            model_id = API_URL.replace("https://api-inference.huggingface.co/models/", "")
+            API_URL = f"{base_url}/{model_id}"
     
-    print(f"Model path: {model_path}")  # Debug log
+    # Only update URL if we're not using a custom URL format (like for replicate)
+    if provider != "replicate" and provider != "together" and "https://api-inference.huggingface.co/models/" in API_URL:
+        model_id = API_URL.replace("https://api-inference.huggingface.co/models/", "")
+        API_URL = f"{base_url}/models/{model_id}"
+            
+    print(f"API URL set to: {API_URL}")  # Debug log
+        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
 
     # Define the payload for the request
     payload = {
@@ -299,21 +382,46 @@ def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7,
     }
     print(f"Payload: {json.dumps(payload, indent=2)}")  # Debug log
 
-    # 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
+    # Adjust payload for different providers if needed
+    if provider == "replicate" or provider == "together":
+        # These providers might have different API formats
+        if provider == "replicate":
+            payload = {
+                "version": model_id,
+                "input": {
+                    "prompt": prompt,
+                    "negative_prompt": is_negative,
+                    "num_inference_steps": steps,
+                    "guidance_scale": cfg_scale,
+                    "seed": seed if seed != -1 else random.randint(1, 1000000000),
+                    "strength": strength,
+                    "width": width,
+                    "height": height
+                }
+            }
+        elif provider == "together":
+            payload = {
+                "model": model_id,
+                "prompt": prompt,
+                "negative_prompt": is_negative,
+                "num_inference_steps": steps,
+                "guidance_scale": cfg_scale,
+                "seed": seed if seed != -1 else random.randint(1, 1000000000),
+                "strength": strength,
+                "width": width,
+                "height": height
+            }
+
+    # Make a request to the API to generate the image
     try:
-        response = handler(prompt, model_path, headers, payload, timeout)
+        # Log which provider we're using
+        print(f"Sending request to {provider} provider")
+        
+        # Add provider to headers for HF inference if needed
+        if provider != "hf-inference" and "api-inference.huggingface.co" in API_URL:
+            headers["X-Provider"] = provider
+            
+        response = requests.post(API_URL, headers=headers, json=payload, timeout=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
@@ -338,8 +446,21 @@ def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7,
             raise gr.Error(f"{response.status_code}: An unexpected error occurred.")
 
     try:
-        # Attempt to read the image from the response content
-        image_bytes = response.content
+        # Handle different response formats based on provider
+        if provider == "replicate":
+            # Replicate might return a URL to the image rather than the image itself
+            result = response.json()
+            if "output" in result and isinstance(result["output"], list) and len(result["output"]) > 0:
+                image_url = result["output"][0]
+                # Fetch the image from the URL
+                image_response = requests.get(image_url, timeout=timeout)
+                image_bytes = image_response.content
+            else:
+                raise Exception(f"Unexpected Replicate response format: {result}")
+        else:
+            # Standard response with image content
+            image_bytes = response.content
+            
         image = Image.open(io.BytesIO(image_bytes))
         print(f'Generation {key} completed! ({prompt})')  # Debug log
         return image
@@ -495,26 +616,58 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme_5') as dalle:
     # Tab for advanced settings
     with gr.Tab("Advanced Settings"):
         with gr.Row():
-            # Provider selection
+            # 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():
+            # Provider selection radio buttons
+            providers_list = [
+                "hf-inference",  # Default Hugging Face Inference
+                "fal-ai",        # Fal.ai
+                "nebius",        # Nebius
+                "replicate",     # Replicate
+                "together",      # Together AI
+            ]
+            
             provider_radio = gr.Radio(
-                choices=["hf-inference", "fal-ai", "nebius", "replicate", "together"],
+                choices=providers_list,
                 value="hf-inference",
                 label="Inference Provider",
-                info="Select the provider to use for image generation"
+                info="Select which provider to use for image generation. Providers other than HF Inference require a custom API key.",
+                elem_id="provider-radio"
             )
+        
         with gr.Row():
             # BYOK (Bring Your Own Key) textbox
             byok_textbox = gr.Textbox(
                 value="",
                 label="BYOK (Bring Your Own Key)",
-                info="Enter a custom API key here. When provided, this key will be used instead of the default keys.",
+                info="Enter a custom API key here. Required for all providers except HF Inference.",
                 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")
+            
+        # Function to validate provider selection based on BYOK
+        def validate_provider(api_key, provider):
+            # If no custom API key is provided, only "hf-inference" can be used
+            if not api_key.strip() and provider != "hf-inference":
+                return gr.update(value="hf-inference")
+            return gr.update(value=provider)
+            
+        # Connect the BYOK textbox to validate provider selection
+        byok_textbox.change(
+            fn=validate_provider,
+            inputs=[byok_textbox, provider_radio],
+            outputs=provider_radio
+        )
+        
+        # Also validate provider when the radio changes to ensure consistency
+        provider_radio.change(
+            fn=validate_provider,
+            inputs=[byok_textbox, provider_radio],
+            outputs=provider_radio
+        )
         with gr.Row():
             # Slider for selecting the image width
             width = gr.Slider(label="Width", value=1024, minimum=64, maximum=1216, step=32)
@@ -536,23 +689,6 @@ 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
@@ -610,46 +746,24 @@ 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.
 
+            ## Inference Provider
+            ###### Select which AI provider to use for generating images. Different providers may have different capabilities, pricing, and performance characteristics:
+            ###### - HF Inference: Default Hugging Face inference API (uses application's default keys)
+            ###### - Fal AI: Optimized for low latency and high throughput (requires your own API key)
+            ###### - Nebius: Cloud provider with enterprise-grade infrastructure (requires your own API key)
+            ###### - Replicate: Wide variety of models with flexible deployment (requires your own API key)
+            ###### - Together: High-performance inference service (requires your own API key)
+
+            ## 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. A custom key is required for all providers except HF Inference.
+
             ## Width & Height
             ###### These sliders allow you to specify the resolution of your image. Default value is 1024x1024, and maximum output is 1216x1216.
 
@@ -679,8 +793,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 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)
+    # Set up button click event to call the query function with provider and BYOK parameters
+    text_button.click(query, inputs=[text_prompt, model, custom_lora, negative_prompt, steps, cfg, method, seed, strength, width, height, provider_radio, byok_textbox], outputs=image_output)
 
 print("Launching Gradio interface...")  # Debug log
 # Launch the Gradio interface without showing the API or sharing externally