Update app.py

app.py

@@ -2,22 +2,20 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 
 model_value = {
-    "Trinity 13B (Everyone)": "WhiteRabbitNeo/Trinity-13B",
-    "Trinity 33B (Plus)": "WhiteRabbitNeo/Trinity-33B-v1.0",
-    "White Rabbit Neo 33B v1.5 (Pro)": "WhiteRabbitNeo/WhiteRabbitNeo-33B-v1.5"
+    "Lake 1": "WhiteRabbitNeo/Trinity-13B",
+    "Lake 1 Plus": "WhiteRabbitNeo/Trinity-33B-v1.0",
+    "Lake 1 Pro": "WhiteRabbitNeo/WhiteRabbitNeo-33B-v1.5"
 }
 
 client = InferenceClient(list(model_value.values()))
 
-# Define a dictionary for access codes
 access_codes = {
-    "wF99-zXDg-WRiN-qVp8": "pro",    # Example code for pro access
-    "8tj82-2UvU-8Lft-Dupb": "plus"  # Example code for plus access
+    "wF99-zXDg-WRiN-qVp8": "pro",
+    "8tj82-2UvU-8Lft-Dupb": "plus"
 }
 
-# Define model access levels
 model_access_levels = {
-    "WhiteRabbitNeo/Trinity-13B": "everyone",  # No access code needed
+    "WhiteRabbitNeo/Trinity-13B": "everyone",
     "WhiteRabbitNeo/Trinity-33B-v1.0": "plus",
     "WhiteRabbitNeo/WhiteRabbitNeo-33B-v1.5": "pro"
 }
@@ -25,6 +23,14 @@ model_access_levels = {
 def check_access_code(code):
     return access_codes.get(code, None)
 
+def recommend_model(current_model, access_level):
+    # Suggest a newer model based on the current model and access level
+    if current_model == "Lake 1" and access_level == "everyone":
+        return "Consider upgrading to Lake 1 Plus for more features."
+    elif current_model == "Lake 1 Plus" and access_level == "plus":
+        return "Consider upgrading to Lake 1 Pro for advanced features."
+    return None
+
 def respond(message, history, model, access_level):
     messages = []
 
@@ -56,16 +62,18 @@ def main(message, history, model_name, access_code):
     model = model_value[model_name]
     access_level = check_access_code(access_code)
     
-    if model == model_value["Trinity 13B (Everyone)"]:
-        # Everyone can access this model without an access code
+    if model == model_value["Lake 1"]:
         return respond(message, history, model, "everyone")
     elif access_level:
-        # Check if the access level is sufficient for the selected model
         required_access_level = model_access_levels[model]
         if access_level == required_access_level:
             return respond(message, history, model, access_level)
         else:
-            return f"You do not have access to the {model_name}. Please enter a valid access code for this model.", history
+            recommendation = recommend_model(model_name, access_level)
+            if recommendation:
+                return f"You do not have access to the {model_name}. {recommendation} Please enter a valid access code for this model.", history
+            else:
+                return f"You do not have access to the {model_name}. Please enter a valid access code for this model.", history
     else:
         return "Invalid access code. Please enter a valid code to use this service.", history
 
@@ -79,7 +87,7 @@ with gr.Blocks() as demo:
             model_dropdown = gr.Dropdown(
                 label="Choose Model",
                 choices=list(model_value.keys()),
-                value="Trinity 13B (Everyone)"
+                value="Lake 1"  # Changed default value to match model_value keys
             )
             response_output = gr.Textbox(label="Response", interactive=False, lines=10)
             message_input = gr.Textbox(label="Message", placeholder="Type your message here...")