Update app.py

app.py

@@ -16,7 +16,10 @@ def create_new_vday_card_tool(theme:str) -> str: #it's import to specify the ret
         theme: A string representing the theme for the Valentine's Day card.
     """
     try:
-        image = image_generation_tool(theme)
+        # Import tool from Hub
+        image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+        prompt = "A high-resolution image of a valentine's day card themed around " + theme
+        image = image_generation_tool(prompt)
         return final_answer(image)
     except Exception as e:
         return f"Error creating image for theme '{theme}': {str(e)}"
@@ -50,17 +53,12 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may
 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)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, create_new_vday_card_tool, image_generation_tool],  # add your tools here (don't remove final answer)
+    tools=[final_answer, create_new_vday_card_tool],  # add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,