Parse initial Message for question

app.py

@@ -48,11 +48,16 @@ def load_config(request: gr.Request):
     return survey_question, survey_template, initial_message, model_args, userid
 
 # Post-loading
-def update_system_message(question: str, template: PromptTemplate | str) -> str:
+def update_template(question: str, template: PromptTemplate | str) -> str:
     """
-    On questionBox|templateBox update, read questionBox|templateBox and update SystemMessageBox
+    Updates templates. Currently only accepts a "question" variable, but can add future templating in the future.
     """
-    return template.format(question=question)
+    if isinstance(template, str):
+        template = PromptTemplate(template)
+    if "question" in template.variables:
+        return template.format(question=question)
+    else:
+        return str(template)
 
 
 def reset_interview() -> tuple[list[list[str | None]], gr.Button, gr.Button]:
@@ -218,9 +223,13 @@ with gr.Blocks() as demo:
             userid,
         ]
     ).then(
-        update_system_message,
+        update_template,
         inputs=[surveyQuestion, surveyTemplate],
         outputs=[systemMessage],
+    ).then(
+        update_template,
+        inputs=[surveyQuestion, initialMessage],
+        outputs=initialMessage,
     ).then(
         initialize_interview,
         inputs=[systemMessage, initialMessage, modelArgs],