Update app.py

app.py

@@ -5,6 +5,7 @@ import os
 
 API_ENDPOINT = os.getenv("API_ENDPOINT", "none")
 API_TOKEN = os.getenv("API_TOKEN")
+MODEL_ID = os.getenv("MODEL_ID", "none")  # Default value if not set
 
 def get_ai_response(message, history):
     """Fetch AI response from the API using the modern messages format."""
@@ -20,7 +21,7 @@ def get_ai_response(message, history):
     messages.append({"role": "user", "content": message})
     
     payload = {
-        "model": "chutesai/Llama-4-Maverick-17B-128E-Instruct-FP8",  
+        "model": MODEL_ID,  # Use the environment variable here
         "messages": messages,
         "stream": False,
         "max_tokens": 10000,
@@ -89,10 +90,13 @@ summary { cursor: pointer; color: #70a9e6; }
 .reasoning-content { padding: 10px; margin-top: 5px; background-color: #404040; border-radius: 5px; }
 """
 
-with gr.Blocks(css=custom_css, title="chutesai/Llama-4-Maverick-17B-128E-Instruct-FP8") as demo:
+# Get model name for display (use the full model ID from environment variable)
+model_display_name = MODEL_ID
+
+with gr.Blocks(css=custom_css, title=model_display_name) as demo:
     with gr.Column():
-        gr.Markdown("## Llama-4-Maverick-17B-128E-Instruct")
-        gr.Markdown("This is a demo of Llama-4-Maverick-17B-128E-Instruct")
+        gr.Markdown(f"## {model_display_name}")
+        gr.Markdown(f"This is a demo of {model_display_name}")
         chatbot = gr.Chatbot(elem_id="chatbot", render_markdown=False, bubble_full_width=True)
         with gr.Row():
             message = gr.Textbox(placeholder="Type your message...", show_label=False, container=False)
@@ -171,4 +175,4 @@ with gr.Blocks(css=custom_css, title="chutesai/Llama-4-Maverick-17B-128E-Instruc
     )
 
 demo.queue()
-demo.launch()
+demo.launch()