Spaces:

sahilmayekar
/

CodeAgent

Sleeping

App Files Files Community

sahilmayekar commited on Sep 25, 2025

Commit

e7addf3

1 Parent(s): 16eff4c

Final

Browse files

Files changed (39) hide show

.gitignore +2 -1
Dockerfile +1 -1
crew_result.txt +0 -0
entrypoint.sh +1 -1
frontend/app/.gitignore +24 -0
frontend/app/README.md +16 -0
frontend/app/components.json +22 -0
frontend/app/eslint.config.js +29 -0
frontend/app/index.html +13 -0
frontend/app/jsconfig.json +8 -0
frontend/app/package-lock.json +0 -0
frontend/app/package.json +40 -0
frontend/app/public/vite.svg +1 -0
frontend/app/src/App.css +42 -0
frontend/app/src/App.jsx +634 -0
frontend/app/src/assets/react.svg +1 -0
frontend/app/src/components/ui/button.jsx +54 -0
frontend/app/src/components/ui/card.jsx +101 -0
frontend/app/src/index.css +186 -0
frontend/app/src/lib/utils.js +6 -0
frontend/app/src/main.jsx +10 -0
frontend/app/vite.config.js +14 -0
poem.txt +1 -0
requirements.txt +0 -0
sql_files/app.db +0 -0
sql_files/schema.sql +17 -0
src/__pycache__/app.cpython-313.pyc +0 -0
src/app.py +202 -8
src/codeagent/.gitignore +4 -0
src/codeagent/README.md +56 -0
src/codeagent/pyproject.toml +21 -0
src/codeagent/src/codeagent/__init__.py +0 -0
src/codeagent/src/codeagent/crews/SqlCrew/__init__.py +1 -0
src/codeagent/src/codeagent/crews/SqlCrew/config/agents.yaml +56 -0
src/codeagent/src/codeagent/crews/SqlCrew/config/tasks.yaml +33 -0
src/codeagent/src/codeagent/crews/SqlCrew/sqlcrew.py +160 -0
src/codeagent/src/codeagent/main.py +86 -0
src/codeagent/src/codeagent/tools/__init__.py +0 -0
src/codeagent/src/codeagent/tools/custom_tools.py +64 -0

.gitignore CHANGED Viewed

	@@ -1 +1,2 @@
1	- /python


1	+ /python
2	+ /frontend/node_modules

Dockerfile CHANGED Viewed

@@ -41,6 +41,6 @@ ENV PORT=7860
 EXPOSE 11434 7860 5000
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 ENTRYPOINT ["/usr/local/bin/entrypoint.sh"]

 EXPOSE 11434 7860 5000
+HEALTHCHECK CMD curl --fail http://localhost:7860/_stcore/health || exit 1
 ENTRYPOINT ["/usr/local/bin/entrypoint.sh"]

crew_result.txt ADDED Viewed

File without changes

entrypoint.sh CHANGED Viewed

@@ -9,6 +9,6 @@ done
 ollama pull nomic-embed-text:latest
-ollama pull gpt-oss:20b || true
 exec flask --app src/app run --host=0.0.0.0 --port=$PORT

 ollama pull nomic-embed-text:latest
+ollama pull llama3.1:8b-instruct-q4_0 || true
 exec flask --app src/app run --host=0.0.0.0 --port=$PORT

frontend/app/.gitignore ADDED Viewed

	@@ -0,0 +1,24 @@

+# Logs
+logs
+*.log
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+pnpm-debug.log*
+lerna-debug.log*
+node_modules
+dist
+dist-ssr
+*.local
+# Editor directories and files
+.vscode/*
+!.vscode/extensions.json
+.idea
+.DS_Store
+*.suo
+*.ntvs*
+*.njsproj
+*.sln
+*.sw?

frontend/app/README.md ADDED Viewed

	@@ -0,0 +1,16 @@

+# React + Vite
+This template provides a minimal setup to get React working in Vite with HMR and some ESLint rules.
+Currently, two official plugins are available:
+- [@vitejs/plugin-react](https://github.com/vitejs/vite-plugin-react/blob/main/packages/plugin-react) uses [Babel](https://babeljs.io/) for Fast Refresh
+- [@vitejs/plugin-react-swc](https://github.com/vitejs/vite-plugin-react/blob/main/packages/plugin-react-swc) uses [SWC](https://swc.rs/) for Fast Refresh
+## React Compiler
+The React Compiler is not enabled on this template. To add it, see [this documentation](https://react.dev/learn/react-compiler/installation).
+## Expanding the ESLint configuration
+If you are developing a production application, we recommend using TypeScript with type-aware lint rules enabled. Check out the [TS template](https://github.com/vitejs/vite/tree/main/packages/create-vite/template-react-ts) for information on how to integrate TypeScript and [`typescript-eslint`](https://typescript-eslint.io) in your project.

frontend/app/components.json ADDED Viewed

	@@ -0,0 +1,22 @@

+{
+  "$schema": "https://ui.shadcn.com/schema.json",
+  "style": "new-york",
+  "rsc": false,
+  "tsx": false,
+  "tailwind": {
+    "config": "",
+    "css": "src/index.css",
+    "baseColor": "neutral",
+    "cssVariables": true,
+    "prefix": ""
+  },
+  "iconLibrary": "lucide",
+  "aliases": {
+    "components": "@/components",
+    "utils": "@/lib/utils",
+    "ui": "@/components/ui",
+    "lib": "@/lib",
+    "hooks": "@/hooks"
+  },
+  "registries": {}
+}

frontend/app/eslint.config.js ADDED Viewed

	@@ -0,0 +1,29 @@

+import js from '@eslint/js'
+import globals from 'globals'
+import reactHooks from 'eslint-plugin-react-hooks'
+import reactRefresh from 'eslint-plugin-react-refresh'
+import { defineConfig, globalIgnores } from 'eslint/config'
+export default defineConfig([
+  globalIgnores(['dist']),
+  {
+    files: ['**/*.{js,jsx}'],
+    extends: [
+      js.configs.recommended,
+      reactHooks.configs['recommended-latest'],
+      reactRefresh.configs.vite,
+    ],
+    languageOptions: {
+      ecmaVersion: 2020,
+      globals: globals.browser,
+      parserOptions: {
+        ecmaVersion: 'latest',
+        ecmaFeatures: { jsx: true },
+        sourceType: 'module',
+      },
+    },
+    rules: {
+      'no-unused-vars': ['error', { varsIgnorePattern: '^[A-Z_]' }],
+    },
+  },
+])

frontend/app/index.html ADDED Viewed

	@@ -0,0 +1,13 @@

+<!doctype html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <link rel="icon" type="image/svg+xml" href="/vite.svg" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>app</title>
+  </head>
+  <body>
+    <div id="root"></div>
+    <script type="module" src="/src/main.jsx"></script>
+  </body>
+</html>

frontend/app/jsconfig.json ADDED Viewed

	@@ -0,0 +1,8 @@

+{
+  "compilerOptions": {
+    "baseUrl": ".",
+    "paths": {
+      "@/*": ["./src/*"]
+    }
+  }
+}

frontend/app/package-lock.json ADDED Viewed

The diff for this file is too large to render. See raw diff

frontend/app/package.json ADDED Viewed

	@@ -0,0 +1,40 @@

+{
+  "name": "app",
+  "private": true,
+  "version": "0.0.0",
+  "type": "module",
+  "scripts": {
+    "dev": "vite",
+    "build": "vite build",
+    "lint": "eslint .",
+    "preview": "vite preview"
+  },
+  "dependencies": {
+    "@monaco-editor/react": "^4.7.0",
+    "@radix-ui/react-slot": "^1.2.3",
+    "@tailwindcss/vite": "^4.1.13",
+    "class-variance-authority": "^0.7.1",
+    "clsx": "^2.1.1",
+    "lucide-react": "^0.544.0",
+    "monaco-editor": "^0.53.0",
+    "react": "^19.1.1",
+    "react-dom": "^19.1.1",
+    "socket.io-client": "^4.8.1",
+    "tailwind-merge": "^3.3.1",
+    "tailwind-variants": "^3.1.1",
+    "tailwindcss": "^4.1.13",
+    "zustand": "^5.0.8"
+  },
+  "devDependencies": {
+    "@eslint/js": "^9.36.0",
+    "@types/react": "^19.1.13",
+    "@types/react-dom": "^19.1.9",
+    "@vitejs/plugin-react": "^5.0.3",
+    "eslint": "^9.36.0",
+    "eslint-plugin-react-hooks": "^5.2.0",
+    "eslint-plugin-react-refresh": "^0.4.20",
+    "globals": "^16.4.0",
+    "tw-animate-css": "^1.3.8",
+    "vite": "^7.1.7"
+  }
+}

