Update app.py

app.py

@@ -1,3 +1,5 @@
+import os
+
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
@@ -50,20 +52,50 @@ custom_role_conversions=None,
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-    
+print(f"Current working directory: {os.getcwd()}")
+print(f"Files in current directory: {os.listdir('.')}")
+print(f"Does prompts.yaml exist? {os.path.exists('prompts.yaml')}")
+
+try:
+    with open("prompts.yaml", 'r') as stream:
+        prompt_templates = yaml.safe_load(stream)
+    print("--- LOADED PROMPT TEMPLATES ---")
+    print(f"Type: {type(prompt_templates)}")
+    if isinstance(prompt_templates, dict):
+        print(f"Keys: {prompt_templates.keys()}")
+    else:
+        print(f"Content: {prompt_templates}")
+    print("--- END LOADED PROMPT TEMPLATES ---")
+except Exception as e:
+    print(f"!!! ERROR LOADING prompts.yaml: {e} !!!")
+    prompt_templates = {} # Assign empty dict on error to potentially see a different error later
+
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer],
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=prompt_templates
+    prompt_templates=prompt_templates # Pass the potentially loaded dict
 )
 
+#with open("prompts.yaml", 'r') as stream:
+#    prompt_templates = yaml.safe_load(stream)
+#    
+#agent = CodeAgent(
+#    model=model,
+#    tools=[final_answer], ## add your tools here (don't remove final answer)
+#    max_steps=6,
+#    verbosity_level=1,
+#    grammar=None,
+#    planning_interval=None,
+#    name=None,
+#    description=None,
+#    prompt_templates=prompt_templates
+#)
+
 
 GradioUI(agent).launch()