Update app.py

app.py

@@ -1,20 +1,11 @@
-import threading
-import time
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, tool
 import datetime
 import pytz
 import yaml
 import requests
-import base64
-from io import BytesIO
-from PIL import Image
+import time
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
-from huggingface_hub import InferenceClient
-
-# --- Set up the Inference Client for Image Generation ---
-# This client calls the Studio Ghibli Image Generator V2 model.
-image_client = InferenceClient(model_id="artificialguybr/StudioGhibli.Redmond-V2")
 
 # --- Define Tools ---
 
@@ -51,44 +42,40 @@ def get_current_time_in_timezone(timezone: str) -> str:
         if hour < 6 or hour >= 22:
             comment = "🌙 It's late, and the night whispers secrets. What dreams keep you awake?"
         else:
-            comment = "☀️ The day unfolds with gentle light. Where are you basking in this beautiful moment?"
+            comment = "☀️ The day unfolds with gentle light. Where are you enjoying this beautiful moment?"
         return f"The current local time in {timezone} is: {local_time_str}. {comment}"
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 @tool
-def probe_story_preferences(likes: str, dislikes: str, favorite_theme: str) -> str:
+def get_weather(location: str) -> str:
     """
-    Captures your story preferences and returns a poetic summary.
+    Simulates fetching poetic weather information for a given location.
     
     Args:
-        likes: What you cherish in stories.
-        dislikes: What you prefer to leave behind.
-        favorite_theme: Your favored narrative theme (e.g., adventure, mystery).
+        location: The name of your location.
     
     Returns:
-        A summary that helps weave your personalized tale.
+        A poetic description of the current weather.
     """
-    return f"Splendid! You delight in {likes} 🌸, wish to steer clear of {dislikes} 🚫, and are enchanted by {favorite_theme} themes. Let us craft your story accordingly."
+    # In a real implementation, this tool would call a weather API.
+    # For now, we'll return a simulated weather description.
+    return f"In {location}, the weather is radiant and gentle, with soft breezes and a sky that whispers tales of wonder."
 
 @tool
-def generate_studio_ghibli_image(prompt: str) -> str:
+def probe_story_preferences(likes: str, dislikes: str, favorite_theme: str) -> str:
     """
-    Generates a Studio Ghibli style image based on the provided prompt.
+    Captures your story preferences and returns a poetic summary.
     
     Args:
-        prompt: The descriptive prompt for the desired image.
+        likes: What you cherish in stories.
+        dislikes: What you prefer to leave behind.
+        favorite_theme: Your favored narrative theme (e.g., adventure, mystery).
     
     Returns:
-        A base64-encoded PNG string representing the generated image.
+        A summary that helps shape your personalized narrative.
     """
-    try:
-        # Call the image-generation API with our prompt and desired parameters.
-        result = image_client.image_generation(prompt, parameters={"width": 1024, "height": 1024, "prompt": prompt})
-        # Assume result is a base64 string or a URL; here we return it directly.
-        return result
-    except Exception as e:
-        return f"Error generating image: {str(e)}"
+    return f"Splendid! You delight in {likes} 🌸, wish to steer clear of {dislikes} 🚫, and are enchanted by {favorite_theme} themes. Let us craft your story accordingly."
 
 @tool
 def offer_interactive_options(scene: str) -> str:
@@ -96,33 +83,20 @@ def offer_interactive_options(scene: str) -> str:
     Provides interactive options based on the current scene description.
     
     Args:
-        scene: The detailed description of the current scene.
+        scene: A detailed description of the current scene.
     
     Returns:
