Update app.py

app.py

@@ -6,6 +6,7 @@ import requests
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
+# 1. A placeholder tool for future creativity.
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
     """
@@ -20,13 +21,14 @@ def my_custom_tool(arg1: str, arg2: int) -> str:
     """
     return "What magic will you build?"
 
+# 2. A tool to get the current local time for a given timezone.
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """
     Fetches the current local time in a specified timezone.
     
     Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+        timezone: A valid timezone string (e.g., 'America/New_York').
     
     Returns:
         A string with the current local time.
@@ -38,28 +40,30 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+# 3. A tool to capture your story preferences.
 @tool
 def probe_story_preferences(likes: str, dislikes: str, favorite_theme: str) -> str:
     """
-    Captures the user's story preferences and returns a personalized summary.
+    Captures your story preferences and returns a personalized summary.
     
     Args:
-        likes: Elements the user loves in stories.
-        dislikes: Elements the user dislikes.
-        favorite_theme: The story theme the user prefers (e.g., adventure, mystery).
+        likes: Elements you love in stories.
+        dislikes: Elements you dislike.
+        favorite_theme: Your preferred story theme (e.g., adventure, mystery).
     
     Returns:
-        A summary string to help shape a personalized narrative.
+        A summary string to shape your personalized narrative.
     """
     return f"Noted! You enjoy {likes}, prefer to avoid {dislikes}, and love {favorite_theme} themes. Let's weave these into your tale!"
 
+# 4. A tool that generates a Studio Ghibli–style image.
 @tool
 def generate_studio_ghibli_image(prompt: str) -> str:
     """
     Generates a Studio Ghibli style image based on the provided text prompt.
     
     Args:
-        prompt: The text prompt for image generation.
+        prompt: The text prompt describing the image.
     
     Returns:
         A URL or identifier for the generated image.
@@ -74,18 +78,43 @@ def generate_studio_ghibli_image(prompt: str) -> str:
     except Exception as e:
         return f"Error generating image: {str(e)}"
 
+# 5. A tool to offer interactive options based on the current scene.
+@tool
+def offer_interactive_options(scene: str) -> str:
+    """
+    Provides interactive options for the next step in your adventure based on the scene description.
+    
+    Args:
+        scene: The description of the current scene.
+    
+    Returns:
+        A string listing numbered options for you to choose from.
+    """
+    return (
+        "Options:\n"
+        "1. Follow the mysterious character that appears in the scene.\n"
+        "2. Explore the magical surroundings further.\n"
+        "3. Interact with a unique object you notice.\n"
+        "4. Ask for more details about the scene.\n"
+        "Please enter the number of your choice."
+    )
+
+# 6. The final answer tool that consolidates the agent’s response.
 final_answer = FinalAnswerTool()
 
+# 7. Set up the language model.
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",  # Use an alternative endpoint if necessary.
     custom_role_conversions=None,
 )
 
+# 8. Load prompt templates from a YAML file.
 with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)
 
+# 9. Create the interactive agent.
 agent = CodeAgent(
     model=model,
     tools=[
@@ -94,7 +123,8 @@ agent = CodeAgent(
         get_current_time_in_timezone,
         probe_story_preferences,
         generate_studio_ghibli_image,
-        DuckDuckGoSearchTool(),
+        offer_interactive_options,
+        DuckDuckGoSearchTool(),  # Optional: for additional web-based info.
     ],
     max_steps=6,
     verbosity_level=1,
@@ -102,11 +132,12 @@ agent = CodeAgent(
     planning_interval=None,
     name="Ghibili Interactive Tale",
     description=(
-        "A highly interactive and fun agent that co-creates a personalized Studio Ghibli-inspired adventure. "
-        "It guides you through a magical narrative, asks for your story preferences, and generates captivating images in Ghibili style. "
-        "Enjoy a seamless blend of interactive storytelling and visual artistry!"
+        "A highly interactive and dynamic agent that co-creates a personalized Studio Ghibli-inspired adventure. "
+        "It engages you by asking probing questions about your preferences, crafts vivid narrative scenes complete with enchanting imagery, "
+        "and offers interactive choices at every step to shape the story. Enjoy a seamless blend of storytelling, art, and creativity!"
     ),
     prompt_templates=prompt_templates
 )
 
+# 10. Launch the Gradio web interface to interact with the agent.
 GradioUI(agent).launch()