Update app.py

app.py

@@ -4,28 +4,58 @@ 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, provider="hf-inference", 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
     print(f"Prompt: {prompt}")
     print(f"Model: {model}")
     print(f"Custom LoRA: {custom_lora}")
-    print(f"Provider: {provider}")
     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
@@ -42,14 +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:
-        # 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")
+        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}"}
 
@@ -57,315 +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}"
-            
-    # 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}"
+        # 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}"
     
-    # 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
+    print(f"Model path: {model_path}")  # Debug log
 
     # Define the payload for the request
     payload = {
@@ -382,46 +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
 
-    # 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
+    # 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:
-        # 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)
+        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
@@ -446,21 +338,8 @@ 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:
-        # 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
-            
+        # Attempt to read the image from the response content
+        image_bytes = response.content
         image = Image.open(io.BytesIO(image_bytes))
         print(f'Generation {key} completed! ({prompt})')  # Debug log
         return image
@@ -616,58 +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")
-        
-        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 selection
             provider_radio = gr.Radio(
-                choices=providers_list,
+                choices=["hf-inference", "fal-ai", "nebius", "replicate", "together"],
                 value="hf-inference",
                 label="Inference Provider",
-                info="Select which provider to use for image generation. Providers other than HF Inference require a custom API key.",
-                elem_id="provider-radio"
+                info="Select the provider to use for image generation"
             )
-        
         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. Required for all providers except HF Inference.",
+                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"
             )
-            
-        # 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():
+            # 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)
@@ -689,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
@@ -746,23 +610,45 @@ 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(
             """
-            ## 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)
-
+            ###### 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. A custom key is required for all providers except HF Inference.
+            ###### 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.
 
             ## Width & Height
             ###### These sliders allow you to specify the resolution of your image. Default value is 1024x1024, and maximum output is 1216x1216.
@@ -793,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 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)
+    # 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