frontend/app/public/vite.svg ADDED Viewed

frontend/app/src/App.css ADDED Viewed

	@@ -0,0 +1,42 @@

+#root {
+  max-width: 1280px;
+  margin: 0 auto;
+  padding: 2rem;
+  text-align: center;
+}
+.logo {
+  height: 6em;
+  padding: 1.5em;
+  will-change: filter;
+  transition: filter 300ms;
+}
+.logo:hover {
+  filter: drop-shadow(0 0 2em #646cffaa);
+}
+.logo.react:hover {
+  filter: drop-shadow(0 0 2em #61dafbaa);
+}
+@keyframes logo-spin {
+  from {
+    transform: rotate(0deg);
+  }
+  to {
+    transform: rotate(360deg);
+  }
+}
+@media (prefers-reduced-motion: no-preference) {
+  a:nth-of-type(2) .logo {
+    animation: logo-spin infinite 20s linear;
+  }
+}
+.card {
+  padding: 2em;
+}
+.read-the-docs {
+  color: #888;
+}

frontend/app/src/App.jsx ADDED Viewed

	@@ -0,0 +1,634 @@

+import React, { useEffect, useState, useRef } from "react";
+import { Editor, DiffEditor } from "@monaco-editor/react";
+import { create } from "zustand";
+import { io } from "socket.io-client";
+const BASE_URL = "http://localhost:7860";
+const useStore = create((set, get) => ({
+  files: [],
+  selectedFile: null,
+  editorContent: "",
+  dbSchema: { tables: [] },
+  chat: [],
+  consoleOutput: "",
+  loading: false,
+  fileProposals: {},
+  showDiffForFile: null,
+  fileContents: {},
+  setFiles: (files) => set({ files }),
+  setSelectedFile: (name) => set({ selectedFile: { name } }),
+  setEditorContent: (editorContent) => set({ editorContent }),
+  setDbSchema: (dbSchema) => set({ dbSchema }),
+  pushChat: (msg) => set((s) => ({ chat: [...s.chat, msg] })),
+  setConsoleOutput: (consoleOutput) => set({ consoleOutput }),
+  setLoading: (loading) => set({ loading }),
+  setFileProposal: (name, details) => set((s) => ({ fileProposals: { ...s.fileProposals, [name]: { ...details, active: true } } })),
+  setShowDiffForFile: (fileName) => set({ showDiffForFile: fileName }),
+  openFile: async (name) => {
+    const currentState = get();
+    if (currentState.selectedFile?.name && currentState.editorContent) {
+      set({ fileContents: { ...currentState.fileContents, [currentState.selectedFile.name]: currentState.editorContent } });
+    }
+    set({ loading: true });
+    let content = "";
+    const state = get();
+    const proposal = state.fileProposals[name];
+    if (proposal?.isNew) {
+      content = "";
+    } else {
+      content = state.fileContents[name];
+      if (!content) {
+        try {
+          const res = await fetch(`${BASE_URL}/files/${encodeURIComponent(name)}`);
+          if (!res.ok) throw new Error("Failed to load file");
+          const data = await res.json();
+          content = data.content;
+          set({ fileContents: { ...get().fileContents, [name]: content } });
+        } catch (e) {
+          console.warn(e);
+          content = `-- demo content for ${name}\nSELECT 1;`;
+          set({ fileContents: { ...get().fileContents, [name]: content } });
+        }
+      }
+    }
+    set({ selectedFile: { name }, editorContent: content, loading: false });
+    const finalState = get();
+    const prop = finalState.fileProposals[name];
+    if (prop?.active) {
+      set({ showDiffForFile: name });
+    } else {
+      set({ showDiffForFile: null });
+    }
+  },
+  acceptProposal: () => {
+    const state = get();
+    const file = state.selectedFile?.name;
+    if (state.showDiffForFile !== file) return;
+    const prop = state.fileProposals[file];
+    if (!prop) return;
+    if (prop.isNew) {
+      const confirmedName = prompt("Save new file as (.sql required):", prop.suggestedName);
+      if (!confirmedName?.trim() || !confirmedName.endsWith(".sql")) {
+        return;
+      }
+      const finalName = confirmedName.trim();
+      fetch(`${BASE_URL}/files`, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ filename: finalName, content: prop.code }),
+      })
+        .then((res) => {
+          if (!res.ok) throw new Error(`HTTP ${res.status}: ${res.statusText}`);
+          return res.json();
+        })
+        .then(() => {
+          const newProps = { ...state.fileProposals };
+          delete newProps[file];
+          const currentFiles = get().files;
+          set({
+            files: [...currentFiles, { name: finalName }],
+            editorContent: prop.code,
+            fileContents: { ...get().fileContents, [finalName]: prop.code },
+            showDiffForFile: null,
+            fileProposals: newProps,
+          });
+          if (finalName !== file) {
+            set({ selectedFile: { name: finalName } });
+          }
+        })
+        .catch((e) => console.error("Failed to create file:", e));
+    } else {
+      fetch(`${BASE_URL}/files/${encodeURIComponent(file)}`, {
+        method: "PUT",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ content: prop.code }),
+      })
+        .then((res) => {
+          if (!res.ok) throw new Error(`HTTP ${res.status}: ${res.statusText}`);
+          const newProps = { ...state.fileProposals };
+          delete newProps[file];
+          set({
+            editorContent: prop.code,
+            fileContents: { ...state.fileContents, [file]: prop.code },
+            showDiffForFile: null,
+            fileProposals: newProps,
+          });
+        })
+        .catch((e) => console.error("Failed to save after accept:", e));
+    }
+  },
+  rejectProposal: () => {
+    const state = get();
+    const file = state.selectedFile?.name;
+    if (state.showDiffForFile !== file) return;
+    const newProps = { ...state.fileProposals };
+    delete newProps[file];
+    set({ showDiffForFile: null, fileProposals: newProps });
+  },
+  refreshFiles: async () => {
+    try {
+      const res = await fetch(`${BASE_URL}/files`);
+      if (!res.ok) throw new Error("Failed to load files");
+      let data = await res.json();
+      data = data.filter((f) => f.name.endsWith(".sql"));
+      set({ files: data || [] });
+    } catch (e) {
+      console.warn(e);
+    }
+  },
+}));
+const Card = ({ children, className = "" }) => (
+  <div className={`rounded-xl border bg-white shadow-sm ${className}`}>{children}</div>
+);
+const CardContent = ({ children, className = "" }) => (
+  <div className={`p-3 ${className}`}>{children}</div>
+);
+const IconButton = ({ children, onClick, className = "" }) => (
+  <button
+    onClick={onClick}
+    className={`px-3 py-1 rounded-md border hover:shadow-sm text-sm bg-blue-500 text-white ${className}`}
+  >
+    {children}
+  </button>
+);
+function FileExplorer() {
+  const { files, selectedFile, loading, fileProposals, fileContents, editorContent } = useStore();
+  const openFile = useStore((s) => s.openFile);
+  const loadFiles = async () => {
+    useStore.setState({ loading: true });
+    try {
+      const res = await fetch(`${BASE_URL}/files`);
+      if (!res.ok) throw new Error("Failed to load files");
+      let data = await res.json();
+      data = data.filter((f) => f.name.endsWith(".sql"));
+      useStore.setState({ files: data || [] });
+    } catch (e) {
+      console.warn(e);
+    } finally {
+      useStore.setState({ loading: false });
+    }
+  };
+  useEffect(() => {
+    loadFiles();
+  }, []);
+  useEffect(() => {
+    if (files.length > 0 && !selectedFile?.name) {
+      openFile(files[0].name);
+    }
+  }, [files, selectedFile?.name, openFile]);
+  async function createNewFile() {
+    const name = prompt("New file name (must end with .sql):");
+    if (!name || !name.endsWith(".sql")) return;
+    try {
+      await fetch(`${BASE_URL}/files`, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ filename: name, content: "" }),
+      });
+      loadFiles();
+    } catch (e) {
+      console.warn(e);
+    }
+  }
+  const proposedNews = Object.entries(fileProposals).filter(([n, p]) => p.isNew);
+  return (
+    <Card className="h-full flex flex-col">
+      <CardContent className="flex items-center justify-between">
+        <h3 className="text-sm font-semibold">Files</h3>
+        <div className="flex gap-2">
+          <IconButton onClick={createNewFile}>New</IconButton>
+        </div>
+      </CardContent>
+      <div className="px-3 pb-3 overflow-auto">
+        <ul className="space-y-1">
+          {files.map((f) => {
+            const hasProposal = fileProposals[f.name]?.active;
+            const isDirty = f.name === selectedFile?.name && (!fileContents[f.name] || editorContent !== fileContents[f.name]);
+            return (
+              <li key={f.name}>
+                <button
+                  onClick={() => openFile(f.name)}
+                  className={`w-full text-left px-2 py-1 rounded-md text-white hover:bg-slate-50 ${selectedFile?.name === f.name ? "bg-slate-100 font-medium" : ""}`}
+                >
+                  <div className="flex items-center justify-between">
+                    <span>{f.name}{isDirty ? " *" : ""}</span>
+                    {hasProposal && <div className="w-2 h-2 bg-green-500 rounded-full"></div>}
+                  </div>
+                </button>
+              </li>
+            );
+          })}
+          {proposedNews.map(([name]) => {
+            const prop = fileProposals[name];
+            const isDirty = name === selectedFile?.name && (!fileContents[name] || editorContent !== fileContents[name]);
+            return (
+              <li key={`p-${name}`}>
+                <button
+                  onClick={() => openFile(name)}
+                  className={`w-full text-left px-2 py-1 rounded-md text-white hover:bg-slate-50 ${selectedFile?.name === name ? "bg-slate-100 font-medium" : ""}`}
+                >
+                  <div className="flex items-center justify-between">
+                    <span className="italic text-blue-600">{name} (proposed){isDirty ? " *" : ""}</span>
+                    <div className="w-2 h-2 bg-yellow-500 rounded-full"></div>
+                  </div>
+                </button>
+              </li>
+            );
+          })}
+        </ul>
+      </div>
+      <div className="mt-auto p-3 border-t">
+        <div className="text-xs text-slate-500">App Status: {loading ? "loading" : "ready"}</div>
+      </div>
+    </Card>
+  );
+}
+function DBViewer() {
+  const { dbSchema, setDbSchema, setConsoleOutput, setFileProposal, setShowDiffForFile} = useStore();
+  useEffect(() => {
+    async function loadSchema() {
+      try {
+        const tres = await fetch(`${BASE_URL}/tables`);
+        if (!tres.ok) throw new Error("no tables");
+        const tables = await tres.json();
+        const tablesWithCols = await Promise.all(
+          tables.map(async (t) => {
+            const qres = await fetch(`${BASE_URL}/query`, {
+              method: "POST",
+              headers: { "Content-Type": "application/json" },
+              body: JSON.stringify({ query: `PRAGMA table_info(${t})` }),
+            });
+            const colsData = await qres.json();
+            console.log(colsData)
+            const columns = colsData.map((c) => `${c.name} ${c.type}`);
+            return { name: t, columns };
+          })
+        );
+        setDbSchema({ tables: tables.map((t) => ({ name: t, columns: [] })) });
+      } catch (e) {
+      }
+    }
+    loadSchema();
+  }, [setDbSchema]);
+  async function inspectTable(table) {
+    try {
+      const res = await fetch(`${BASE_URL}/query`, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ query: `SELECT * FROM ${table} LIMIT 50;` }),
+      });
+      const data = await res.json();
+      setConsoleOutput(JSON.stringify(data, null, 2));
+    } catch (e) {
+      setConsoleOutput(`Failed to query table ${table}: ${e.message}`);
+    }
+  }
+  return (
+    <Card className="h-full">
+      <CardContent>
+        <div className="flex items-center justify-between">
+          <h3 className="text-sm font-semibold">DB Viewer</h3>
+          <div className="text-xs text-slate-500">SQLite</div>
+        </div>
+        <div className="mt-3 space-y-3 overflow-auto" style={{ maxHeight: "40vh" }}>
+          {dbSchema.tables.map((t) => (
+            <div key={t.name} className="p-2 rounded border hover:bg-slate-50">
+              <div className="flex justify-between items-center">
+                <div>
+                  <div className="font-medium">{t.name}</div>
+                  <div className="text-xs text-slate-500">{t.columns.length} columns</div>
+                </div>
+                <div className="flex gap-2">
+                  <IconButton onClick={() => inspectTable(t.name)}>Preview</IconButton>
+                </div>
+              </div>
+            </div>
+          ))}
+        </div>
+      </CardContent>
+    </Card>
+  );
+}
+function WSListener() {
+  const { setFileProposal, setShowDiffForFile, selectedFile, pushChat } = useStore();
+  const [isWaiting, setIsWaiting] = useState(true);
+  useEffect(() => {
+    const socket = io("http://localhost:7860");
+    socket.on("connect", () => {
+      console.log("Connected to WebSocket server");
+      setIsWaiting(false);
+    });
+    socket.on("event", (data) => {
+      console.log("WS message:", data);
+      if (data.kind === "code_change") {
+        setFileProposal(data.filename || selectedFile?.name, {
+          code: data.proposedCode,
+          isNew: data.isNew,
+          suggestedName: data.newFileName,
+        });
+        setShowDiffForFile(data.filename || selectedFile?.name);
+      }
+      console.log(data)
+      if (data.kind === "agent_message") {
+        pushChat({ role: "agent", text: `${data.output}`});
+      }
+    });
+    socket.on("disconnect", () => {
+      console.log("Disconnected from WebSocket server");
+      setIsWaiting(true);
+    });
+    return () => {
+      socket.disconnect();
+    };
+  }, []);
+  return isWaiting ? <div>Waiting for agent response...</div> : null;
+}
+function AgentChat() {
+  const { chat, pushChat, setFileProposal, setShowDiffForFile, selectedFile, openFile } = useStore();
+  const [input, setInput] = useState("");
+  const [sending, setSending] = useState(false);
+  async function sendMessage() {
+    if (!input.trim() || sending) return;
+    pushChat({ role: "user", text: input });
+    setInput("");
+    setSending(true);
+    try {
+      const res = await fetch(`${BASE_URL}/run_crew`, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ instructions: input, file: selectedFile?.name }),
+      });
+      const data = await res.json();
+      pushChat({ role: "agent", text: data.status || "Crew is Working..." });
+    } catch (e) {
+      pushChat({ role: "agent", text: `Error: ${e.message}` });
+    } finally {
+      setSending(false);
+    }
+  }
+  return (
+    <Card className="flex flex-col h-full">
+      <CardContent className="flex-1 flex flex-col">
+        <div className="flex items-center justify-between mb-2">
+          <h3 className="text-sm font-semibold">Agent</h3>
+        </div>
+        <div className="flex-1 overflow-auto border rounded p-2 bg-slate-50">
+          {chat.map((m, i) => (
+            <div key={i} className={`mb-2 ${m.role === "user" ? "text-right" : ""}`}>
+              <div
+                className={`${
+                  m.role === "user" ? "inline-block bg-blue-600 text-white" : "inline-block bg-white text-slate-800"
+                } px-3 py-1 rounded-lg`}
+              >
+                {m.text}
+                {m.proposedCode && (
+                  <div className="mt-2 p-2 bg-gray-100 rounded">
+                    <pre className="text-xs overflow-auto">{m.proposedCode}</pre>
+                  </div>
+                )}
+              </div>
+            </div>
+          ))}
+        </div>
+        <div className="mt-2 flex gap-2">
+          <input
+            className="flex-1 border rounded px-3 py-2"
+            value={input}
+            onChange={(e) => setInput(e.target.value)}
+            placeholder="Instruct SQL agent"
+            onKeyDown={(e) => {
+              if (e.key === "Enter") sendMessage();
+            }}
+          />
+          <IconButton onClick={sendMessage} className="bg-blue-600 text-white">
+            {sending ? "..." : "Send"}
+          </IconButton>
+        </div>
+      </CardContent>
+    </Card>
+  );
+}
+function OutputConsole() {
+  const { consoleOutput } = useStore();
+  return (
+    <Card>
+      <CardContent className="p-0">
+        <div className="bg-black text-green-300 font-mono text-sm p-3 h-40 overflow-auto">
+          {consoleOutput || "Console output will appear here."}
+        </div>
+      </CardContent>
+    </Card>
+  );
+}
+function SQLMonacoEditor() {
+  const {
+    editorContent,
+    setEditorContent,
+    selectedFile,
+    showDiffForFile,
+    fileProposals,
+    fileContents,
+    acceptProposal,
+    rejectProposal,
+  } = useStore();
+  const proposal = fileProposals[selectedFile?.name];
+  const showDiff = !!proposal?.active && showDiffForFile === selectedFile?.name;
+  const proposedContent = proposal?.code || "";
+  const isDirty = selectedFile?.name && (!fileContents[selectedFile.name] || editorContent !== fileContents[selectedFile.name]);
+  const saveFile = async () => {
+    let fileName = selectedFile?.name;
+    if (!fileName) {
+      const name = prompt("Save as (must end with .sql):");
+      if (!name || !name.endsWith(".sql")) {
+        alert("Invalid filename");
+        return;
+      }
+      fileName = name;
+      try {
+        const res = await fetch(`${BASE_URL}/files`, {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify({ filename: name, content: editorContent }),
+        });
+        if (!res.ok) throw new Error("Create failed");
+        const currentFiles = useStore.getState().files;
+        useStore.setState({ files: [...currentFiles, { name }] });
+        useStore.setState({ selectedFile: { name }, editorContent, fileContents: { ...useStore.getState().fileContents, [name]: editorContent } });
+        alert("File created and saved");
+        return;
+      } catch (e) {
+        alert("Failed to create file: " + e.message);
+        return;
+      }
+    }
+    try {
+      await fetch(`${BASE_URL}/files/${encodeURIComponent(fileName)}`, {
+        method: "PUT",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ content: editorContent }),
+      });
+      useStore.setState({ fileContents: { ...useStore.getState().fileContents, [fileName]: editorContent } });
+      alert("Saved");
+    } catch (e) {
+      alert("Save failed: " + e.message);
+    }
+  };
+  const runQuery = async () => {
+    let fileName = selectedFile?.name;
+    if (!fileName) {
+      const name = prompt("Save query as (must end with .sql):");
+      if (!name || !name.endsWith(".sql")) {
+        alert("Invalid filename");
+        return;
+      }
+      fileName = name;
+      try {
+        const res = await fetch(`${BASE_URL}/files`, {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify({ filename: name, content: editorContent }),
+        });
+        if (!res.ok) throw new Error("Create failed");
+        const currentFiles = useStore.getState().files;
+        useStore.setState({ files: [...currentFiles, { name }], selectedFile: { name }, editorContent, fileContents: { ...useStore.getState().fileContents, [name]: editorContent } });
+      } catch (e) {
+        alert("Failed to create file: " + e.message);
+        return;
+      }
+    }
+    try {
+      const res = await fetch(`${BASE_URL}/query`, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ query: editorContent }),
+      });
+      if (!res.ok) {
+        let err;
+        try {
+          err = await res.json();
+        } catch {
+          err = "Query failed";
+        }
+        throw new Error(err.error || JSON.stringify(err) || err);
+      }
+      const data = await res.json();
+      useStore.setState({ consoleOutput: JSON.stringify(data, null, 2) });
+    } catch (e) {
+      useStore.setState({ consoleOutput: "Run failed: " + e.message });
+    }
+  };
+  const commonOptions = { minimap: { enabled: false }, fontSize: 13, wordWrap: "on" };
+  const editorKey = `${selectedFile?.name}-${showDiff ? "diff" : "normal"}`;
+  return (
+    <Card className="h-full flex flex-col">
+      <CardContent className="flex-1 flex flex-col p-0">
+        <div className="p-3 border-b flex items-center justify-between">
+          <div>
+            <div className="text-sm font-semibold">{selectedFile?.name || "No file selected"}{isDirty ? " *" : ""}</div>
+            <div className="text-xs text-slate-500">SQL Editor {showDiff ? "(Diff View)" : ""}</div>
+          </div>
+          <div className="flex gap-2">
+            {showDiff ? (
+              <>
+                <IconButton onClick={acceptProposal}>Accept</IconButton>
+                <IconButton onClick={rejectProposal} className="bg-red-500">
+                  Reject
+                </IconButton>
+              </>
+            ) : (
+              <>
+                <IconButton onClick={saveFile}>Save</IconButton>
+                <IconButton onClick={runQuery}>Run</IconButton>
+              </>
+            )}
+          </div>
+        </div>
+        <div key={editorKey} className="flex-1">
+          {showDiff ? (
+            <DiffEditor
+              height="100%"
+              language="sql"
+              original={editorContent}
+              modified={proposedContent}
+              options={{ ...commonOptions, renderSideBySide: false, diffCodeLens: false }}
+            />
+          ) : (
+            <Editor
+              height="100%"
+              defaultLanguage="sql"
+              value={editorContent}
+              onChange={(v) => setEditorContent(v || "")}
+              options={commonOptions}
+            />
+          )}
+        </div>
+      </CardContent>
+    </Card>
+  );
+}
+export default function App() {
+  return (
+    <div className="h-screen w-screen bg-slate-50 p-4 box-border">
+      <div className="grid grid-cols-12 gap-4 h-full">
+        <div className="col-span-2 flex flex-col gap-4 h-full">
+          <FileExplorer />
+          <div className="h-1/3"></div>
+        </div>
+        <div className="col-span-7 flex flex-col gap-4 h-full">
+          <SQLMonacoEditor />
+        </div>
+        <div className="col-span-3 flex flex-col gap-4 h-full">
+          <div className="flex-1">
+            <AgentChat />
+          </div>
+          <div className="h-1/3">
+            <DBViewer />
+          </div>
+          <div className="h-1/3">
+            <OutputConsole />
+          </div>
+        </div>
+      </div>
+      <WSListener />
+    </div>
+  );
+}

