Update app.py

app.py

@@ -108,47 +108,7 @@ with gr.Blocks() as demo:
                 value=list(PROMPT_STRATEGIES.keys())[0],
                 label="Select Prompt Strategy"
             )
-            
-    with gr.Row():
-        override_params = gr.Checkbox(
-            label="Override Template Parameters",
-            value=False
-        )
-    
-    with gr.Row():
-        with gr.Column(visible=False) as param_controls:
-            max_tokens = gr.Slider(
-                minimum=1,
-                maximum=2048,
-                value=512,
-                step=1,
-                label="Max new tokens"
-            )
-            temperature = gr.Slider(
-                minimum=0.1,
-                maximum=4.0,
-                value=0.7,
-                step=0.1,
-                label="Temperature"
-            )
-            top_p = gr.Slider(
-                minimum=0.1,
-                maximum=1.0,
-                value=0.95,
-                step=0.05,
-                label="Top-p (nucleus sampling)"
-            )
     
-    with gr.Accordion("Current Prompt Details", open=False):
-        system_prompt_display = gr.TextArea(
-            label="System Prompt",
-            interactive=False,
-            lines=3
-        )
-        current_messages_display = gr.JSON(
-            label="Full Conversation Context",
-        )
-
     chatbot = gr.ChatInterface(
         fn=respond,
         additional_inputs=[
@@ -160,6 +120,50 @@ with gr.Blocks() as demo:
             top_p,
         ]
     )
+
+    # Parameters and Prompt Details section below the chat
+    with gr.Row():
+        with gr.Column():
+            # Parameter Controls
+            override_params = gr.Checkbox(
+                label="Override Template Parameters",
+                value=False
+            )
+            
+            with gr.Column(visible=False) as param_controls:
+                max_tokens = gr.Slider(
+                    minimum=1,
+                    maximum=2048,
+                    value=512,
+                    step=1,
+                    label="Max new tokens"
+                )
+                temperature = gr.Slider(
+                    minimum=0.1,
+                    maximum=4.0,
+                    value=0.7,
+                    step=0.1,
+                    label="Temperature"
+                )
+                top_p = gr.Slider(
+                    minimum=0.1,
+                    maximum=1.0,
+                    value=0.95,
+                    step=0.05,
+                    label="Top-p (nucleus sampling)"
+                )
+
+        with gr.Column():
+            # Prompt Details
+            with gr.Accordion("Current Prompt Details", open=False):
+                system_prompt_display = gr.TextArea(
+                    label="System Prompt",
+                    interactive=False,
+                    lines=3
+                )
+                current_messages_display = gr.JSON(
+                    label="Full Conversation Context",
+                )
     
     def toggle_param_controls(override):
         return gr.Column(visible=override)
@@ -169,13 +173,10 @@ with gr.Blocks() as demo:
         system_context = prompts["system_context"]
         formatted_system_message = system_context.format(prompt_strategy=prompt_template.template)
         
-        # Get the template parameters
-        params = prompt_template.parameters
-        
         return (
             formatted_system_message,
             {
-                "Template Parameters": params,
+                "Template Parameters": prompt_template.parameters,
                 "Prompt Strategy": prompt_template.template
             }
         )