Update app.py

app.py

@@ -9,6 +9,8 @@ import time
 import openai
 from tenacity import retry, stop_after_attempt, wait_exponential, retry_if_exception_type
 import random
+import re
+from collections import Counter
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -16,6 +18,95 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+
+
+class TextSummarizationTool(Tool):
+    """Summarizes a long text into a concise version by extracting leading sentences."""
+
+    name = "text_summarization"
+    description = "Summarizes a long input text into a short paragraph."
+
+    inputs = {
+        "text": {
+            "type": "string",
+            "description": "The long text that needs to be summarized.",
+        }
+    }
+    output_type = "string"
+
+    def forward(self, text: str) -> str:
+        try:
+            sentences = text.split('. ')
+            if len(sentences) <= 3:
+                return text
+            return '. '.join(sentences[:3]) + '.'
+        except Exception as e:
+            return f"Error summarizing text: {e}"
+
+class KeywordExtractorTool(Tool):
+    """Extracts top keywords from a given block of text based on frequency."""
+
+    name = "keyword_extractor"
+    description = "Extracts the most frequent keywords from the provided text."
+
+    inputs = {
+        "text": {
+            "type": "string",
+            "description": "The text to analyze for keywords.",
+        }
+    }
+    output_type = "string"
+
+    def forward(self, text: str) -> str:
+        try:
+            words = re.findall(r'\b\w+\b', text.lower())
+            stop_words = {'the', 'and', 'is', 'in', 'it', 'of', 'to', 'a'}
+            filtered_words = [w for w in words if w not in stop_words]
+            word_counts = Counter(filtered_words)
+            keywords = ', '.join(word for word, _ in word_counts.most_common(5))
+            return keywords
+        except Exception as e:
+            return f"Error extracting keywords: {e}"
+
+class TextTranslationTool(Tool):
+    """Translates simple words from source to target language using a dictionary lookup."""
+
+    name = "text_translation"
+    description = "Translates simple words from English to Spanish using a fixed dictionary."
+
+    inputs = {
+        "text": {
+            "type": "string",
+            "description": "The text to translate word-by-word.",
+        },
+        "source_lang": {
+            "type": "string",
+            "description": "Source language code (e.g., 'en').",
+        },
+        "target_lang": {
+            "type": "string",
+            "description": "Target language code (e.g., 'es').",
+        }
+    }
+    output_type = "string"
+
+    def __init__(self):
+        self.translation_dict = {
+            'hello': 'hola',
+            'world': 'mundo',
+            'goodbye': 'adiós',
+            'thank': 'gracias',
+            'you': 'tú'
+        }
+
+    def forward(self, text: str, source_lang: str, target_lang: str) -> str:
+        try:
+            words = text.split()
+            translated_words = [self.translation_dict.get(word.lower(), word) for word in words]
+            return ' '.join(translated_words)
+        except Exception as e:
+            return f"Error translating text: {e}"
+
 # --- Retry Helper for Agent Call ---
 def safe_agent_call(agent, question, retries=5, wait_time=20):
     """
@@ -43,6 +134,9 @@ class BasicAgent:
             tools=[
                 DuckDuckGoSearchTool(),
                 WikipediaSearchTool(),
+                KeywordExtractorTool(),
+                TextSummarizationTool(),
+                TextTranslationTool()
             ],
             add_base_tools=True,
         )