Update app.py

app.py

@@ -79,18 +79,27 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Convert string values to nested dictionaries
+# Convert all templates to nested dicts
 for key in list(prompt_templates.keys()):
     if isinstance(prompt_templates[key], str):
         prompt_templates[key] = {"content": prompt_templates[key]}
 
-# Convert final_answer to the required nested structure
+# Ensure final_answer structure
 if 'final_answer' in prompt_templates:
     prompt_templates['final_answer'] = {
         'pre_messages': '',
-        'content': prompt_templates['final_answer'],
+        'content': prompt_templates['final_answer']['content'],
         'post_messages': ''
     }
+
+# Resolve {final_answer} in system_prompt
+if 'system_prompt' in prompt_templates and 'final_answer' in prompt_templates:
+    sys_content = prompt_templates['system_prompt']['content']
+    final_content = prompt_templates['final_answer']['content']
+    prompt_templates['system_prompt']['content'] = sys_content.replace(
+        "{final_answer}", 
+        final_content
+    )
     
 agent = CodeAgent(
     model=model,