Update app.py

app.py

@@ -96,12 +96,25 @@ with gr.Blocks(title="Athena Playground Chat", css=css, theme=theme) as demo:
     gr.Markdown("# 🚀 Athena Playground Chat")
     gr.Markdown("*Powered by HuggingFace ZeroGPU*")
 
-    # Placeholders for config components
-    model_choice = gr.State("Athena-R3X 8B")
-    max_length = gr.State(512)
-    temperature = gr.State(0.7)
+    # 1. Declare config components FIRST
+    model_choice = gr.Dropdown(
+        label="📱 Model",
+        choices=list(MODELS.keys()),
+        value="Athena-R3X 8B",
+        info="Select which Athena model to use"
+    )
+    max_length = gr.Slider(
+        32, 2048, value=512, 
+        label="📝 Max Tokens",
+        info="Maximum number of tokens to generate"
+    )
+    temperature = gr.Slider(
+        0.1, 2.0, value=0.7, 
+        label="🎨 Creativity",
+        info="Higher values = more creative responses"
+    )
 
-    # Main chat interface
+    # 2. Create the chat interface, passing the controls as additional_inputs
     chat_interface = gr.ChatInterface(
         fn=respond,
         additional_inputs=[model_choice, max_length, temperature],
@@ -124,26 +137,10 @@ with gr.Blocks(title="Athena Playground Chat", css=css, theme=theme) as demo:
         type="messages"
     )
 
-    # Accordion at the bottom for configurations
+    # 3. Place the controls in an Accordion for display (they are still linked!)
     with gr.Accordion("Configurations", open=False):
-        model_choice = gr.Dropdown(
-            label="📱 Model",
-            choices=list(MODELS.keys()),
-            value="Athena-R3X 8B",
-            info="Select which Athena model to use"
-        )
-        max_length = gr.Slider(
-            32, 2048, value=512, 
-            label="📝 Max Tokens",
-            info="Maximum number of tokens to generate"
-        )
-        temperature = gr.Slider(
-            0.1, 2.0, value=0.7, 
-            label="🎨 Creativity",
-            info="Higher values = more creative responses"
-        )
-        # Link the config components to the chat interface's additional_inputs
-        chat_interface.additional_inputs = [model_choice, max_length, temperature]
+        gr.Markdown("### Change Model and Generation Settings")
+        gr.Row([model_choice, max_length, temperature])
 
 if __name__ == "__main__":
     demo.launch()