Update app.py

app.py

@@ -68,6 +68,7 @@ with open("prompts.yaml", 'r') as stream:
 tool=[
     DuckDuckGoSearchTool,
     get_current_time_in_timezone,
+    FinalAnswerTool()
     load_tool("agents-course/text-to-image", trust_remote_code=True),
 ]
 
@@ -76,7 +77,11 @@ model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
 max_tokens=2048,
 temperature=0.5
 )
+with open("prompts.yaml", "r") as stream:
+    prompt_templates = yaml.safe_load(stream)
+
 # --- Agent setup ---
+
 agent = CodeAgent(
     tools=tool,
     model= model, 
@@ -85,7 +90,12 @@ agent = CodeAgent(
     description="Agent with web search and time lookup tools"
 )
 
+
+# Entry Point
+
+
 if __name__ == "__main__":
     # Example run
     print(agent.run("Search for the latest AI news"))
+    from Gradio_UI import GradioUI
     GradioUI(agent).launch()