Fix: Update app.py with YAML validation and pin smolagents version

app.py

@@ -70,7 +70,13 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 
 with open("prompts.yaml", 'r') as stream:
     yaml_content = yaml.safe_load(stream)
-    prompt_templates = yaml_content['prompt_templates']  # Extract the prompt_templates section
+    if not isinstance(yaml_content, dict):
+        raise ValueError("YAML content must be a dictionary")
+    if 'prompt_templates' not in yaml_content:
+        raise ValueError("YAML must contain 'prompt_templates' key")
+    prompt_templates = yaml_content['prompt_templates']
+    if not isinstance(prompt_templates, dict):
+        raise ValueError("prompt_templates must be a dictionary")
     
 agent = CodeAgent(
     model=model,

requirements.txt

@@ -1,5 +1,5 @@
 markdownify
-smolagents>=0.1.0
+smolagents==0.1.0
 requests
 duckduckgo_search
 pandas