Move txt2img aspect ratios to config

lib/config.py

@@ -5,13 +5,21 @@ Config = SimpleNamespace(
     ICON="⚡",
     LAYOUT="wide",
     TXT2IMG_NEGATIVE_PROMPT="ugly, bad, asymmetrical, malformed, mutated, disgusting, blurry, grainy, oversaturated, undersaturated, overexposed, underexposed, worst quality, low details, lowres, watermark, signature, autograph, trademark",
-    TXT2IMG_DEFAULT_INDEX=2,
+    TXT2IMG_DEFAULT_MODEL=2,
     TXT2IMG_MODELS=[
         "black-forest-labs/flux.1-dev",
         "black-forest-labs/flux.1-schnell",
         "stabilityai/stable-diffusion-xl-base-1.0",
     ],
-    TXT2TXT_DEFAULT_INDEX=4,
+    TXT2IMG_DEFAULT_AR="1:1",
+    TXT2IMG_AR={
+        "9:7": (1152, 896),
+        "7:4": (1344, 768),
+        "1:1": (1024, 1024),
+        "4:7": (768, 1344),
+        "7:9": (896, 1152),
+    },
+    TXT2TXT_DEFAULT_MODEL=4,
     TXT2TXT_MODELS=[
         "codellama/codellama-34b-instruct-hf",
         "meta-llama/llama-2-13b-chat-hf",

pages/1_💬_Text_Generation.py

@@ -42,7 +42,7 @@ model = st.sidebar.selectbox(
     "Model",
     placeholder="Select a model",
     format_func=lambda x: x.split("/")[1],
-    index=Config.TXT2TXT_DEFAULT_INDEX,
+    index=Config.TXT2TXT_DEFAULT_MODEL,
     options=Config.TXT2TXT_MODELS,
 )
 max_tokens = st.sidebar.slider(

pages/2_🎨_Text_to_Image.py

@@ -13,13 +13,6 @@ from lib import Config, Presets
 HF_TOKEN = os.environ.get("HF_TOKEN")
 API_URL = "https://api-inference.huggingface.co/models"
 HEADERS = {"Authorization": f"Bearer {HF_TOKEN}", "X-Wait-For-Model": "true", "X-Use-Cache": "false"}
-SIZE_AR = {
-    "9:7": (1152, 896),
-    "7:4": (1344, 768),
-    "1:1": (1024, 1024),
-    "4:7": (768, 1344),
-    "7:9": (896, 1152),
-}
 PRESET_MODEL = {
     "black-forest-labs/flux.1-dev": Presets.FLUX_1_DEV,
     "black-forest-labs/flux.1-schnell": Presets.FLUX_1_SCHNELL,
@@ -66,12 +59,12 @@ model = st.sidebar.selectbox(
     "Model",
     format_func=lambda x: x.split("/")[1],
     options=Config.TXT2IMG_MODELS,
-    index=Config.TXT2IMG_DEFAULT_INDEX,
+    index=Config.TXT2IMG_DEFAULT_MODEL,
 )
 aspect_ratio = st.sidebar.select_slider(
     "Aspect Ratio",
-    options=list(SIZE_AR.keys()),
-    value=list(SIZE_AR.keys())[1],
+    options=list(Config.TXT2IMG_AR.keys()),
+    value=Config.TXT2IMG_DEFAULT_AR,
 )
 
 # heading
@@ -85,9 +78,9 @@ parameters = {}
 preset = PRESET_MODEL[model]
 for param in preset["parameters"]:
     if param == "width":
-        parameters[param] = SIZE_AR[aspect_ratio][0]
+        parameters[param] = Config.TXT2IMG_AR[aspect_ratio][0]
     if param == "height":
-        parameters[param] = SIZE_AR[aspect_ratio][1]
+        parameters[param] = Config.TXT2IMG_AR[aspect_ratio][1]
     if param == "guidance_scale":
         parameters[param] = st.sidebar.slider(
             "Guidance Scale",