Update app.py

app.py

@@ -61,6 +61,15 @@ theme = gr.themes.Soft(
 title = "Bilder Builder"
 
 
+def check_api(url):
+    try:
+        response = requests.get(url, timeout=2)
+        return response.status_code
+    except requests.exceptions.RequestException as e:
+        print(f"An error occurred: {e}")
+        return None
+
+    
 # MARK: GET PREV. IMAGES
 def get_gallery_images(dirpath):
     gallery_images = [dirpath + "/" + s for s in os.listdir(dirpath) if os.path.isfile(os.path.join(dirpath, s))]
@@ -222,8 +231,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", "FLUX", "FLUX-Realism", "FLUX-Pixel", "FLUX-Anime", "FLUX-3D", "FLUX-Disney", "ANY-DARK", "Stable-Diffusion-XL-Base", "Stable-Diffusion-XL-Lightning"]
-                            select_model = gr.Dropdown(choices=inference_models, value="FLUX", label="Model", elem_id="select_model", elem_classes="select-model")
+                            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) == 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)
                             image_height = gr.Number(label="Höhe", minimum=256, maximum=MAX_IMAGE_SIZE, value=1024, step=32, elem_id="image_height_selector", elem_classes="image-height-selector", scale=1, visible=False)