frontend/app/src/assets/react.svg ADDED Viewed

frontend/app/src/components/ui/button.jsx ADDED Viewed

	@@ -0,0 +1,54 @@

+import * as React from "react"
+import { Slot } from "@radix-ui/react-slot"
+import { cva } from "class-variance-authority";
+import { cn } from "@/lib/utils"
+const buttonVariants = cva(
+  "inline-flex items-center justify-center gap-2 whitespace-nowrap rounded-md text-sm font-medium transition-all disabled:pointer-events-none disabled:opacity-50 [&_svg]:pointer-events-none [&_svg:not([class*='size-'])]:size-4 shrink-0 [&_svg]:shrink-0 outline-none focus-visible:border-ring focus-visible:ring-ring/50 focus-visible:ring-[3px] aria-invalid:ring-destructive/20 dark:aria-invalid:ring-destructive/40 aria-invalid:border-destructive",
+  {
+    variants: {
+      variant: {
+        default: "bg-primary text-primary-foreground hover:bg-primary/90",
+        destructive:
+          "bg-destructive text-white hover:bg-destructive/90 focus-visible:ring-destructive/20 dark:focus-visible:ring-destructive/40 dark:bg-destructive/60",
+        outline:
+          "border bg-background shadow-xs hover:bg-accent hover:text-accent-foreground dark:bg-input/30 dark:border-input dark:hover:bg-input/50",
+        secondary:
+          "bg-secondary text-secondary-foreground hover:bg-secondary/80",
+        ghost:
+          "hover:bg-accent hover:text-accent-foreground dark:hover:bg-accent/50",
+        link: "text-primary underline-offset-4 hover:underline",
+      },
+      size: {
+        default: "h-9 px-4 py-2 has-[>svg]:px-3",
+        sm: "h-8 rounded-md gap-1.5 px-3 has-[>svg]:px-2.5",
+        lg: "h-10 rounded-md px-6 has-[>svg]:px-4",
+        icon: "size-9",
+      },
+    },
+    defaultVariants: {
+      variant: "default",
+      size: "default",
+    },
+  }
+)
+function Button({
+  className,
+  variant,
+  size,
+  asChild = false,
+  ...props
+}) {
+  const Comp = asChild ? Slot : "button"
+  return (
+    <Comp
+      data-slot="button"
+      className={cn(buttonVariants({ variant, size, className }))}
+      {...props} />
+  );
+}
+export { Button, buttonVariants }

