more tests

app.py

@@ -80,43 +80,6 @@ def get_base_filename(filepath):
     filename = os.path.splitext(basename)[0]
     return filename
 
- # Function to check authorization
-def check_authorization(profile: gr.OAuthProfile | None):
-    
-    is_authorized = profile.username in AUTHORIZED_USER_IDS
-    if is_authorized:
-        # Define preferred and additional models directly in the function
-        preferred_models_auth = [
-            ("Gemini 2.0 Flash (cheap)", "google/gemini-2.0-flash-001"),
-            ("GPT-4.1 Mini", "gpt-4.1-mini"),
-            ("GPT-4.1 (Recommended)", "gpt-4.1"),
-            ("Claude 3.7 Sonnet", "anthropic/claude-3.7-sonnet"),
-            ("Gemini 2.5 Pro", "google/gemini-2.5-pro-preview-03-25"),
-            ("Gemini 2.5 Flash Thinking (Recommended)","google/gemini-2.5-flash-preview")
-        ]
-        
-        additional_models = [
-            ("GPT-4.1 Nano", "gpt-4.1-nano"),
-            ("ChatGPT Latest", "openai/chatgpt-4o-latest"),
-            ("Llama 4 Maverick", "meta-llama/llama-4-maverick")
-        ]
-        
-        # Calculate all models once
-        all_models_list = preferred_models + additional_models
-
-        text = f"""**Current Model**: Gemini 2.0 Flash (cheap) 
-    **Estimated cost per 100 Images**: {MODEL_PRICING["google/gemini-2.0-flash-001"]}"""
-        
-        return gr.Update(choices=preferred_models_auth, label="Select Model",value="google/gemini-2.0-flash-001"),False,text
-    else:
-        all_models_list = preferred_models
-
-        # Default model value
-        default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
-        text = f"""**Current Model**: Llama 4 Maverick (free)
-    **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
-        return gr.Update(choices=preferred_models, label="Select Model",value=default_model),True,text
-
 # Define the Gradio interface
 def create_demo():
     custom_css = """
@@ -172,6 +135,43 @@ def create_demo():
             with gr.Column(scale=1):
 
                 ##########################################################################
+                # Function to check authorization
+                def check_authorization(profile: gr.OAuthProfile | None):
+                    
+                    is_authorized = profile.username in AUTHORIZED_USER_IDS
+                    if is_authorized:
+                        # Define preferred and additional models directly in the function
+                        preferred_models_auth = [
+                            ("Gemini 2.0 Flash (cheap)", "google/gemini-2.0-flash-001"),
+                            ("GPT-4.1 Mini", "gpt-4.1-mini"),
+                            ("GPT-4.1 (Recommended)", "gpt-4.1"),
+                            ("Claude 3.7 Sonnet", "anthropic/claude-3.7-sonnet"),
+                            ("Gemini 2.5 Pro", "google/gemini-2.5-pro-preview-03-25"),
+                            ("Gemini 2.5 Flash Thinking (Recommended)","google/gemini-2.5-flash-preview")
+                        ]
+                        
+                        additional_models = [
+                            ("GPT-4.1 Nano", "gpt-4.1-nano"),
+                            ("ChatGPT Latest", "openai/chatgpt-4o-latest"),
+                            ("Llama 4 Maverick", "meta-llama/llama-4-maverick")
+                        ]
+                        
+                        # Calculate all models once
+                        all_models_list = preferred_models_auth + additional_models
+                
+                        text = f"""**Current Model**: Gemini 2.0 Flash (cheap) 
+                    **Estimated cost per 100 Images**: {MODEL_PRICING["google/gemini-2.0-flash-001"]}"""
+                        
+                        return gr.Update(choices=preferred_models_auth, label="Select Model",value="google/gemini-2.0-flash-001"),False,text
+                    else:
+                        all_models_list = preferred_models
+                
+                        # Default model value
+                        default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
+                        text = f"""**Current Model**: Llama 4 Maverick (free)
+                    **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
+                        return gr.Update(choices=preferred_models, label="Select Model",value=default_model),True,text
+                        
                 # Define preferred and additional models directly in the function
                 preferred_models = [
                     ("Llama 4 Maverick (free)", "meta-llama/llama-4-maverick:free")