-        A string listing interactive options.
+        A string listing interactive options for the next step.
     """
     return (
         "Options:\n"
         "1. Follow the mysterious figure in the distance.\n"
         "2. Wander deeper into the enchanted landscape.\n"
-        "3. Approach a peculiar, glowing object.\n"
-        "4. Inquire for more secrets of the scene.\n"
+        "3. Approach a peculiar object you notice.\n"
+        "4. Ask for more details about the scene.\n"
         "Please enter the number of your choice."
     )
 
-@tool
-def get_weather(location: str) -> str:
-    """
-    Simulates fetching poetic weather information for a given location.
-    
-    Args:
-        location: The name of your location.
-    
-    Returns:
-        A poetic description of the current weather.
-    """
-    return f"In {location}, the weather is radiant and gentle, with soft breezes and a sky that tells tales of wonder."
-
 final_answer = FinalAnswerTool()
 
 # --- Set up the Language Model ---
@@ -144,9 +118,8 @@ agent = CodeAgent(
         my_custom_tool,
         get_current_time_in_timezone,
         probe_story_preferences,
-        generate_studio_ghibli_image,
-        offer_interactive_options,
         get_weather,
+        offer_interactive_options,
         DuckDuckGoSearchTool(),
     ],
     max_steps=6,
@@ -156,9 +129,9 @@ agent = CodeAgent(
     name="Ghibili Interactive Tale",
     description=(
         "Ghibili Interactive Tale is your creative partner on a magical journey. "
-        "Together, we'll co-author a story as enchanting as a classic tale, with every scene painted in Studio Ghibli style. "
-        "I begin by reflecting on your current time and mood, ask for your location to capture the weather, "
-        "and then weave a narrative filled with vivid imagery and interactive choices. Let's create magic together!"
+        "I will guide you through a personalized adventure by reflecting on your current time and mood, "
+        "asking about your location to capture the weather, and using your preferences to craft a captivating narrative. "
+        "Let's begin this interactive story together!"
     ),
     prompt_templates=prompt_templates
 )
@@ -166,35 +139,14 @@ agent = CodeAgent(
 # --- Set an Initial Opener with a Storybook-Like Tone and Emojis ---
 initial_opener = (
     "🌟 Welcome, dear traveler, to the enchanting realm of Ghibili Interactive Tale! 🌟\n\n"
-    "Imagine stepping into the pages of a timeless storybook, where each page is filled with wonder and magic. "
+    "Imagine stepping into the pages of a timeless storybook, where every moment is filled with wonder and magic. "
     "I am here to be your guide and creative partner on this journey, crafting a narrative that reflects your dreams and desires.\n\n"
-    "To begin, please share your timezone (for example, 'America/New_York') so I may whisper a gentle comment on the hour. "
+    "To begin, please share your timezone (for example, 'America/New_York') so I may gently comment on the hour. "
     "Then, tell me your location so we can capture the weather and set the perfect scene for our adventure.\n\n"
     "What kind of tale are you yearning for today—a whimsical fairy tale, a mysterious quest, or a serene escape? "
     "Your story awaits, and I am excited to walk beside you every step of the way. 📖✨"
 )
 agent.conversation = [{"role": "assistant", "content": initial_opener}]
 
-# --- Proactive Story Loop ---
-# This loop proactively generates narrative segments if the user is idle.
-def proactive_story_loop(agent, interval=15, steps=5):
-    for i in range(steps):
-        # Simulate generation of the next narrative segment.
-        # Here, we assume that `agent.generate_next_message()` generates a new story segment based on the current conversation.
-        try:
-            new_segment = agent.generate_next_message()
-        except Exception as e:
-            new_segment = f"(Error generating next message: {e})"
-        if new_segment:
-            agent.conversation.append({"role": "assistant", "content": new_segment})
-        time.sleep(interval)
-    # After the narrative loop, generate a final image representing the culmination of the story.
-    image_prompt = "A breathtaking final scene that encapsulates the magical journey we have created together, in vivid Studio Ghibli style."
-    final_image = generate_studio_ghibli_image(image_prompt)
-    agent.conversation.append({"role": "assistant", "content": f"Here is the final image of our adventure: {final_image}"})
-
-# Start the proactive story loop in a background thread.
-threading.Thread(target=proactive_story_loop, args=(agent, 15, 5), daemon=True).start()
-
 # --- Launch the Interactive UI ---
 GradioUI(agent).launch()