frontend/app/src/components/ui/card.jsx ADDED Viewed

	@@ -0,0 +1,101 @@

+import * as React from "react"
+import { cn } from "@/lib/utils"
+function Card({
+  className,
+  ...props
+}) {
+  return (
+    <div
+      data-slot="card"
+      className={cn(
+        "bg-card text-card-foreground flex flex-col gap-6 rounded-xl border py-6 shadow-sm",
+        className
+      )}
+      {...props} />
+  );
+}
+function CardHeader({
+  className,
+  ...props
+}) {
+  return (
+    <div
+      data-slot="card-header"
+      className={cn(
+        "@container/card-header grid auto-rows-min grid-rows-[auto_auto] items-start gap-1.5 px-6 has-data-[slot=card-action]:grid-cols-[1fr_auto] [.border-b]:pb-6",
+        className
+      )}
+      {...props} />
+  );
+}
+function CardTitle({
+  className,
+  ...props
+}) {
+  return (
+    <div
+      data-slot="card-title"
+      className={cn("leading-none font-semibold", className)}
+      {...props} />
+  );
+}
+function CardDescription({
+  className,
+  ...props
+}) {
+  return (
+    <div
+      data-slot="card-description"
+      className={cn("text-muted-foreground text-sm", className)}
+      {...props} />
+  );
+}
+function CardAction({
+  className,
+  ...props
+}) {
+  return (
+    <div
+      data-slot="card-action"
+      className={cn(
+        "col-start-2 row-span-2 row-start-1 self-start justify-self-end",
+        className
+      )}
+      {...props} />
+  );
+}
+function CardContent({
+  className,
+  ...props
+}) {
+  return (<div data-slot="card-content" className={cn("px-6", className)} {...props} />);
+}
+function CardFooter({
+  className,
+  ...props
+}) {
+  return (
+    <div
+      data-slot="card-footer"
+      className={cn("flex items-center px-6 [.border-t]:pt-6", className)}
+      {...props} />
+  );
+}
+export {
+  Card,
+  CardHeader,
+  CardFooter,
+  CardTitle,
+  CardAction,
+  CardDescription,
+  CardContent,
+}

