Update app.py

app.py

@@ -52,15 +52,25 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 
 @tool
 def generate_image(prompt: str) -> str:
-    """Generates an image based on a text prompt using a Hugging Face model.
-    Returns the path or object of the generated image.
+    """
+    Generates an image using Stable Diffusion XL based on a text prompt.
+    Returns the path to the generated image file.
     
     Args:
         prompt: A descriptive text prompt for the image (e.g., 'A cat on the beach').
     """
     try:
-        # 调用远程工具
-        return image_generation_tool(prompt)
+        # 我们显式指定一个免费的模型 ID：stable-diffusion-xl-base-1.0
+        client = InferenceClient("stabilityai/stable-diffusion-xl-base-1.0")
+        
+        # 生成图片
+        image = client.text_to_image(prompt)
+        
+        # 将图片保存到本地文件
+        filename = "generated_image.png"
+        image.save(filename)
+        
+        return f"Image generated successfully and saved to {filename}"
     except Exception as e:
         return f"Error generating image: {str(e)}"