Update app.py

app.py

@@ -18,6 +18,32 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+@tool                                                                                                              
+def get_huggingface_model_info(model_id: str) -> str:                                                              
+    """Fetches information about a model from the Hugging Face Hub.                                                
+    Args:                                                                                                          
+        model_id: The model identifier (e.g., 'meta-llama/Llama-2-7b' or 'bert-base-uncased').                     
+    """                                                                                                            
+    try:                                                                                                           
+        url = f"https://huggingface.co/api/models/{model_id}"                                                      
+        response = requests.get(url, timeout=10)                                                                   
+        if response.status_code == 404:                                                                            
+            return f"Model '{model_id}' not found on Hugging Face Hub."                                            
+        response.raise_for_status()                                                                                
+        data = response.json()                                                                                     
+        downloads = data.get('downloads')                                                                                  
+                                                                                      
+        info = f"**{data.get('id', model_id)}**\n"                                                                 
+        info += f"- Downloads: {downloads:,}\n" if downloads else "- Downloads: N/A\n"                                                 
+        info += f"- Likes: {data.get('likes', 'N/A')}\n"                                                           
+        info += f"- Pipeline: {data.get('pipeline_tag', 'N/A')}\n"                                                 
+        info += f"- Library: {data.get('library_name', 'N/A')}\n"                                                  
+        if data.get('tags'):                                                                                       
+            info += f"- Tags: {', '.join(data['tags'][:5])}\n"                                                     
+        return info                                                                                                
+    except Exception as e:                                                                                         
+        return f"Error fetching model info: {str(e)}"
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -55,7 +81,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, get_huggingface_model_info, get_current_time_in_timezone, image_generation_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,