Update app.py

app.py

@@ -50,7 +50,6 @@ def respond(
     history: list[tuple[str, str]],
     model_name,
     prompt_strategy,
-    system_message,
     override_params: bool,
     max_tokens,
     temperature,
@@ -58,11 +57,12 @@ def respond(
 ):
     assistant = create_assistant(model_name)
     
-    # Get prompt template
+    # Get selected prompt template and system context
     prompt_template: PromptTemplate = prompts[PROMPT_STRATEGIES[prompt_strategy]]
+    system_context: PromptTemplate = prompts["system_context"]
     
-    # Generate system message using prompt template
-    formatted_system_message = prompt_template.format(prompt_strategy=system_message)
+    # Format system context with the selected prompt strategy
+    formatted_system_message = system_context.format(prompt_strategy=prompt_template.template)
     
     # Prepare messages
     messages = [{"role": "system", "content": formatted_system_message}]
@@ -102,10 +102,6 @@ with gr.Blocks() as demo:
                 value=list(PROMPT_STRATEGIES.keys())[0],
                 label="Select Prompt Strategy"
             )
-            system_message = gr.Textbox(
-                value="You are a friendly and helpful AI assistant.",
-                label="System Message"
-            )
             
     with gr.Row():
         override_params = gr.Checkbox(
@@ -142,7 +138,6 @@ with gr.Blocks() as demo:
         additional_inputs=[
             model_dropdown,
             prompt_strategy_dropdown,
-            system_message,
             override_params,
             max_tokens,
             temperature,