Update app.py

app.py

@@ -198,12 +198,14 @@ WEB_TOOLS = [
     TextInspectorTool(model, text_limit),
 ]
 
+flux_tool = FluxLoRATool()
+
 # Agent creation in a factory function
 def create_agent():
     """Creates a fresh agent instance for each session"""
     return CodeAgent(
         model=model,
-        tools=[visualizer] + WEB_TOOLS,
+        tools=[visualizer] + WEB_TOOLS + flux_tool
         max_steps=10,
         verbosity_level=1,
         additional_authorized_imports=AUTHORIZED_IMPORTS,