Add `huggingface_text` and `allowed_domains_override` configuration

app.py

@@ -51,7 +51,10 @@ def init_config():
     global clients
     global llm_host_names
     config = json.loads(os.environ['CONFIG'])
+    reserved_keys = ("huggingface_text", "allowed_domains_override")
     for name in config:
+        if name in reserved_keys:
+            continue
         model_personas = config[name].get("personas", {})
         client = Client(
             api_url=os.environ.get(config[name]['api_url'],
@@ -66,16 +69,21 @@ def init_config():
 
 def get_allowed_models(user_domain: str) -> List[str]:
     """
-    Get a list of allowed endpoints for a specified user domain
+    Get a list of allowed endpoints for a specified user domain. Allowed domains
+    are configured in each model's configuration and may optionally be overridden
+    in the Gradio demo configuration.
     :param user_domain: User domain (i.e. neon.ai, google.com, guest)
     :return: List of allowed endpoints from configuration
     """
+    overrides = config.get("allowed_domains_override", {})
     allowed_endpoints = []
     for client in clients:
-        if clients[client].config.inference.allowed_domains is None:
+        allowed_domains = overrides.get(client,
+                                        clients[client].config.inference.allowed_domains)
+        if allowed_domains is None:
             # Allowed domains not specified; model is public
             allowed_endpoints.append(client)
-        elif user_domain in clients[client].config.inference.allowed_domains:
+        elif user_domain in allowed_domains:
             # User domain is in the allowed domain list
             allowed_endpoints.append(client)
     return allowed_endpoints
@@ -278,12 +286,12 @@ def init_gradio() -> gr.Blocks:
             return clean_state, *clean_state
 
         # Compile
-        # TODO: Define a configuration structure for this information
-        accordion_info = config.get("accordian_info") or \
+        hf_config = config.get("huggingface_text") or dict()
+        accordion_info = hf_config.get("accordian_info") or \
             "Persona and LLM Options - Choose one:"
-        version = config.get("version") or \
+        version = hf_config.get("version") or \
             f"v{datetime.now().strftime('%Y-%m-%d')}"
-        title = config.get("title") or \
+        title = hf_config.get("title") or \
             f"Neon AI BrainForge Personas and Large Language Models ({version})"
 
         with gr.Accordion(label=accordion_info, open=True,