frontend/app/src/index.css ADDED Viewed

	@@ -0,0 +1,186 @@

+@import "tailwindcss";
+@import "tw-animate-css";
+@custom-variant dark (&:is(.dark *));
+:root {
+  font-family: system-ui, Avenir, Helvetica, Arial, sans-serif;
+  line-height: 1.5;
+  font-weight: 400;
+  color-scheme: light dark;
+  color: rgba(255, 255, 255, 0.87);
+  background-color: #242424;
+  font-synthesis: none;
+  text-rendering: optimizeLegibility;
+  -webkit-font-smoothing: antialiased;
+  -moz-osx-font-smoothing: grayscale;
+  --radius: 0.625rem;
+  --background: oklch(1 0 0);
+  --foreground: oklch(0.145 0 0);
+  --card: oklch(1 0 0);
+  --card-foreground: oklch(0.145 0 0);
+  --popover: oklch(1 0 0);
+  --popover-foreground: oklch(0.145 0 0);
+  --primary: oklch(0.205 0 0);
+  --primary-foreground: oklch(0.985 0 0);
+  --secondary: oklch(0.97 0 0);
+  --secondary-foreground: oklch(0.205 0 0);
+  --muted: oklch(0.97 0 0);
+  --muted-foreground: oklch(0.556 0 0);
+  --accent: oklch(0.97 0 0);
+  --accent-foreground: oklch(0.205 0 0);
+  --destructive: oklch(0.577 0.245 27.325);
+  --border: oklch(0.922 0 0);
+  --input: oklch(0.922 0 0);
+  --ring: oklch(0.708 0 0);
+  --chart-1: oklch(0.646 0.222 41.116);
+  --chart-2: oklch(0.6 0.118 184.704);
+  --chart-3: oklch(0.398 0.07 227.392);
+  --chart-4: oklch(0.828 0.189 84.429);
+  --chart-5: oklch(0.769 0.188 70.08);
+  --sidebar: oklch(0.985 0 0);
+  --sidebar-foreground: oklch(0.145 0 0);
+  --sidebar-primary: oklch(0.205 0 0);
+  --sidebar-primary-foreground: oklch(0.985 0 0);
+  --sidebar-accent: oklch(0.97 0 0);
+  --sidebar-accent-foreground: oklch(0.205 0 0);
+  --sidebar-border: oklch(0.922 0 0);
+  --sidebar-ring: oklch(0.708 0 0);
+}
+a {
+  font-weight: 500;
+  color: #646cff;
+  text-decoration: inherit;
+}
+a:hover {
+  color: #535bf2;
+}
+body {
+  margin: 0;
+  display: flex;
+  place-items: center;
+  min-width: 320px;
+  min-height: 100vh;
+}
+h1 {
+  font-size: 3.2em;
+  line-height: 1.1;
+}
+button {
+  border-radius: 8px;
+  border: 1px solid transparent;
+  padding: 0.6em 1.2em;
+  font-size: 1em;
+  font-weight: 500;
+  font-family: inherit;
+  background-color: #1a1a1a;
+  cursor: pointer;
+  transition: border-color 0.25s;
+}
+button:hover {
+  border-color: #646cff;
+}
+button:focus,
+button:focus-visible {
+  outline: 4px auto -webkit-focus-ring-color;
+}
+@media (prefers-color-scheme: light) {
+  :root {
+    color: #213547;
+    background-color: #ffffff;
+  }
+  a:hover {
+    color: #747bff;
+  }
+  button {
+    background-color: #f9f9f9;
+  }
+}
+@theme inline {
+  --radius-sm: calc(var(--radius) - 4px);
+  --radius-md: calc(var(--radius) - 2px);
+  --radius-lg: var(--radius);
+  --radius-xl: calc(var(--radius) + 4px);
+  --color-background: var(--background);
+  --color-foreground: var(--foreground);
+  --color-card: var(--card);
+  --color-card-foreground: var(--card-foreground);
+  --color-popover: var(--popover);
+  --color-popover-foreground: var(--popover-foreground);
+  --color-primary: var(--primary);
+  --color-primary-foreground: var(--primary-foreground);
+  --color-secondary: var(--secondary);
+  --color-secondary-foreground: var(--secondary-foreground);
+  --color-muted: var(--muted);
+  --color-muted-foreground: var(--muted-foreground);
+  --color-accent: var(--accent);
+  --color-accent-foreground: var(--accent-foreground);
+  --color-destructive: var(--destructive);
+  --color-border: var(--border);
+  --color-input: var(--input);
+  --color-ring: var(--ring);
+  --color-chart-1: var(--chart-1);
+  --color-chart-2: var(--chart-2);
+  --color-chart-3: var(--chart-3);
+  --color-chart-4: var(--chart-4);
+  --color-chart-5: var(--chart-5);
+  --color-sidebar: var(--sidebar);
+  --color-sidebar-foreground: var(--sidebar-foreground);
+  --color-sidebar-primary: var(--sidebar-primary);
+  --color-sidebar-primary-foreground: var(--sidebar-primary-foreground);
+  --color-sidebar-accent: var(--sidebar-accent);
+  --color-sidebar-accent-foreground: var(--sidebar-accent-foreground);
+  --color-sidebar-border: var(--sidebar-border);
+  --color-sidebar-ring: var(--sidebar-ring);
+}
+.dark {
+  --background: oklch(0.145 0 0);
+  --foreground: oklch(0.985 0 0);
+  --card: oklch(0.205 0 0);
+  --card-foreground: oklch(0.985 0 0);
+  --popover: oklch(0.205 0 0);
+  --popover-foreground: oklch(0.985 0 0);
+  --primary: oklch(0.922 0 0);
+  --primary-foreground: oklch(0.205 0 0);
+  --secondary: oklch(0.269 0 0);
+  --secondary-foreground: oklch(0.985 0 0);
+  --muted: oklch(0.269 0 0);
+  --muted-foreground: oklch(0.708 0 0);
+  --accent: oklch(0.269 0 0);
+  --accent-foreground: oklch(0.985 0 0);
+  --destructive: oklch(0.704 0.191 22.216);
+  --border: oklch(1 0 0 / 10%);
+  --input: oklch(1 0 0 / 15%);
+  --ring: oklch(0.556 0 0);
+  --chart-1: oklch(0.488 0.243 264.376);
+  --chart-2: oklch(0.696 0.17 162.48);
+  --chart-3: oklch(0.769 0.188 70.08);
+  --chart-4: oklch(0.627 0.265 303.9);
+  --chart-5: oklch(0.645 0.246 16.439);
+  --sidebar: oklch(0.205 0 0);
+  --sidebar-foreground: oklch(0.985 0 0);
+  --sidebar-primary: oklch(0.488 0.243 264.376);
+  --sidebar-primary-foreground: oklch(0.985 0 0);
+  --sidebar-accent: oklch(0.269 0 0);
+  --sidebar-accent-foreground: oklch(0.985 0 0);
+  --sidebar-border: oklch(1 0 0 / 10%);
+  --sidebar-ring: oklch(0.556 0 0);
+}
+@layer base {
+  * {
+    @apply border-border outline-ring/50;
+  }
+  body {
+    @apply bg-background text-foreground;
+  }
+}

frontend/app/src/lib/utils.js ADDED Viewed

	@@ -0,0 +1,6 @@

+import { clsx } from "clsx";
+import { twMerge } from "tailwind-merge"
+export function cn(...inputs) {
+  return twMerge(clsx(inputs));
+}

frontend/app/src/main.jsx ADDED Viewed

	@@ -0,0 +1,10 @@

+import { StrictMode } from 'react'
+import { createRoot } from 'react-dom/client'
+import './index.css'
+import App from './App.jsx'
+createRoot(document.getElementById('root')).render(
+  <StrictMode>
+    <App />
+  </StrictMode>,
+)

frontend/app/vite.config.js ADDED Viewed

	@@ -0,0 +1,14 @@

+import path from "path"
+import tailwindcss from "@tailwindcss/vite"
+import react from "@vitejs/plugin-react"
+import { defineConfig } from "vite"
+// https://vite.dev/config/
+export default defineConfig({
+  plugins: [react(), tailwindcss()],
+  resolve: {
+    alias: {
+      "@": path.resolve(__dirname, "./src"),
+    },
+  },
+})

poem.txt ADDED Viewed

	@@ -0,0 +1 @@


