feat: enhance chat functionality with flowchart tools integration and improved UI interactions

app/api/chat/route.ts

@@ -1,13 +1,40 @@
 import { google } from "@ai-sdk/google";
+import { Message } from "ai/react";
 import { streamText } from "ai";
-
+import { z } from "zod";
 export const maxDuration = 30;
 
+// Define tool interfaces
+interface DisplayFlowChartArgs {
+  xml: string;
+}
+
+interface ToolContext {
+  getCurrentXML: () => string;
+  displayChart: (xml: string) => void;
+}
+
 export async function POST(req: Request) {
   const { messages } = await req.json();
-  const result = streamText({
+  const response = streamText({
     model: google("gemini-2.0-flash"),
     messages,
+    tools: {
+      display_flow_chart: {
+        description: "Display a flowchart on draw.io",
+        parameters: z.object(
+          {
+            xml: z.string().describe("XML string to be displayed on draw.io"),
+          },
+        )
+      },
+      fetch_flow_chart: {
+        description: "Get the current flowchart XML from draw.io",
+        parameters: z.object({}),
+      }
+    },
+    temperature: 0,
   });
-  return result.toDataStreamResponse();
-}
+
+  return response.toDataStreamResponse();
+}