Update app.py

app.py

@@ -4,10 +4,14 @@ import random
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Define the tell_joke tool
+# Tool: tell a random joke (no arguments)
 @tool
-def tell_joke(_: str = "") -> str:
-    """A tool that tells a random joke."""
+def tell_joke() -> str:
+    """A tool that tells a random joke.
+    
+    Returns:
+        A funny one-liner from a predefined list.
+    """
     jokes = [
         "Why don't scientists trust atoms? Because they make up everything!",
         "Why was the math book sad? Because it had too many problems.",
@@ -20,18 +24,18 @@ def tell_joke(_: str = "") -> str:
 # Final answer tool
 final_answer = FinalAnswerTool()
 
-# LLM model
+# Model
 model = HfApiModel(
     max_tokens=1024,
     temperature=0.7,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 )
 
-# Load the prompt template
-with open("prompts.yaml", "r") as stream:
+# Load system prompt
+with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Create the agent
+# Create agent
 agent = CodeAgent(
     model=model,
     tools=[final_answer, tell_joke],
@@ -40,5 +44,5 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Launch Gradio interface
+# Launch UI
 GradioUI(agent).launch()