add authentication

app.py

@@ -279,40 +279,101 @@ def create_demo():
             ]
         )
         
-        # Function to update UI based on authentication status (called on demo.load)
-        def handle_auth_change_and_ui_setup(profile: gr.OAuthProfile | None, current_user_pref_show_all: bool):
-            is_authorized = check_authorization(profile)
-            
-            if is_authorized:
-                # Authorized User
-                current_choices = list(all_models_list) if current_user_pref_show_all else list(preferred_models)
-                current_value = AUTHENTICATED_DEFAULT_MODEL_ID # Reset to default for authed user
-                # You could add logic here to try and preserve model_choice.value if it's valid in current_choices
+        # Handle "Show Additional Models" checkbox change
+        def persist_preference_and_toggle_models_visibility(show_all_checkbox_val: bool, current_model_id: str, is_authorized: bool):
+            if not is_authorized: # Should not be triggered if checkbox is not interactive
+                mdl_choice_update = gr.Dropdown.update(choices=UNAUTHENTICATED_MODEL_CHOICES, value=UNAUTHENTICATED_DEFAULT_MODEL_ID)
+                return show_all_checkbox_val, mdl_choice_update # Persist the (likely False) checkbox value
 
-                model_dropdown_update = gr.Dropdown.update(choices=current_choices, value=current_value, label="Select Model", interactive=True)
-                accordion_update = gr.Accordion.update(visible=True)
-                model_info_text_update = get_model_info_markdown(current_value, True)
-                show_all_checkbox_update = gr.Checkbox.update(value=current_user_pref_show_all, interactive=True)
-            else:
-                # Unauthenticated User
-                model_dropdown_update = gr.Dropdown.update(choices=UNAUTHENTICATED_MODEL_CHOICES, value=UNAUTHENTICATED_DEFAULT_MODEL_ID, label="Select Model (Free Tier)", interactive=True)
-                accordion_update = gr.Accordion.update(visible=False)
-                model_info_text_update = get_model_info_markdown(UNAUTHENTICATED_DEFAULT_MODEL_ID, False)
-                show_all_checkbox_update = gr.Checkbox.update(value=False, interactive=False)
+            # Authorized user
+            new_choices = list(all_models_list) if show_all_checkbox_val else list(preferred_models)
             
-            return is_authorized, model_dropdown_update, accordion_update, model_info_text_update, show_all_checkbox_update
+            new_model_value = current_model_id
+            if not any(val == current_model_id for _, val in new_choices): # If current model not in new list
+                new_model_value = AUTHENTICATED_DEFAULT_MODEL_ID # Revert to default
+                if not any(val == new_model_value for _, val in new_choices): # If default not in new list (edge case)
+                    new_model_value = new_choices[0][1] if new_choices else None
 
-        demo.load(
-            fn=handle_auth_change_and_ui_setup,
-            inputs=[user_show_all_models_preference_state], # Pass the state that holds user's preference
-            outputs=[
-                auth_status_state,
-                model_choice,
-                advanced_settings_accordion,
-                model_info,
-                show_all_models_checkbox
-            ]
+
+            mdl_choice_update = gr.Dropdown.update(choices=new_choices, value=new_model_value)
+            return show_all_checkbox_val, mdl_choice_update # Return new preference and dropdown update
+
+        show_all_models_checkbox.change(
+            fn=persist_preference_and_toggle_models_visibility,
+            inputs=[show_all_models_checkbox, model_choice, auth_status_state],
+            outputs=[user_show_all_models_preference_state, model_choice]
+        )
+
+        # Update model info when model selection changes
+        def update_model_info_display(selected_model_id: str, is_authorized: bool):
+            return get_model_info_markdown(selected_model_id, is_authorized)
+
+        model_choice.change(
+            fn=update_model_info_display,
+            inputs=[model_choice, auth_status_state],
+            outputs=[model_info]
         )
+        
+        # Handle file uploads
+        def handle_upload(files, current_paths, current_filenames):
+            # ... (your existing function, no changes needed here)
+            file_paths = []
+            file_names = []
+            if files:
+                for file in files:
+                    file_paths.append(file.name)
+                    file_names.append(get_base_filename(file.name))
+            return file_paths, file_paths, file_names, 0, None, "0 of 0", "Upload images and click 'Generate Alt-Text'."
+
+
+        upload_button.upload(
+            fn=handle_upload,
+            inputs=[upload_button, image_state, filename_state],
+            outputs=[input_gallery, image_state, filename_state,
+                     current_index, current_image, image_counter, analysis_text]
+        )
+
+        # Analyze images
+        def analyze_images(image_paths, current_model_choice_id, length_choice, filenames, content_type_choice):
+            # Note: current_model_choice_id will be correctly set based on auth status by the UI
+            if not image_paths:
+                return [], [], 0, None, "0 of 0", "No images uploaded to analyze.", None
+
+            is_photography = content_type_choice == "Photography"
+            sys_prompt = get_sys_prompt(length_choice, photograph=is_photography)
+            image_results = []
+            analysis_progress = gr.Progress(track_tqdm=True)
+
+            for i, image_path in enumerate(analysis_progress.tqdm(image_paths, desc="Analyzing Images")):
+                image_id = filenames[i] if i < len(filenames) and filenames[i] else f"Image_{i+1}_{os.path.basename(image_path)}"
+                try:
+                    img = Image.open(image_path)
+                    prompt0 = prompt_new() # Assuming prompt_new() is defined in your utils
+                    model_name = current_model_choice_id 
+                    client_to_use = OR # Ensure OR can handle "meta-llama/llama-4-maverick:free"
+                                     # Or add logic here: if model_name == UNAUTHENTICATED_DEFAULT_MODEL_ID: client_to_use = DifferentClient()
+
+                    result = client_to_use.generate_caption(
+                        img, model=model_name, max_image_size=512,
+                        prompt=prompt0, prompt_dev=sys_prompt, temperature=1
+                    )
+                    image_results.append({"image_id": image_id, "content": result.strip()})
+                except FileNotFoundError:
+                     error_message = f"Error: File not found at path '{image_path}'"
+                     print(error_message)
+                     image_results.append({"image_id": image_id, "content": error_message})
+                except Exception as e:
+                    error_message = f"Error processing {image_id}: {str(e)}"
+                    print(error_message)
+                    image_results.append({"image_id": image_id, "content": error_message})
+
+            csv_path = create_csv_file_simple(image_results)
+            initial_image = image_paths[0] if image_paths else None
+            initial_counter = f"1 of {len(image_paths)}" if image_paths else "0 of 0"
+            initial_text = image_results[0]["content"] if image_results else "Analysis complete, but no results generated."
+
+            return (image_paths, image_results, 0, initial_image, initial_counter,
+                    initial_text, csv_path)
 
         # Navigate previous
         def go_to_prev(current_idx, images, results):