Update app.py

app.py

@@ -33,6 +33,44 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+# ==============================================
+# YOUR TEXT COUNTER TOOL - ADD THIS WHOLE BLOCK
+# ==============================================
+@tool
+def count_text_stats(text: str) -> str:
+    """Count characters, words, and sentences in a text.
+    
+    Args:
+        text: The text you want to analyze
+    """
+    # Count characters (including spaces)
+    char_count = len(text)
+    
+    # Count words (split by spaces)
+    words = text.split()
+    word_count = len(words)
+    
+    # Handle empty text to avoid division by zero
+    if word_count == 0:
+        return f"""📊 Text Statistics:
+- Characters: {char_count}
+- Words: 0
+- Sentences: 0
+- Average word length: 0 characters"""
+    
+    # Count sentences (split by ., !, ?)
+    sentence_endings = ['.', '!', '?']
+    sentence_count = 0
+    for char in text:
+        if char in sentence_endings:
+            sentence_count += 1
+    
+    return f"""📊 Text Statistics:
+- Characters: {char_count}
+- Words: {word_count}
+- Sentences: {sentence_count}
+- Average word length: {char_count / word_count:.1f} characters"""
+# ==============================================
 
 final_answer = FinalAnswerTool()
 
@@ -55,7 +93,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, count_text_stats],  # <-- ADDED count_text_stats HERE!
     max_steps=6,
     verbosity_level=1,
     grammar=None,