1	+ CrewAI is the ace of poetic pace, crafting verse with AI in its perfect place.

requirements.txt CHANGED Viewed

Binary files a/requirements.txt and b/requirements.txt differ

sql_files/app.db ADDED Viewed

Binary file (24.6 kB). View file

sql_files/schema.sql ADDED Viewed

	@@ -0,0 +1,17 @@

+CREATE TABLE Fruits (
+  fruit_id INT PRIMARY KEY,
+  name VARCHAR(255),
+  type VARCHAR(100)
+);
+CREATE TABLE Tastes (
+  taste_id INT PRIMARY KEY,
+  description VARCHAR(255)
+);
+CREATE TABLE Fruit_Tastes (
+  fruit_id INT NOT NULL,
+  taste_id INT NOT NULL,
+  CONSTRAINT fk_fruit FOREIGN KEY (fruit_id) REFERENCES Fruits(fruit_id),
+  CONSTRAINT fk_taste FOREIGN KEY (taste_id) REFERENCES Tastes(taste_id)
+);

src/__pycache__/app.cpython-313.pyc ADDED Viewed

Binary file (11.8 kB). View file

src/app.py CHANGED Viewed

@@ -1,11 +1,205 @@
-from flask import Flask
-app = Flask(__name__)
-@app.route("/")
-def home():
-    return "Flask + Ollama on Hugging Face Spaces!"
-@app.route("/health")
-def health():
-    return {"status": "ok"}

+from flask import Flask, request, jsonify, send_from_directory, render_template
+import os
+import sqlite3
+import subprocess
+import threading
+import queue
+from flask import Response
+from flask_cors import CORS
+import json
+import re
+from datetime import datetime
+from flask_socketio import SocketIO, emit
+event_queue = queue.Queue()
+app = Flask(__name__, static_folder='../frontend/app/dist', static_url_path='/')
+socketio = SocketIO(app, cors_allowed_origins="*")
+__all__ = ["app", "socketio"]
+from codeagent.src.codeagent.main import CodeAgentFlow
+CORS(app)
+SQL_FILES_DIR = 'sql_files'
+os.makedirs(SQL_FILES_DIR, exist_ok=True)
+DB_FILE = 'app.db'
+@app.route('/')
+def index():
+    return send_from_directory(app.static_folder, 'index.html')
+@app.route('/files', methods=['GET'])
+def get_files():
+    files = os.listdir(SQL_FILES_DIR)
+    return jsonify([{'name': f} for f in files])
+@app.route('/files', methods=['POST'])
+def create_file():
+    data = request.json
+    filename = data['filename']
+    content = data.get('content', '')
+    filepath = os.path.join(SQL_FILES_DIR, filename)
+    with open(filepath, 'w') as f:
+        f.write(content)
+    return jsonify({"status": "created", "file": filename})
+@app.route('/files/<filename>', methods=['PUT'])
+def update_file(filename):
+    content = request.json.get('content', '')
+    filepath = os.path.join(SQL_FILES_DIR, filename)
+    if not os.path.exists(filepath):
+        return jsonify({"error": "file not found"}), 404
+    with open(filepath, 'w') as f:
+        f.write(content)
+    return jsonify({"status": "updated", "file": filename})
+@app.route('/files/<filename>', methods=['GET'])
+def get_file_content(filename):
+    filepath = os.path.join(SQL_FILES_DIR, filename)
+    if os.path.exists(filepath):
+        with open(filepath, 'r') as f:
+            content = f.read()
+        return jsonify({"filename": filename, "content": content})
+    return jsonify({"error": "file not found"}), 404
+@app.route('/files/<filename>', methods=['DELETE'])
+def delete_file(filename):
+    filepath = os.path.join(SQL_FILES_DIR, filename)
+    if os.path.exists(filepath):
+        os.remove(filepath)
+        return jsonify({"status": "deleted", "file": filename})
+    return jsonify({"error": "file not found"}), 404
+def get_db_connection():
+    conn = sqlite3.connect(os.path.join(SQL_FILES_DIR,DB_FILE))
+    conn.row_factory = sqlite3.Row
+    return conn
+@app.route('/tables', methods=['GET'])
+def get_tables():
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
+    tables = [row['name'] for row in cursor.fetchall()]
+    conn.close()
+    return jsonify(tables)
+recent_result = None
+@app.route('/query', methods=['POST'])
+def run_query():
+    query = request.json.get('query', '')
+    print(query)
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    try:
+        query_lc = str(query).strip().lower()
+        if ";" in query.strip() and not query_lc.startswith("select") and not query_lc.startswith("pragma"):
+            # Multiple statements → executescript
+            cursor.executescript(query)
+            conn.commit()
+            result = {"rows_affected": cursor.rowcount}
+        else:
+            cursor.execute(query)
+            if query_lc.startswith("pragma"):
+                rows = cursor.fetchall()
+                result = [dict(row) for row in rows] if rows else []
+            elif query_lc.startswith("select"):
+                result = [dict(row) for row in cursor.fetchall()]
+            else:
+                conn.commit()
+                result = {"rows_affected": cursor.rowcount}
+        conn.close()
+        global recent_result
+        recent_result = {"query": query, "result": result}
+        return jsonify(result)
+    except Exception as e:
+        conn.close()
+        return jsonify({"error": str(e)}), 400
+@app.route('/recent_query', methods=['GET'])
+def recent_query():
+    return jsonify(recent_result)
+@app.route('/execute_file', methods=['POST'])
+def execute_file():
+    """
+    Execute a .sql file located in SQL_FILES_DIR using sqlite3.executescript.
+    Accepts: { "filename": "schema.sql" }
+    Returns: { "status": "success", "file": "...", "started_at": "...", "finished_at": "..."}
+             or { "error": "message" }
+    """
+    data = request.json or {}
+    filename = data.get('filename')
+    if not filename:
+        return jsonify({"error": "filename is required"}), 400
+    filepath = os.path.join(SQL_FILES_DIR, filename)
+    if not os.path.exists(filepath):
+        return jsonify({"error": f"file not found: {filename}"}), 404
+    started_at = datetime.utcnow().isoformat()
+    conn = get_db_connection()
+    try:
+        with open(filepath, 'r', encoding='utf-8') as f:
+            sql_text = f.read()
+        conn.executescript(sql_text)
+        conn.commit()
+        finished_at = datetime.utcnow().isoformat()
+        return jsonify({
+            "status": "success",
+            "file": filename,
+            "started_at": started_at,
+            "finished_at": finished_at
+        })
+    except Exception as e:
+        conn.rollback()
+        return jsonify({"error": str(e), "file": filename}), 400
+    finally:
+        conn.close()
+def run_crew_task(user_input):
+    codeAgent = CodeAgentFlow(user_input=user_input)
+    codeAgent.kickoff()
+@app.route('/run_crew', methods=['POST'])
+def run_crew():
+    instructions = request.json.get('instructions', '')
+    threading.Thread(target=run_crew_task, args=(instructions,)).start()
+    return jsonify({"status": "Crew is Working..."})
+def send_event(msg):
+    socketio.emit("event", msg)
+@app.route('/send')
+def sendSSE():
+    msg = {
+        "filename": "Users_query.sql",
+        "proposedCode": "SELECT * FROM Users;",
+        "isNew": False,
+        "newFileName": "Users_query.sql",
+        "kind": "code_change"
+    }
+    send_event(msg)
+    return "sent"
+@socketio.on('connect')
+def handle_connect():
+    print("Client connected")
+    emit("event", {"status": "connected"})
+@socketio.on('disconnect')
+def handle_disconnect():
+    print("Client disconnected")
+if __name__ == '__main__':
+    socketio.run(app, host="0.0.0.0", port=7860, debug=False)

src/codeagent/.gitignore ADDED Viewed

	@@ -0,0 +1,4 @@

+.env
+__pycache__/
+lib/
+.DS_Store

src/codeagent/README.md ADDED Viewed

	@@ -0,0 +1,56 @@

