OAuth

app.py

@@ -126,82 +126,76 @@ def create_demo():
                              show_share_button=False, show_fullscreen_button=False,
                              container=False, elem_id="nml-logo")
         
-        # Store model choices and state
-        show_all_models_state = gr.State(False)
-
-        # Define preferred and additional models directly in the function
-        preferred_models = [
-            ("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:thinking")
-        ]
-        
-        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
-        
-        # Default model value
-        default_model = "google/gemini-2.0-flash-001"
-        
         with gr.Row():
             # Left column: Controls and uploads
             with gr.Column(scale=1):
 
                 ##########################################################################
-                # Function to check authorization
-                def check_authorization(profile: gr.OAuthProfile | None):
-                    if profile is None:
-                        return True
-                    
-                    is_authorized = profile.username not in AUTHORIZED_USER_IDS
-                    return is_authorized
-            
-                #new bit with authroisation
-                # define it false to begin with
-                #auth_state = gr.State(False)
-
-                #demo.load(check_authorization, inputs=None, outputs=[auth_state])
+                # Define preferred and additional models directly in the function
+                preferred_models = [
+                    ("Llama 4 Maverick (free)", "meta-llama/llama-4-maverick:free")
+                ]
 
-                with gr.Group(visible=False) as unauthenticated_ui:
-                    # free version - yeah baby
-                    login_button = gr.LoginButton()
+                login_button = gr.LoginButton(visible=False)
 
-                    # Define preferred and additional models directly in the function
-                    preferred_models = [
-                        ("Llama 4 Maverick (free)", "meta-llama/llama-4-maverick:free")
-                    ]
-
-                    all_models_list = preferred_models
-        
-                    # Default model value
-                    default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
-                demo.load(
-                    fn=check_authorization,
-                    inputs=None,
-                    outputs=[unauthenticated_ui]
-                )
-                ################################## normal stuff below  
                 upload_button = gr.UploadButton(
                     "Click to Upload Images",
                     file_types=["image"],
                     file_count="multiple"
                 )
-                
-                # Model dropdown
+                        
                 model_choice = gr.Dropdown(
                     choices=preferred_models, 
                     label="Select Model",
-                    value=default_model
+                    value="meta-llama/llama-4-maverick:free"
                 )
+
+                default_model = gr.state()
                 
+                # Function to check authorization
+                def check_authorization(profile: gr.OAuthProfile | None):
+                    if profile is None:
+                        return False
+                    
+                    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:thinking")
+                        ]
+                        
+                        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
+                        
+                        # Default model value
+                        default_model = "google/gemini-2.0-flash-001"
+
+                        return gr.Dropdown(choices=preferred_models, label="Select Model",value=default_model),False,default_model
+                    else:
+                        all_models_list = preferred_models
+            
+                        # Default model value
+                        default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
+                        return gr.Dropdown(choices=preferred_models, label="Select Model",value=default_model),True,default_model
+            
+                demo.load(
+                    fn=check_authorization,
+                    inputs=None,
+                    outputs=[model_choice,login_button,default_model]
+                )
+                ################################## normal stuff below  
+
                 length_choice = gr.Radio(
                     choices=["short", "medium", "long"], 
                     label="Response Length",