add image generation and Hugging Face login tools

app.py

@@ -170,15 +170,47 @@ model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',#
 custom_role_conversions=None,
 )
 
+hf_token_setted = False
 
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+@tool
+def image_generation_tool(text: str) -> str:
+    """Generates an image based on the input text using the 'text-to-image' tool.
+    Args:
+        text: The input text to generate an image from.
+    Returns:
+        str: The URL of the generated image.
+    """
+    #if the user is not autenthicated to hugging face the following code will not work 
+    if hf_token_setted == False:
+        return final_answer("You need to be authenticated to Hugging Face to use this tool. Provide your 'HF_TOKEN'.")
+    
+    image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+    return image_generation_tool(text)
+
+
+from huggingface_hub import login
+@tool
+def login_to_hugging_face(HF_TOKEN: str) -> str:
+    """Logs in to the Hugging Face API using the provided API token.
+    Args:
+        HF_TOKEN: The Hugging Face API token.
+    Returns:
+        str: A message indicating the result of the login attempt.
+    """
+    global hf_token_setted
+    try:
+        login(HF_TOKEN)
+    except Exception as e:
+        return f"Failed to log in to Hugging Face: {str(e)}"
+    hf_token_setted = True
+    return "Logged in to Hugging Face successfully."
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, robot_memory_tool, image_generation_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, robot_memory_tool, image_generation_tool, login_to_hugging_face], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -189,4 +221,5 @@ agent = CodeAgent(
 )
 
 
+
 GradioUI(agent).launch()