Update app.py

app.py

@@ -27,33 +27,7 @@ def save_image_to_disk(image: Image.Image, filename: str) -> str:
     except Exception as e:
         return f"Error saving image: {str(e)}"
 
-@tool
-def reliable_image_generator(prompt: str) -> Image.Image:
-    """A highly reliable tool to generate images.
-    Args:
-        prompt: The description of the image to generate.
-    """
-    try:
-        # 将提示词编码为 URL 格式
-        encoded_prompt = urllib.parse.quote(prompt)
-        url = f"https://image.pollinations.ai/prompt/{encoded_prompt}"
-        
-        # 关键修复：伪装成正常的 Windows Chrome 浏览器
-        headers = {
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
-        }
-        
-        response = requests.get(url, headers=headers)
-        
-        # 提前拦截错误：如果不是 200 成功，直接把服务器的骂人话打印出来
-        if response.status_code != 200:
-            raise Exception(f"API Error {response.status_code}: 返回了非图片内容")
-            
-        # 将网络请求返回的图片数据转化为真正的 PIL 图片对象
-        image = Image.open(BytesIO(response.content))
-        return image
-    except Exception as e:
-        raise Exception(f"Image generation failed: {str(e)}")
+
         
 @tool
 def get_hacker_news_top_stories(limit: int = 5) -> str:
@@ -136,8 +110,8 @@ custom_role_conversions=None,
 
 
 # Import tool from Hub
-# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-image_generation_tool = reliable_image_generator
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)