Update app.py

app.py

@@ -42,6 +42,14 @@ def create_instructions_html():
     </div>
     """
 
+# Get available models with error handling
+try:
+    available_models = get_available_models_secure()
+    print(f"Available models: {available_models}")
+except Exception as e:
+    print(f"Error getting models: {e}")
+    available_models = ["flux", "turbo", "kontext"]  # Fallback
+
 # Create the Gradio interface
 with gr.Blocks(css=get_custom_css(), title=f"{COMPANY_NAME} AI Image Generator") as demo:
     with gr.Row():
@@ -76,7 +84,7 @@ with gr.Blocks(css=get_custom_css(), title=f"{COMPANY_NAME} AI Image Generator")
             
             with gr.Row():
                 model = gr.Dropdown(
-                    choices=get_available_models_secure(),
+                    choices=available_models,
                     value="flux",
                     label="Model"
                 )