Update .env.local.template

.env.local.template

@@ -8,7 +8,7 @@ MONGODB_DIRECT_CONNECTION=false
 
 COOKIE_NAME=chat-ui
 HF_ACCESS_TOKEN=#hf_<token> from from https://huggingface.co/settings/token
-
+HF_TOKEN=${HF_TOKEN}
 # used to activate search with web functionality. disabled if not defined
 SERPAPI_KEY=#your serpapi key here
 
@@ -38,6 +38,59 @@ TOOLS=`[
     ],
     "_id": "5f2c1b9d8a9a4d3b6a7f9e2c",
     "showOutput": true
+  },
+  {
+    "name": "generate_image",
+    "description": "Use this tool to generate images based on a prompt.",
+    "color": "yellow",
+    "icon": "camera",
+    "baseUrl": "BSC-LT/FLUX.1-schnell",
+    "endpoint": "/infer",
+    "outputComponent": "image",
+    "displayName":"Image Generation",
+    "outputComponentIdx": 0,
+    "inputs":[
+      {
+        name: 'prompt',
+        description: 'Prompt to generate the image',
+        paramType: 'required',
+        type: 'str'
+      },
+      { 
+        name: 'seed', 
+        paramType: 'fixed',
+        value: '0',
+        type: 'float' 
+      },
+      {
+        name: 'randomize_seed',
+        paramType: 'fixed',
+        value: 'true',
+        type: 'bool'
+      },
+      {
+        name: 'width',
+        description: 'numeric value between 256 and 2048',
+        paramType: 'optional',
+        default: '1024',
+        type: 'float'
+      },
+      {
+        name: 'height',
+        description: 'numeric value between 256 and 2048',
+        paramType: 'optional',
+        default: '1024',
+        type: 'float'
+      },
+      {
+        name: 'num_inference_steps',
+        paramType: 'fixed',
+        value: '4',
+        type: 'float'
+      }
+    ],
+    "_id": "5f2c1b9d8a9a4d3b6a7f9e2d",
+    "showOutput": true
   }
 ]`