Update app.py

app.py

@@ -4,34 +4,31 @@ import random
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Tool: tells a random joke
+# Define the tell_joke tool
 @tool
 def tell_joke(_: str = "") -> str:
-    """A tool that tells a random joke.
-    Args:
-        _: Dummy argument (not used)
-    """
+    """A tool that tells a random joke."""
     jokes = [
         "Why don't scientists trust atoms? Because they make up everything!",
-        "I told my computer I needed a break, and it said 'No problem — I'll go to sleep.'",
+        "Why was the math book sad? Because it had too many problems.",
         "Why did the scarecrow win an award? Because he was outstanding in his field!",
-        "What’s a programmer’s favorite hangout place? The Foo Bar.",
-        "Parallel lines have so much in common… it’s a shame they’ll never meet."
+        "Why did the tomato turn red? Because it saw the salad dressing!",
+        "Why did the computer get cold? Because it forgot to close its windows."
     ]
     return random.choice(jokes)
 
 # Final answer tool
 final_answer = FinalAnswerTool()
 
-# Model
+# LLM model
 model = HfApiModel(
     max_tokens=1024,
     temperature=0.7,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 )
 
-# Load prompts
-with open("prompts.yaml", 'r') as stream:
+# Load the prompt template
+with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)
 
 # Create the agent
@@ -43,5 +40,5 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Launch UI
+# Launch Gradio interface
 GradioUI(agent).launch()