Update app.py

app.py

@@ -33,6 +33,39 @@ 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 check_hf_model(model_name: str) -> str:
+    """A tool that checks if a Hugging Face model is available and gets basic information.
+    
+    Args:
+        model_name: The name of the model on Hugging Face (e.g., 'google/bert-base-uncased')
+    """
+    try:
+        # Construct the HF API URL
+        api_url = f"https://huggingface.co/api/models/{model_name}"
+        
+        # Make the request
+        response = requests.get(api_url)
+        
+        if response.status_code == 200:
+            data = response.json()
+            
+            # Extract useful information
+            pipeline_tag = data.get('pipeline_tag', 'Not specified')
+            downloads = data.get('downloads', 'Not available')
+            likes = data.get('likes', 'Not available')
+            
+            return f"""Model '{model_name}' is available:
+- Task: {pipeline_tag}
+- Downloads: {downloads}
+- Likes: {likes}
+- URL: https://huggingface.co/{model_name}"""
+        else:
+            return f"Model '{model_name}' not found or not accessible."
+            
+    except Exception as e:
+        return f"Error checking model '{model_name}': {str(e)}"
+
 
 final_answer = FinalAnswerTool()
 
@@ -55,7 +88,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,check_hf_model], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,