Update app.py

app.py

@@ -67,13 +67,19 @@ def generate_image(prompt: str) -> AgentImage:
     Args:
         prompt: The image description.
     """
-    res = image_generator(prompt=prompt)
-    if isinstance(res, PILImage.Image):
-        return AgentImage(res)
-    # 如果 hub 工具未来直接返回 AgentImage，就直接透传
-    if hasattr(res, "image"):
-        return res
-    raise RuntimeError(f"Unexpected image type from image_generator: {type(res)}")
+    hf_token = os.getenv("HF_TOKEN")
+    if not hf_token:
+        raise RuntimeError("HF_TOKEN is missing. Add it in Settings → Variables and Secrets.")
+
+    headers = {"Authorization": f"Bearer {hf_token}", "Accept": "image/png"}
+    url = "https://api-inference.huggingface.co/models/stabilityai/sdxl-turbo"
+
+    resp = requests.post(url, headers=headers, json={"inputs": prompt}, timeout=120)
+    if resp.status_code != 200:
+        raise RuntimeError(f"Inference API error {resp.status_code}: {resp.text[:300]}")
+
+    img = Image.open(io.BytesIO(resp.content)).convert("RGB")
+    return AgentImage(img)
 
 final_answer = FinalAnswerTool()