Rename BFL to Black Forest Labs

lib/api.py

@@ -37,7 +37,7 @@ def txt2img_generate(api_key, service, model, inputs, parameters, **kwargs):
     if service == "Fal":
         headers["Authorization"] = f"Key {api_key}"
 
-    if service == "BFL":
+    if service == "Black Forest Labs":
         headers["x-key"] = api_key
 
     json = {}
@@ -51,7 +51,7 @@ def txt2img_generate(api_key, service, model, inputs, parameters, **kwargs):
         json = {**parameters, **kwargs}
         json["prompt"] = inputs
 
-    if service == "BFL":
+    if service == "Black Forest Labs":
         json = {**parameters, **kwargs}
         json["prompt"] = inputs
 
@@ -75,7 +75,7 @@ def txt2img_generate(api_key, service, model, inputs, parameters, **kwargs):
 
             # BFL is async so we need to poll for result
             # https://api.bfl.ml/docs
-            if service == "BFL":
+            if service == "Black Forest Labs":
                 id = response.json()["id"]
                 url = f"{Config.SERVICES[service]}/get_result?id={id}"
 

lib/config.py

@@ -5,7 +5,7 @@ Config = SimpleNamespace(
     ICON="⚡",
     LAYOUT="wide",
     SERVICES={
-        "BFL": "https://api.bfl.ml/v1",
+        "Black Forest Labs": "https://api.bfl.ml/v1",
         "Fal": "https://fal.run",
         "Hugging Face": "https://api-inference.huggingface.co/models",
         "Perplexity": "https://api.perplexity.ai",
@@ -28,13 +28,13 @@ Config = SimpleNamespace(
     TXT2IMG_NEGATIVE_PROMPT="ugly, unattractive, disfigured, deformed, mutated, malformed, blurry, grainy, noisy, oversaturated, undersaturated, overexposed, underexposed, worst quality, low details, lowres, watermark, signature, autograph, trademark, sloppy, cluttered",
     TXT2IMG_DEFAULT_MODEL={
         # The index of model in below lists
-        "BFL": 2,
+        "Black Forest Labs": 2,
         "Fal": 0,
         "Hugging Face": 2,
     },
     TXT2IMG_MODELS={
         # Model IDs referenced in Text_to_Image.py
-        "BFL": [
+        "Black Forest Labs": [
             "flux-dev",
             "flux-pro",
             "flux-pro-1.1",

pages/2_🎨_Text_to_Image.py

@@ -5,14 +5,15 @@ import streamlit as st
 
 from lib import Config, ModelPresets, txt2img_generate
 
+# The token name is the service in lower_snake_case
 SERVICE_SESSION = {
-    "BFL": "api_key_bfl",
+    "Black Forest Labs": "api_key_black_forest_labs",
     "Fal": "api_key_fal",
     "Hugging Face": "api_key_hugging_face",
 }
 
 SESSION_TOKEN = {
-    "api_key_bfl": os.environ.get("BFL_API_KEY") or None,
+    "api_key_black_forest_labs": os.environ.get("BFL_API_KEY") or None,
     "api_key_fal": os.environ.get("FAL_KEY") or None,
     "api_key_hugging_face": os.environ.get("HF_TOKEN") or None,
 }
@@ -42,8 +43,8 @@ st.set_page_config(
 )
 
 # Initialize Streamlit session state
-if "api_key_bfl" not in st.session_state:
-    st.session_state.api_key_bfl = ""
+if "api_key_black_forest_labs" not in st.session_state:
+    st.session_state.api_key_black_forest_labs = ""
 
 if "api_key_fal" not in st.session_state:
     st.session_state.api_key_fal = ""
@@ -60,13 +61,13 @@ if "txt2img_messages" not in st.session_state:
 if "txt2img_seed" not in st.session_state:
     st.session_state.txt2img_seed = 0
 
-st.logo("logo.svg")
+st.logo("logo.png")
 st.sidebar.header("Settings")
 service = st.sidebar.selectbox(
     "Service",
     options=list(SERVICE_SESSION.keys()),
-    index=1,
     disabled=st.session_state.running,
+    index=2,  # Hugging Face
 )
 
 # Disable API key input and hide value if set by environment variable; handle empty string value later.