Upload app.py with huggingface_hub

app.py

@@ -64,6 +64,19 @@ def query_databank(filename: str) -> str:
     content = load_from_databank(filename)
     return content if content else "File not found."
 
+@mcp.tool()
+def generate_image(prompt: str) -> str:
+    """Generate an image using a text-to-image model on Hugging Face."""
+    try:
+        # Using a popular model like FLUX or SDXL via InferenceClient
+        image = client.text_to_image(prompt, model="black-forest-labs/FLUX.1-schnell")
+        os.makedirs("exports/images", exist_ok=True)
+        image_path = f"exports/images/{hash(prompt)}.png"
+        image.save(image_path)
+        return f"Image generated and saved to {image_path}"
+    except Exception as e:
+        return f"Error generating image: {str(e)}"
+
 # --- AGENT LOGIC (Aussie Domain Router) ---
 
 def aussie_router(user_input, history):