+# {{crew_name}} Crew
+Welcome to the {{crew_name}} Crew project, powered by [crewAI](https://crewai.com). This template is designed to help you set up a multi-agent AI system with ease, leveraging the powerful and flexible framework provided by crewAI. Our goal is to enable your agents to collaborate effectively on complex tasks, maximizing their collective intelligence and capabilities.
+## Installation
+Ensure you have Python >=3.10 <3.14 installed on your system. This project uses [UV](https://docs.astral.sh/uv/) for dependency management and package handling, offering a seamless setup and execution experience.
+First, if you haven't already, install uv:
+```bash
+pip install uv
+```
+Next, navigate to your project directory and install the dependencies:
+(Optional) Lock the dependencies and install them by using the CLI command:
+```bash
+crewai install
+```
+### Customizing
+**Add your `OPENAI_API_KEY` into the `.env` file**
+- Modify `src/codeagent/config/agents.yaml` to define your agents
+- Modify `src/codeagent/config/tasks.yaml` to define your tasks
+- Modify `src/codeagent/crew.py` to add your own logic, tools and specific args
+- Modify `src/codeagent/main.py` to add custom inputs for your agents and tasks
+## Running the Project
+To kickstart your flow and begin execution, run this from the root folder of your project:
+```bash
+crewai run
+```
+This command initializes the codeagent Flow as defined in your configuration.
+This example, unmodified, will run the create a `report.md` file with the output of a research on LLMs in the root folder.
+## Understanding Your Crew
+The codeagent Crew is composed of multiple AI agents, each with unique roles, goals, and tools. These agents collaborate on a series of tasks, defined in `config/tasks.yaml`, leveraging their collective skills to achieve complex objectives. The `config/agents.yaml` file outlines the capabilities and configurations of each agent in your crew.
+## Support
+For support, questions, or feedback regarding the {{crew_name}} Crew or crewAI.
+- Visit our [documentation](https://docs.crewai.com)
+- Reach out to us through our [GitHub repository](https://github.com/joaomdmoura/crewai)
+- [Join our Discord](https://discord.com/invite/X4JWnZnxPb)
+- [Chat with our docs](https://chatg.pt/DWjSBZn)
+Let's create wonders together with the power and simplicity of crewAI.

src/codeagent/pyproject.toml ADDED Viewed

	@@ -0,0 +1,21 @@

+[project]
+name = "codeagent"
+version = "0.1.0"
+description = "codeagent using crewAI"
+authors = [{ name = "Your Name", email = "you@example.com" }]
+requires-python = ">=3.10,<3.14"
+dependencies = [
+    "crewai[tools]>=0.177.0,<1.0.0",
+]
+[project.scripts]
+kickoff = "codeagent.main:kickoff"
+run_crew = "codeagent.main:kickoff"
+plot = "codeagent.main:plot"
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+[tool.crewai]
+type = "flow"

src/codeagent/src/codeagent/__init__.py ADDED Viewed

File without changes

src/codeagent/src/codeagent/crews/SqlCrew/__init__.py ADDED Viewed

	@@ -0,0 +1 @@


1	+ """Poem crew template."""

src/codeagent/src/codeagent/crews/SqlCrew/config/agents.yaml ADDED Viewed

	@@ -0,0 +1,56 @@

+manager:
+  role: "Crew Manager"
+  goal: >
+    Understand the user's intent and choose the minimal set of tasks and the best-suited agents
+    to fulfill it safely. Avoid destructive ops unless explicitly requested.
+    If user only needs a SELECT, do not run DDL or DML. Do not allow any agent except
+    the SQL Writing Agent to author SQL queries or write .sql files.
+  backstory: >
+    A seasoned release lead who routes work precisely.
+    Always follows the principle of least privilege and avoids unnecessary steps.
+  allow_delegation: true
+  verbose: true
+schema_agent:
+  role: "Schema Agent"
+  goal: >
+    Design robust, normalized database schemas and validate correctness
+    (tables, columns, types, primary keys, foreign keys).
+    Do NOT write SQL queries or save .sql files.
+  backstory: >
+    A meticulous database engineer focused on relational integrity,
+    evolution-friendly design, and practical constraints.
+  verbose: true
+sql_exec_agent:
+  role: "SQL Execution Agent"
+  goal: >
+    Execute .sql files in the correct order (setup → schema → data),
+    and report results clearly with errors if any.
+  backstory: >
+    A release engineer for database changes, automating the sequence of DDL and DML executions.
+  verbose: true
+code_quality_agent:
+  role: "Code Quality Agent"
+  goal: >
+    Review SQL for best practices, detect anti-patterns, and suggest improvements.
+    Do NOT author new SQL unless explicitly asked to propose a corrected snippet.
+    Never write files.
+  backstory: >
+    A vigilant reviewer who enforces consistency, performance awareness,
+    and safe data modification patterns.
+  verbose: true
+sql_writing_agent:
+  role: "SQL Writing Agent"
+  goal: >
+    Generate safe, performant SQL queries based on the user's intent and the schema context,
+    lint them for common issues, and propose the final SQL via propose_code_changes tool for UI review and save.
+    Only produce SELECT/READ queries unless the user explicitly requests DML/DDL.
+    Avoid SELECT *; prefer explicit columns; document filters at the top of the file.
+  backstory: >
+    An analyst who writes clear, maintainable SQL. Prioritizes readability, safety,
+    and reproducibility. Collaborates with Schema Agent for schema awareness
+    but alone is authorized to write .sql files.
+  verbose: true

src/codeagent/src/codeagent/crews/SqlCrew/config/tasks.yaml ADDED Viewed

	@@ -0,0 +1,33 @@

+tasks:
+  analyze_intent:
+    description: >
+      Analyze the user's request to determine required actions: schema design, query writing, review, or execution.
+      Delegate to appropriate agents only if necessary. Prioritize safety and minimal steps.
+    expected_output: A clear plan of delegated tasks or direct response if no delegation needed.
+    agent: manager
+  design_schema:
+    description: >
+      If schema changes or validation needed, design normalized schema with tables, columns, types, PKs, FKs.
+    expected_output: Detailed schema proposal in text format, ready for SQL generation.
+    agent: schema_agent
+  write_sql:
+    description: >
+      Based on user intent and schema, generate SQL query or script. Lint for issues. Save to .sql file via tool.
+      Default to SELECT unless DML/DDL explicitly requested.
+    expected_output: Generated SQL code saved to file, with lint report.
+    agent: sql_writing_agent
+  review_sql:
+    description: >
+      Review generated or existing SQL for best practices, anti-patterns, performance, safety.
+      Suggest improvements without writing new code unless requested.
+    expected_output: Review report with issues and suggestions.
+    agent: code_quality_agent
+  execute_sql:
+    description: >
+      If execution requested, run .sql files in order (schema first, then data). Report results and errors.
+    expected_output: Execution log with success/failure and output.
+    agent: sql_exec_agent

src/codeagent/src/codeagent/crews/SqlCrew/sqlcrew.py ADDED Viewed

	@@ -0,0 +1,160 @@

+from typing import List
+from crewai import Agent, Crew, Process, Task, LLM
+from crewai.project import CrewBase, agent, crew, task
+from crewai.agents.agent_builder.base_agent import BaseAgent
+import sys
+import os
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../..")))
+from app import socketio
+from codeagent.src.codeagent.tools.custom_tools import execute_sql_file, get_files_and_schema, propose_code_changes, get_recent_query_result
+llm = LLM(model="ollama/llama3.1:8b-instruct-q4_0", base_url="http://localhost:11434")
+def send_chat_message(output: str) -> None:
+    socketio.emit("event", {'kind':'agent_message', 'output': output})
+@CrewBase
+class SqlCrew:
+    agents: List[BaseAgent]
+    tasks: List[Task]
+    # def manager(self) -> Agent:
+    #     return Agent(
+    #         role="Crew Manager",
+    #         goal=(
+    #             "Understand user instruction and delegate tasks according to the instruction, \"{task}\" "
+    #             "Schema-related → Schema Agent, SQL writing → SQL Writing Agent, "
+    #             "execution → SQL Execution Agent, reviews/errors → Code Quality Agent."
+    #         ),
+    #         backstory="A release manager who ensures the right agent handles the right task.",
+    #         llm=llm,
+    #         allow_delegation=True,
+    #         verbose=True,
+    #     )
+    @agent
+    def schema_agent(self) -> Agent:
+        return Agent(
+            role="Schema Agent",
+            goal=(
+                "Check if you have anything to do with the task else delegate to another Co-Worker, Task: {task}"
+                "Design or validate database schema, if it is asked by user"
+                "Focus on tables, columns, keys, and relations. "
+                "Do NOT write SQL files; just propose schema."
+            ),
+            backstory="A careful database engineer ensuring solid relational design.",
+            llm=llm,
+            tools=[get_files_and_schema],
+            allow_delegation=True,
+            step_callback = send_chat_message,
+            verbose=True
+        )
+    @agent
+    def sql_writing_agent(self) -> Agent:
+        return Agent(
+            role="SQL Writing Agent",
+            goal=(
+                "Check if you have anything to do with the task else delegate to another Co-Worker, Task: {task}"
+                "Write safe, performant SQL queries. "
+                "Propose code changes for UI approval instead of direct execution."
+            ),
+            backstory="An SQL analyst who produces clean, maintainable queries.",
+            llm=llm,
+            tools=[propose_code_changes, get_files_and_schema],
+            allow_delegation=True,
+            step_callback = send_chat_message,
+            verbose=True,
+        )
+    @agent
+    def code_quality_agent(self) -> Agent:
+        return Agent(
+            role="Code Quality Agent",
+            goal=(
+                "Check if you have anything to do with the task else delegate to another Co-Worker, Task: {task}"
+                "Review SQL for performance, safety, and best practices. "
+                "Suggest improvements but don’t auto-correct unless asked."
+            ),
+            backstory="A strict reviewer ensuring SQL is clean and safe.",
+            llm=llm,
+            tools=[get_recent_query_result, get_files_and_schema],
+            allow_delegation=True,
+            step_callback = send_chat_message,
+            verbose=True,
+        )
+    @agent
+    def sql_exec_agent(self) -> Agent:
+        return Agent(
+            role="SQL Execution Agent",
+            goal=(
+                "Check if you have anything to do with the task else delegate to another Co-Worker, Task: {task}"
+                "Execute .sql files when asked. "
+                "If execution fails → send to Code Quality Agent. "
+                "If successful → report back to Manager."
+            ),
+            backstory="A release engineer who safely runs DB updates.",
+            llm=llm,
+            tools=[execute_sql_file, get_files_and_schema],
+            allow_delegation=True,
+            step_callback = send_chat_message,
+            verbose=True,
+        )
+    # @task
+    # def analyze_intent(self) -> Task:
+    #     return Task(
+    #         description="Analyze user request and route to the right agent.",
+    #         expected_output="Clear delegation plan or direct response.",
+    #         agent=self.manager(),
+    #     )
+    @task
+    def design_schema(self) -> Task:
+        return Task(
+            description="Propose schema (tables, columns, PKs, FKs) if needed meaning if user specifically said so to create database or to change database.",
+            expected_output="Schema design in text format.",
+            agent=self.schema_agent(),
+        )
+    @task
+    def write_sql(self) -> Task:
+        return Task(
+            description="Generate SQL based on schema and intent. Save via propose_code_changes.",
+            expected_output="SQL file proposal + lint notes.",
+            agent=self.sql_writing_agent(),
+        )
+    @task
+    def execute_sql(self) -> Task:
+        return Task(
+            description="Run SQL files if execution requested. Report results/errors.",
+            expected_output="Execution log with success/failure details.",
+            agent=self.sql_exec_agent(),
+        )
+    @task
+    def review_sql(self) -> Task:
+        return Task(
+            description="Check SQL for issues and suggest improvements.",
+            expected_output="Review report with issues & recommendations.",
+            agent=self.code_quality_agent(),
+        )
+    @crew
+    def crew(self) -> Crew:
+        return Crew(
+            agents=self.agents,
+            tasks=self.tasks,
+            process=Process.hierarchical,
+            manager_llm = llm,
+            verbose=True,
+            step_callback=send_chat_message,
+            task_callback=send_chat_message,
+        )

src/codeagent/src/codeagent/main.py ADDED Viewed

	@@ -0,0 +1,86 @@

+#!/usr/bin/env python
+from pydantic import BaseModel, Field
+from crewai.flow import Flow, listen, start
+from crewai import Agent, LLM
+from typing import Literal
+# from codeagent.src.codeagent.crews.git_crew.git_crew import GitCrew
+from codeagent.src.codeagent.crews.SqlCrew.sqlcrew import SqlCrew
+class FlowState(BaseModel):
+    user_input: str = ""
+    decision: str = ""
+    result: str = ""
+class Decision(BaseModel):
+    decision:str = Field(..., description='identifer of the Crew to Call to process input further, ["GitCrew","SqlCrew"]')
+class DebugLLM(LLM):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+    def call(self, messages, *args, **kwargs):
+        import json
+        print("=== LLM Payload Sent ===")
+        print(json.dumps(messages, indent=2))
+        print("========================")
+        return super().call(messages, *args, **kwargs)
+llm = DebugLLM(model="ollama/llama3.1:8b-instruct-q4_0", base_url="http://localhost:11434")
+router_agent = Agent(
+    role="Task Router",
+    goal = 'Decide which crew (GitCrew or SqlCrew) should handle the user request. ' \
+        'Return the decision as a **JSON object** with one field "decision", ' \
+        'and make sure the value is a string in double quotes, like: {"decision": "GitCrew"} or {"decision": "SqlCrew"}',
+    backstory= "An intelligent orchestrator that analyzes user input and delegates tasks to specialized crews. You have two crews One is GitCrew and Other one is SqlCrew" \
+        "GitCrew description: Handles tasks related to git, repositories, commits, branches, merges, etc." \
+        "SqlCrew description: Handles tasks related to databases, SQL queries, tables, schemas, etc.",
+    llm=llm,
+    verbose=True,
+)
+class CodeAgentFlow(Flow[FlowState]):
+    @start()
+    def decide_crew(self):
+        print("State ------:> ", self.state.user_input)
+        decision_obj = router_agent.kickoff(self.state.user_input, response_format=Decision)
+        print(decision_obj)
+        decision_obj = Decision.model_validate_json(decision_obj.raw)
+        self.state.decision = decision_obj.decision
+        if self.state.decision == "GitCrew":
+            # result = GitCrew().crew().kickoff(inputs={"task": self.state.user_input})
+            # self.state.result = result.raw
+            pass
+        elif self.state.decision == "SqlCrew":
+            result = SqlCrew().crew().kickoff(inputs={"task": self.state.user_input})
+            self.state.result = result.raw
+        else:
+            self.state.result = f"Invalid decision from router: {self.state.decision}"
+    @listen(decide_crew)
+    def save_result(self):
+        print("Saving result...")
+        with open("crew_result.txt", "w") as f:
+            f.write(self.state.result)
+def kickoff(user_input: str):
+    code_agent_flow = CodeAgentFlow()
+    code_agent_flow.state.user_input = user_input
+    code_agent_flow.kickoff()
+def plot():
+    code_agent_flow = CodeAgentFlow()
+    code_agent_flow.plot()
+if __name__ == "__main__":
+    # Example kickoff
+    kickoff("please push my changes to git repository")

src/codeagent/src/codeagent/tools/__init__.py ADDED Viewed

File without changes

src/codeagent/src/codeagent/tools/custom_tools.py ADDED Viewed

	@@ -0,0 +1,64 @@

+import requests
+import json
+from crewai.tools import tool
+import queue
+import sys
+import os
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../..")))
+from app import socketio
+@tool("execute_sql_file")
+def execute_sql_file(filename: str) -> str:
+    """Executes the provided .sql file via Flask and returns the result."""
+    resp = requests.post("http://localhost:7860/execute_file", json={"filename": filename})
+    if resp.status_code == 200:
+        return json.dumps(resp.json())
+    return f"Error: {resp.text}"
+@tool("get_files_and_schema")
+def get_files_and_schema() -> str:
+    """Retrieves all .sql files with contents and full DB schema (tables, columns, types) via Flask."""
+    files_resp = requests.get("http://localhost:7860/files")
+    sql_files = []
+    if files_resp.ok:
+        file_list = [f['name'] for f in files_resp.json() if f['name'].endswith('.sql')]
+        for fname in file_list:
+            content_resp = requests.get(f"http://localhost:7860/files/{fname}")
+            if content_resp.ok:
+                content = content_resp.json().get('content', '')
+                sql_files.append(f"File: {fname}\n{content}\n{'-'*50}")
+    schema = []
+    tables_resp = requests.get("http://localhost:7860/tables")
+    if tables_resp.ok:
+        tables = tables_resp.json()
+        for t in tables:
+            qresp = requests.post("http://localhost:7860/query", json={"query": f"PRAGMA table_info({t})"})
+            if qresp.ok:
+                cols = qresp.json()
+                col_str = '\n'.join([f"{c['name']} {c.get('type', 'UNKNOWN')}" for c in cols])
+                schema.append(f"Table: {t}\n{col_str}\n{'-'*50}")
+    return f"SQL Files:\n{chr(10).join(sql_files)}\n\nDB Schema:\n{chr(10).join(schema) or "Empty"}"
+@tool("propose_code_changes")
+def propose_code_changes(filename: str, code: str, is_new: bool = False, suggested_name: str = None) -> str:
+    """Proposes code changes or new file for frontend UI via WebSocket."""
+    proposal = {
+        "kind": "code_change",
+        "proposedCode": code,
+        "newFileName": suggested_name if is_new else None,
+        "isNew": is_new,
+        "filename": filename
+    }
+    socketio.emit("event", proposal)
+    return "Proposal sent to frontend via WebSocket"
+@tool("get_recent_query_result")
+def get_recent_query_result() -> str:
+    """Retrieves the most recent SQL query execution result via Flask."""
+    resp = requests.get("http://localhost:7860/recent_query")
+    if resp.status_code == 200:
+        return resp.json().get('result', 'No recent result')
+    return "Error fetching recent result"