update message

app.py

@@ -5,7 +5,7 @@ import os
 import requests
 import json
 from dotenv import load_dotenv
-import openai
+#import openai
 import base64
 import csv
 import tempfile
@@ -138,6 +138,7 @@ def create_demo():
                 gr.Markdown("# MATCHA: Museum Alt-Text for Cultural Heritage with AI 🍵 🌿")
                 gr.Markdown("Upload one or more images to generate accessible alternative text (designed to meet WCAG Guidelines)")
                 gr.Markdown("Developed by the Natural History Museum in Partnership with National Museums Liverpool. Funded by the DCMS Pilot Scheme")
+                auth_state = gr.Markdown()
             with gr.Column(scale=1):
                 with gr.Row():
                     gr.Image("images/nhm_logo.png", show_label=False, height=100,
@@ -159,7 +160,7 @@ def create_demo():
                         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),text
+                        return gr.update(choices=preferred_models, label="Select Model",value=default_model),text,""
                     
                     is_authorized = profile.username in AUTHORIZED_USER_IDS
                     if is_authorized:
@@ -173,7 +174,7 @@ def create_demo():
                         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),text
+                        return gr.update(choices=preferred_models, label="Select Model",value=default_model),text,"Free version - please email chris.addis@nhm.ac.uk for full access."
                         
                 # Define preferred and additional models directly in the function
                 preferred_models = [
@@ -223,13 +224,13 @@ def create_demo():
                 demo.load(
                     fn=check_authorization,
                     inputs=None,
-                    outputs=[model_choice,model_info]
+                    outputs=[model_choice,model_info,auth_state]
                 )
                 
                 login_button.click(
                     fn=check_authorization,
                     inputs=None, # The user profile is automatically passed on login
-                    outputs=[model_choice, model_info]
+                    outputs=[model_choice, model_info,auth_state]
                 )
 
                 gr.Markdown("### Uploaded Images")