feat: postcard generator

app.py

@@ -8,36 +8,73 @@ from tools.visit_webpage import VisitWebpageTool
 from tools.web_search import DuckDuckGoSearchTool
 from Gradio_UI import GradioUI
 
-search_tool = DuckDuckGoSearchTool()
-visit_webpage_tool = VisitWebpageTool()
 
-
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_design_specs_for_year(
-    year: str,
-) -> str:  # it's import to specify the return type
-    """A tool that
+def get_color_palette_based_on_emotion(emotion: str) -> str:
+    """A tool that creates the color palette of a given emotion.
     Args:
-        year: the year of the design specs
+        emotion: the emotion to create the color palette of.
     """
-    return search_tool(f"What is the relevant design specs for {year}?")
+    model = HfApiModel(
+        max_tokens=2096,
+        temperature=0.5,
+        model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+        custom_role_conversions=None,
+    )
+
+    messages = [
+        {
+            "role": "system",
+            "content": (
+                "ROLE: you are a cute color palette generator. "
+                "TASK: take the emotion and create a color palette which will complement the emotion. Use only pastel colors. "
+                "OUTPUT: a list of 6 colors in hex format. "
+                "CONSTRAINTS: Do not add any words or explanations. Just return the list of colors."
+            ),
+        },
+        {
+            "role": "user",
+            "content": emotion,
+        },
+    ]
+    response = model(messages, stop_sequences=["END"])
+    return response.content
 
 
 @tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
+def get_postcard_prompt_based_on_color_palette_and_greeting(
+    color_palette: str, greeting: str
+) -> str:
+    """A tool that creates the image prompt for post card based on the color palette and greeting.
     Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+        color_palette: 6 colors in hex format to create the image prompt of.
+        greeting: the greeting to create the image prompt of.
     """
-    try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+    model = HfApiModel(
+        max_tokens=2096,
+        temperature=0.5,
+        model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+        custom_role_conversions=None,
+    )
+
+    messages = [
+        {
+            "role": "system",
+            "content": (
+                "ROLE: you are a cute postcard image prompt generator. "
+                "TASK: take the color palette and greeting and create a postcard image prompt which simbolses the greeting. "
+                "Use simple vector shapes, and pastel colors from the color palette. "
+                "OUTPUT: a postcard image prompt. "
+                "CONSTRAINTS: Use only the colors from the color palette. Include the greeting into the image prompt so it would be rendered as a text. "
+            ),
+        },
+        {
+            "role": "user",
+            "content": f"Color palette: {color_palette}, Greeting: {greeting}",
+        },
+    ]
+    response = model(messages, stop_sequences=["END"])
+    return response.content
 
 
 final_answer = FinalAnswerTool()
@@ -64,10 +101,8 @@ agent = CodeAgent(
     model=model,
     tools=[
         final_answer,
-        get_current_time_in_timezone,
-        search_tool,
-        visit_webpage_tool,
-        get_design_specs_for_year,
+        get_color_palette_based_on_emotion,
+        get_postcard_prompt_based_on_color_palette_and_greeting,
         image_generation_tool,
     ],  ## add your tools here (don't remove final answer)
     max_steps=6,