Update app.py

app.py

@@ -3,6 +3,7 @@ import datetime
 import requests
 import pytz
 import yaml
+import re
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
@@ -24,11 +25,14 @@ def create_new_vday_card_tool(theme:str) -> str: #it's import to specify the ret
             description="Generate image from prompt"
         )
         '''
-        image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-        image = image_generation_tool(theme)
-        return final_answer(image)
+        if re.search("Valentine's Day card", prompt) != None:
+            image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+            image = image_generation_tool(theme)
+            return final_answer(image)
+        else:
+            return f"You did not prompt the agent for a Valentine's Day card, please try again"
     except Exception as e:
-        return f"Error creating image for theme '{theme}': {str(e)}"
+        return f"Error creating card for theme '{theme}': {str(e)}"
 
 '''
 @tool