Update app.py

app.py

@@ -6,7 +6,6 @@ import requests
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# A simple custom tool (feel free to modify for extra magic!)
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
     """
@@ -21,7 +20,6 @@ def my_custom_tool(arg1: str, arg2: int) -> str:
     """
     return "What magic will you build?"
 
-# Tool to get the current local time in a given timezone.
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """
@@ -40,7 +38,6 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# Tool to probe the user's story preferences.
 @tool
 def probe_story_preferences(likes: str, dislikes: str, favorite_theme: str) -> str:
     """
@@ -56,7 +53,6 @@ def probe_story_preferences(likes: str, dislikes: str, favorite_theme: str) -> s
     """
     return f"Noted! You enjoy {likes}, prefer to avoid {dislikes}, and love {favorite_theme} themes. Let's weave these into your tale!"
 
-# New tool to generate a Studio Ghibli style image by calling the /predict endpoint.
 @tool
 def generate_studio_ghibli_image(prompt: str) -> str:
     """
@@ -74,28 +70,22 @@ def generate_studio_ghibli_image(prompt: str) -> str:
         response = requests.post(url, json=payload)
         response.raise_for_status()
         result = response.json()
-        # The API is expected to return a list in its JSON response.
-        # Adjust the indexing based on the actual output structure.
         return result[0] if result and isinstance(result, list) else "No image output received."
     except Exception as e:
         return f"Error generating image: {str(e)}"
 
-# The final answer tool is required to deliver the conclusive response.
 final_answer = FinalAnswerTool()
 
-# Set up the LLM model (using Qwen 2.5 Coder 32B as an example).
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",  # If overloaded, consider an alternate endpoint.
+    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
     custom_role_conversions=None,
 )
 
-# Load prompt templates from a YAML file.
 with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Create the CodeAgent with an interactive narrative focus.
 agent = CodeAgent(
     model=model,
     tools=[
@@ -104,7 +94,7 @@ agent = CodeAgent(
         get_current_time_in_timezone,
         probe_story_preferences,
         generate_studio_ghibli_image,
-        DuckDuckGoSearchTool(),  # Optional: enable web search if needed.
+        DuckDuckGoSearchTool(),
     ],
     max_steps=6,
     verbosity_level=1,
@@ -119,9 +109,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Launch the Gradio UI with a custom title and description for an appealing display.
-GradioUI(
-    agent,
-    title="Ghibili Interactive Tale",
-    description="Embark on a magical journey where your story comes to life with enchanting visuals and a narrative tailored just for you!"
-).launch()
+GradioUI(agent).launch()