Added image generation and translator tool

app.py

@@ -33,6 +33,181 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def image_generator(prompt: str) -> str:
+    """A tool that generates an image based on a text prompt.
+    Args:
+        prompt: A text description of the image to generate.
+    """
+    try:
+        # Use the imported image generation tool
+        image_path = image_generation_tool(prompt=prompt)
+        return image_path
+    except Exception as e:
+        return f"Error generating image: {str(e)}"
+
+@tool
+def translator(question: str, src_lang: str, tgt_lang: str) -> str:
+    """A tool that translates text from one language to another.
+    Args:
+        question: The text to translate.
+        src_lang: The source language of the text.
+        tgt_lang: The target language to translate to.
+    """
+    try:
+        # Import the deep_translator library
+        from deep_translator import GoogleTranslator
+        
+        # Map common language names to language codes if needed
+        language_map = {
+            "english": "en", "french": "fr", "spanish": "es", "german": "de",
+            "italian": "it", "portuguese": "pt", "russian": "ru", "japanese": "ja",
+            "chinese": "zh-CN", "korean": "ko", "arabic": "ar", "hindi": "hi"
+        }
+        
+        # Convert language names to codes if necessary
+        src = language_map.get(src_lang.lower(), src_lang.lower())
+        tgt = language_map.get(tgt_lang.lower(), tgt_lang.lower())
+        
+        # Perform the translation
+        translated = GoogleTranslator(source=src, target=tgt).translate(question)
+        
+        return translated
+    except Exception as e:
+        return f"Error translating text: {str(e)}"
+
+@tool
+def search(query: str) -> str:
+    """A tool that searches the web for information and returns a list of relevant pages.
+    Args:
+        query: The search query.
+    """
+    try:
+        # Use DuckDuckGo search tool that's already imported
+        search_tool = DuckDuckGoSearchTool()
+        results = search_tool(query)
+        
+        # Check if results is empty
+        if not results:
+            return f"No result found for query \"{query}\"."
+        
+        # Handle different return formats
+        formatted_results = []
+        
+        # Check if results is a list of strings or objects
+        if isinstance(results, list):
+            if results and isinstance(results[0], str):
+                # Handle list of strings
+                formatted_results = results[:5]  # Limit to top 5 results
+            else:
+                # Try to handle objects with title/url attributes
+                for result in results[:5]:
+                    try:
+                        if hasattr(result, 'title') and hasattr(result, 'url'):
+                            formatted_results.append(f"[{result.title}]({result.url})")
+                        elif isinstance(result, dict) and 'title' in result and 'url' in result:
+                            formatted_results.append(f"[{result['title']}]({result['url']})")
+                        else:
+                            formatted_results.append(str(result))
+                    except:
+                        formatted_results.append(str(result))
+        elif isinstance(results, str):
+            # Handle string result
+            return results
+            
+        return f"Found {len(formatted_results)} pages:\n" + "\n".join(formatted_results)
+    except Exception as e:
+        return f"Error searching the web: {str(e)}"
+
+@tool
+def web_search(query: str) -> str:
+    """A tool that searches the web for information and returns detailed results.
+    Args:
+        query: The search query.
+    """
+    try:
+        # Use DuckDuckGo search tool that's already imported
+        search_tool = DuckDuckGoSearchTool()
+        results = search_tool(query)
+        
+        # Check if results is empty
+        if not results:
+            return f"No results found for query \"{query}\"."
+        
+        # Handle different return formats
+        if isinstance(results, str):
+            return results
+            
+        combined_info = ""
+        
+        # Process up to 3 results
+        count = 0
+        for result in results:
+            if count >= 3:
+                break
+                
+            try:
+                if hasattr(result, 'snippet'):
+                    combined_info += f"Result {count+1}: {result.snippet}\n\n"
+                elif isinstance(result, dict) and 'snippet' in result:
+                    combined_info += f"Result {count+1}: {result['snippet']}\n\n"
+                elif hasattr(result, 'title'):
+                    combined_info += f"Result {count+1}: {result.title}\n\n"
+                elif isinstance(result, dict) and 'title' in result:
+                    combined_info += f"Result {count+1}: {result['title']}\n\n"
+                else:
+                    combined_info += f"Result {count+1}: {str(result)}\n\n"
+                count += 1
+            except:
+                combined_info += f"Result {count+1}: {str(result)}\n\n"
+                count += 1
+        
+        return combined_info.strip() if combined_info else "No detailed information found."
+    except Exception as e:
+        return f"Error searching the web: {str(e)}"
+
+@tool
+def visit_webpage(url: str) -> str:
+    """A tool that visits a webpage and returns its content.
+    Args:
+        url: The URL of the webpage to visit.
+    """
+    try:
+        # Add headers to mimic a browser request
+        headers = {
+            'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.114 Safari/537.36'
+        }
+        
+        # Make the request
+        response = requests.get(url, headers=headers, timeout=10)
+        response.raise_for_status()
+        
+        # Try to extract main content using a simple approach
+        from bs4 import BeautifulSoup
+        soup = BeautifulSoup(response.text, 'html.parser')
+        
+        # Remove script and style elements
+        for script in soup(["script", "style"]):
+            script.extract()
+        
+        # Get text
+        text = soup.get_text()
+        
+        # Break into lines and remove leading and trailing space on each
+        lines = (line.strip() for line in text.splitlines())
+        # Break multi-headlines into a line each
+        chunks = (phrase.strip() for line in lines for phrase in line.split("  "))
+        # Remove blank lines
+        text = '\n'.join(chunk for chunk in chunks if chunk)
+        
+        # Limit the length of the response
+        max_length = 5000
+        if len(text) > max_length:
+            text = text[:max_length] + "... (content truncated)"
+            
+        return text
+    except Exception as e:
+        return f"Error visiting webpage: {str(e)}"
 
 final_answer = FinalAnswerTool()
 
@@ -42,7 +217,8 @@ final_answer = FinalAnswerTool()
 model = HfApiModel(
 max_tokens=2096,
 temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+#model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
 custom_role_conversions=None,
 )
 
@@ -55,7 +231,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, image_generator, translator, get_current_time_in_timezone, search, web_search, visit_webpage], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,

requirements.txt

@@ -3,3 +3,4 @@ smolagents
 requests
 duckduckgo_search
 pandas
+deep_translator