Replace static model list with function for removing extension and id

app.py

@@ -10,54 +10,6 @@ import PIL
 from PIL.ExifTags import TAGS
 import html
 
-model_names = {
-    'absolutereality_V16': 'absolutereality_V16.safetensors [37db0fc3]',
-    'absolutereality_v181': 'absolutereality_v181.safetensors [3d9d4d2b]',
-    'analog-diffusion-1.0': 'analog-diffusion-1.0.ckpt [9ca13f02]',
-    'anythingv3_0-pruned': 'anythingv3_0-pruned.ckpt [2700c435]',
-    'anything-v4.5-pruned': 'anything-v4.5-pruned.ckpt [65745d25]',
-    'anythingV5_PrtRE': 'anythingV5_PrtRE.safetensors [893e49b9]',
-    'AOM3A3_orangemixs': 'AOM3A3_orangemixs.safetensors [9600da17]',
-    'childrensStories_v13D': 'childrensStories_v13D.safetensors [9dfaabcb]',
-    'childrensStories_v1SemiReal': 'childrensStories_v1SemiReal.safetensors [a1c56dbb]',
-    'childrensStories_v1ToonAnime': 'childrensStories_v1ToonAnime.safetensors [2ec7b88b]',
-    'cyberrealistic_v33': 'cyberrealistic_v33.safetensors [82b0d085]',
-    'deliberate_v2': 'deliberate_v2.safetensors [10ec4b29]',
-    'deliberate_v3': 'deliberate_v3.safetensors [afd9d2d4]',
-    'dreamlike-anime-1.0': 'dreamlike-anime-1.0.safetensors [4520e090]',
-    'dreamlike-diffusion-1.0': 'dreamlike-diffusion-1.0.safetensors [5c9fd6e0]',
-    'dreamlike-photoreal-2.0': 'dreamlike-photoreal-2.0.safetensors [fdcf65e7]',
-    'dreamshaper_6BakedVae': 'dreamshaper_6BakedVae.safetensors [114c8abb]',
-    'dreamshaper_7': 'dreamshaper_7.safetensors [5cf5ae06]',
-    'dreamshaper_8': 'dreamshaper_8.safetensors [9d40847d]',
-    'edgeOfRealism_eorV20': 'edgeOfRealism_eorV20.safetensors [3ed5de15]',
-    'EimisAnimeDiffusion_V1': 'EimisAnimeDiffusion_V1.ckpt [4f828a15]',
-    'elldreths-vivid-mix': 'elldreths-vivid-mix.safetensors [342d9d26]',
-    'epicrealism_naturalSinRC1VAE': 'epicrealism_naturalSinRC1VAE.safetensors [90a4c676]',
-    'ICantBelieveItsNotPhotography_seco': 'ICantBelieveItsNotPhotography_seco.safetensors [4e7a3dfd]',
-    'juggernaut_aftermath': 'juggernaut_aftermath.safetensors [5e20c455]',
-    'lyriel_v16': 'lyriel_v16.safetensors [68fceea2]',
-    'mechamix_v10': 'mechamix_v10.safetensors [ee685731]',
-    'meinamix_meinaV9': 'meinamix_meinaV9.safetensors [2ec66ab0]',
-    'meinamix_meinaV11': 'meinamix_meinaV11.safetensors [b56ce717]',
-    'openjourney_V4': 'openjourney_V4.ckpt [ca2f377f]',
-    'portraitplus_V1.0': 'portraitplus_V1.0.safetensors [1400e684]',
-    'Realistic_Vision_V1.4-pruned-fp16': 'Realistic_Vision_V1.4-pruned-fp16.safetensors [8d21810b]',
-    'Realistic_Vision_V2.0': 'Realistic_Vision_V2.0.safetensors [79587710]',
-    'Realistic_Vision_V4.0': 'Realistic_Vision_V4.0.safetensors [29a7afaa]',
-    'Realistic_Vision_V5.0': 'Realistic_Vision_V5.0.safetensors [614d1063]',
-    'redshift_diffusion-V10': 'redshift_diffusion-V10.safetensors [1400e684]',
-    'revAnimated_v122': 'revAnimated_v122.safetensors [3f4fefd9]',
-    'rundiffusionFX25D_v10': 'rundiffusionFX25D_v10.safetensors [cd12b0ee]',
-    'rundiffusionFX_v10': 'rundiffusionFX_v10.safetensors [cd4e694d]',
-    'sdv1_4': 'sdv1_4.ckpt [7460a6fa]',
-    'v1-5-pruned-emaonly': 'v1-5-pruned-emaonly.safetensors [d7049739]',
-    'shoninsBeautiful_v10': 'shoninsBeautiful_v10.safetensors [25d8c546]',
-    'theallys-mix-ii-churned': 'theallys-mix-ii-churned.safetensors [5d9225a4]',
-    'timeless-1.0': 'timeless-1.0.ckpt [7c4971d4]',
-    'toonyou_beta6': 'toonyou_beta6.safetensors [980f6b15]'
-}
-
 
 class Prodia:
     def __init__(self, api_key, base=None):
@@ -128,6 +80,16 @@ def image_to_base64(image_path):
 
     return img_str.decode('utf-8')  # Convert bytes to string
 
+def remove_id_and_ext(text):
+    text = re.sub(r'\[.*\]$', '', text)
+    extension = text[-12:].strip()
+    print(extension)
+    if extension == "safetensors":
+        text = text[:-13]
+    elif extension == "ckpt":
+        text = text[:-4]
+    return text
+
 def get_data(text):
     results = {}
     patterns = {
@@ -193,6 +155,12 @@ def send_to_txt2img(image):
 
 
 prodia_client = Prodia(api_key=os.getenv("PRODIA_API_KEY"))
+model_list = prodia_client.list_models()
+model_names = {}
+
+for model_name in model_list:
+    name_without_ext = remove_id_and_ext(model_name)
+    model_names[name_without_ext] = model_name
 
 def flip_text(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
     result = prodia_client.generate({