Update app.py

app.py

@@ -33,6 +33,29 @@ 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 text_to_image(prompt: str) -> AgentImage:
+    """Generate an image from text via Hugging Face Inference API (sdxl-turbo).
+    Args:
+        prompt: The image description in English.
+    """
+    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"
+    }
+    # 你也可以换成其它支持推理的模型，比如：black-forest-labs/FLUX.1-dev
+    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[:200]}")
+
+    img = Image.open(io.BytesIO(resp.content)).convert("RGB")
+    return AgentImage(img)
 
 final_answer = FinalAnswerTool()
 
@@ -55,7 +78,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, DuckDuckGoSearchTool(), image_generation_tool, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer, DuckDuckGoSearchTool(), image_generation_tool, get_current_time_in_timezone, text_to_image ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,