Update app.py

app.py

@@ -210,6 +210,19 @@ def process(Prompt, used_model, image_width, image_height, image_ratio, image_se
         raise gr.Error("Unbekannter Fehler beim API-Aufruf! 🤷‍♂️", duration=15)
 
 
+def get_inference_models():
+    models = []
+    if check_api(API_OLD) == 200:
+        models.append("FLUX-API-OLD")
+        selected_model = "FLUX-API-OLD"
+    if check_api(API_V1_IMAGE) == 200:
+        models_new_api = ["FLUX", "FLUX-Realism", "FLUX-Pixel", "FLUX-Anime", "FLUX-3D", "FLUX-Disney", "ANY-DARK", "Stable-Diffusion-XL-Base", "Stable-Diffusion-XL-Lightning"]
+        for model in models_new_api:
+            models.append(model)
+        selected_model = "FLUX"
+    return models, selected_model
+
+
 # MARK: Gradio BLOCKS UI
 with gr.Blocks(theme=theme, head=custom_head, css=custom_css, js=custom_js, title=title) as demo:
     with gr.Row(elem_classes="row-header"):
@@ -231,13 +244,14 @@ with gr.Blocks(theme=theme, head=custom_head, css=custom_css, js=custom_js, titl
                 with gr.Row(elem_classes="image_size_selctor_wrapper"):
                     with gr.Column(scale=1):
                         with gr.Row():
-                            inference_models = ["FLUX-API-OLD"]
-                            selected_model = "FLUX-API-OLD"
-                            models_new_api = ["FLUX", "FLUX-Realism", "FLUX-Pixel", "FLUX-Anime", "FLUX-3D", "FLUX-Disney", "ANY-DARK", "Stable-Diffusion-XL-Base", "Stable-Diffusion-XL-Lightning"]
-                            if check_api(API_V1_IMAGE) == 200:
-                                selected_model = "FLUX"
-                                for model in models_new_api:
-                                    inference_models.append(model)
+                            inference_models, select_model = get_inference_models()
+                            # inference_models = ["FLUX-API-OLD"]
+                            # selected_model = "FLUX-API-OLD"
+                            # models_new_api = ["FLUX", "FLUX-Realism", "FLUX-Pixel", "FLUX-Anime", "FLUX-3D", "FLUX-Disney", "ANY-DARK", "Stable-Diffusion-XL-Base", "Stable-Diffusion-XL-Lightning"]
+                            # if check_api(API_V1_IMAGE) == 200:
+                            #     selected_model = "FLUX"
+                            #     for model in models_new_api:
+                            #         inference_models.append(model)
                             select_model = gr.Dropdown(choices=inference_models, value=selected_model, label="Model", elem_id="select_model", elem_classes="select-model")
                             # with gr.Row():
                             image_width = gr.Number(label="Breite", minimum=256, maximum=MAX_IMAGE_SIZE, value=576, step=32, elem_id="image_width_selector", elem_classes="image-width-selector", scale=1, visible=False)