OAuth

app.py

@@ -150,6 +150,7 @@ def create_demo():
                     value="meta-llama/llama-4-maverick:free"
                 )
                 
+                
                 # Function to check authorization
                 def check_authorization(profile: gr.OAuthProfile | None):
                     if profile is None:
@@ -175,22 +176,20 @@ def create_demo():
                         
                         # 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.Dropdown(choices=preferred_models, label="Select Model",value="google/gemini-2.0-flash-001"),False
+                        return gr.Dropdown(choices=preferred_models, 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
-                        return gr.Dropdown(choices=preferred_models, label="Select Model",value=default_model),True
+                        text = f"""**Current Model**: Llama 4 Maverick (free)
+                    **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
+                        return gr.Dropdown(choices=preferred_models, label="Select Model",value=default_model),True,text
             
-                demo.load(
-                    fn=check_authorization,
-                    inputs=None,
-                    outputs=[model_choice,login_button]
-                )
-                ################################## normal stuff below  
-
                 length_choice = gr.Radio(
                     choices=["short", "medium", "long"], 
                     label="Response Length",
@@ -212,18 +211,15 @@ def create_demo():
                         value="Museum Object"
                     )
                 
-                # Find the default model's display name
-                default_model_name = "Unknown Model"
-                for name, value in preferred_models:
-                    if value == default_model:
-                        default_model_name = name
-                        break
-                
-                model_info = gr.Markdown(
-                    f"""**Current Model**: {default_model_name}  
-                    **Estimated cost per 100 Images**: {MODEL_PRICING[default_model]}""",
+                model_info = gr.Markdown("",
                     elem_id="model-info-display"
                 )
+
+                demo.load(
+                    fn=check_authorization,
+                    inputs=None,
+                    outputs=[model_choice,login_button,model_info]
+                )
                 
                 gr.Markdown("### Uploaded Images")
                 input_gallery = gr.Gallery(