Update app.py

app.py

@@ -20,10 +20,17 @@ final_answer = FinalAnswer()
 with open(os.path.join(CURRENT_DIR, "prompts.yaml"), "r") as f:
     prompt_templates = yaml.safe_load(f)
 
-# 2) Flatten EVERYTHING into strings
-for key, val in prompt_templates.items():
-    if not isinstance(val, str):
-        prompt_templates[key] = yaml.safe_dump(val, default_flow_style=False)
+# 2) Recursively stringify any non-str leaves, preserving nested dicts
+def stringify_leaves(obj):
+    if isinstance(obj, dict):
+        return { k: stringify_leaves(v) for k, v in obj.items() }
+    elif isinstance(obj, list):
+        # join lists of things into one big string
+        return "".join(stringify_leaves(v) for v in obj)
+    else:
+        return str(obj)
+        
+prompt_templates = stringify_leaves(raw_templates)
 
 agent = CodeAgent(
     model=model,