+
{(data && typeof window !== "undefined") ? (
+
+
+
+
+
+
+
+
back
+
+
+
+
Account
+
Manage your account's settings.
+
{account.name}
+
+
+
Plan
+
You are using open-source version. Subscribe to get access to hosted cognee with your data!
+
+
Select a plan
+
+
+
+
+
+
+
+
+
+
+ );
+}
diff --git a/cognee-frontend/src/app/account/page.tsx b/cognee-frontend/src/app/account/page.tsx
new file mode 100644
index 000000000..f6323c313
--- /dev/null
+++ b/cognee-frontend/src/app/account/page.tsx
@@ -0,0 +1 @@
+export { default } from "./Account";
diff --git a/cognee-frontend/src/app/dashboard/AddDataToCognee.tsx b/cognee-frontend/src/app/dashboard/AddDataToCognee.tsx
new file mode 100644
index 000000000..4f9a3179e
--- /dev/null
+++ b/cognee-frontend/src/app/dashboard/AddDataToCognee.tsx
@@ -0,0 +1,116 @@
+import { FormEvent, useCallback, useState } from "react";
+import { CloseIcon, PlusIcon } from "@/ui/Icons";
+import { useModal } from "@/ui/elements/Modal";
+import { CTAButton, GhostButton, IconButton, Modal, NeutralButton, Select } from "@/ui/elements";
+
+import addData from "@/modules/ingestion/addData";
+import { Dataset } from "@/modules/ingestion/useDatasets";
+import cognifyDataset from "@/modules/datasets/cognifyDataset";
+
+interface AddDataToCogneeProps {
+ datasets: Dataset[];
+ refreshDatasets: () => void;
+ useCloud?: boolean;
+}
+
+export default function AddDataToCognee({ datasets, refreshDatasets, useCloud = false }: AddDataToCogneeProps) {
+ const [filesForUpload, setFilesForUpload] = useState
(null);
+
+ const prepareFiles = useCallback((event: FormEvent) => {
+ const formElements = event.currentTarget;
+ const files = formElements.files;
+
+ setFilesForUpload(files);
+ }, []);
+
+ const processDataWithCognee = useCallback((state: object, event?: FormEvent) => {
+ event!.preventDefault();
+
+ if (!filesForUpload) {
+ return;
+ }
+
+ const formElements = event!.currentTarget;
+ const datasetId = formElements.datasetName.value;
+
+ return addData(
+ datasetId ? {
+ id: datasetId,
+ } : {
+ name: "main_dataset",
+ },
+ Array.from(filesForUpload),
+ useCloud
+ )
+ .then(({ dataset_id, dataset_name }) => {
+ refreshDatasets();
+ setFilesForUpload(null);
+
+ return cognifyDataset({
+ id: dataset_id,
+ name: dataset_name,
+ data: [], // not important, just to mimick Dataset
+ status: "", // not important, just to mimick Dataset
+ }, useCloud);
+ });
+ }, [filesForUpload, refreshDatasets, useCloud]);
+
+ const {
+ isModalOpen: isAddDataModalOpen,
+ openModal: openAddDataModal,
+ closeModal: closeAddDataModal,
+ isActionLoading: isProcessingDataWithCognee,
+ confirmAction: submitDataToCognee,
+ } = useModal(false, processDataWithCognee);
+
+ return (
+ <>
+
+
+ Add data to cognee
+
+
+
+
+
+ Add new data to a dataset?
+
+
+
Please select a dataset to add data in.
If you don't have any, don't worry, we will create one for you.
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ {selectedNotebook && (
+
+ )}
+
+
+
+ {datasets.length === 0 && (
+
+ No datasets here, add one by clicking +
+
+ )}
+ {datasets.map((dataset) => {
+ return (
+
+ {datasetInProcessing?.id == dataset.id ? : }
+ {dataset.name}
+
+
handleDatasetRemove(dataset)} className="hover:bg-gray-100 w-full text-left px-2 cursor-pointer">delete
+
+ No data in a dataset, add by clicking "add data" in a dropdown menu
+
+ )}
+ {dataset.data?.map((data) => (
+
+
{data.name}
+
+ handleDataRemove(data)}>
+
+
+
-Try cognee UI out locally [here](https://docs.cognee.ai/how-to-guides/cognee-ui).
+Try cognee UI by runnning ``` cognee -ui ``` command on your terminal.
## Understand our architecture
diff --git a/alembic/versions/211ab850ef3d_add_sync_operations_table.py b/alembic/versions/211ab850ef3d_add_sync_operations_table.py
new file mode 100644
index 000000000..f22c7c6e2
--- /dev/null
+++ b/alembic/versions/211ab850ef3d_add_sync_operations_table.py
@@ -0,0 +1,98 @@
+"""Add sync_operations table
+
+Revision ID: 211ab850ef3d
+Revises: 9e7a3cb85175
+Create Date: 2025-09-10 20:11:13.534829
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "211ab850ef3d"
+down_revision: Union[str, None] = "9e7a3cb85175"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+ # ### commands auto generated by Alembic - please adjust! ###
+
+ # Check if table already exists (it might be created by Base.metadata.create_all() in initial migration)
+ connection = op.get_bind()
+ inspector = sa.inspect(connection)
+
+ if "sync_operations" not in inspector.get_table_names():
+ # Table doesn't exist, create it normally
+ op.create_table(
+ "sync_operations",
+ sa.Column("id", sa.UUID(), nullable=False),
+ sa.Column("run_id", sa.Text(), nullable=True),
+ sa.Column(
+ "status",
+ sa.Enum(
+ "STARTED",
+ "IN_PROGRESS",
+ "COMPLETED",
+ "FAILED",
+ "CANCELLED",
+ name="syncstatus",
+ create_type=False,
+ ),
+ nullable=True,
+ ),
+ sa.Column("progress_percentage", sa.Integer(), nullable=True),
+ sa.Column("dataset_ids", sa.JSON(), nullable=True),
+ sa.Column("dataset_names", sa.JSON(), nullable=True),
+ sa.Column("user_id", sa.UUID(), nullable=True),
+ sa.Column("created_at", sa.DateTime(timezone=True), nullable=True),
+ sa.Column("started_at", sa.DateTime(timezone=True), nullable=True),
+ sa.Column("completed_at", sa.DateTime(timezone=True), nullable=True),
+ sa.Column("total_records_to_sync", sa.Integer(), nullable=True),
+ sa.Column("total_records_to_download", sa.Integer(), nullable=True),
+ sa.Column("total_records_to_upload", sa.Integer(), nullable=True),
+ sa.Column("records_downloaded", sa.Integer(), nullable=True),
+ sa.Column("records_uploaded", sa.Integer(), nullable=True),
+ sa.Column("bytes_downloaded", sa.Integer(), nullable=True),
+ sa.Column("bytes_uploaded", sa.Integer(), nullable=True),
+ sa.Column("dataset_sync_hashes", sa.JSON(), nullable=True),
+ sa.Column("error_message", sa.Text(), nullable=True),
+ sa.Column("retry_count", sa.Integer(), nullable=True),
+ sa.PrimaryKeyConstraint("id"),
+ )
+ op.create_index(
+ op.f("ix_sync_operations_run_id"), "sync_operations", ["run_id"], unique=True
+ )
+ op.create_index(
+ op.f("ix_sync_operations_user_id"), "sync_operations", ["user_id"], unique=False
+ )
+ else:
+ # Table already exists, but we might need to add missing columns or indexes
+ # For now, just log that the table already exists
+ print("sync_operations table already exists, skipping creation")
+
+ # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+ # ### commands auto generated by Alembic - please adjust! ###
+
+ # Only drop if table exists (might have been created by Base.metadata.create_all())
+ connection = op.get_bind()
+ inspector = sa.inspect(connection)
+
+ if "sync_operations" in inspector.get_table_names():
+ op.drop_index(op.f("ix_sync_operations_user_id"), table_name="sync_operations")
+ op.drop_index(op.f("ix_sync_operations_run_id"), table_name="sync_operations")
+ op.drop_table("sync_operations")
+
+ # Drop the enum type that was created (only if no other tables are using it)
+ sa.Enum(name="syncstatus").drop(op.get_bind(), checkfirst=True)
+ else:
+ print("sync_operations table doesn't exist, skipping downgrade")
+
+ # ### end Alembic commands ###
diff --git a/alembic/versions/45957f0a9849_add_notebook_table.py b/alembic/versions/45957f0a9849_add_notebook_table.py
new file mode 100644
index 000000000..ded7ccd9c
--- /dev/null
+++ b/alembic/versions/45957f0a9849_add_notebook_table.py
@@ -0,0 +1,46 @@
+"""Add notebook table
+
+Revision ID: 45957f0a9849
+Revises: 9e7a3cb85175
+Create Date: 2025-09-10 17:47:58.201319
+
+"""
+
+from datetime import datetime, timezone
+from uuid import uuid4
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "45957f0a9849"
+down_revision: Union[str, None] = "9e7a3cb85175"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+ bind = op.get_bind()
+ inspector = sa.inspect(bind)
+
+ if "notebooks" not in inspector.get_table_names():
+ # Define table with all necessary columns including primary key
+ op.create_table(
+ "notebooks",
+ sa.Column("id", sa.UUID, primary_key=True, default=uuid4), # Critical for SQLite
+ sa.Column("owner_id", sa.UUID, index=True),
+ sa.Column("name", sa.String(), nullable=False),
+ sa.Column("cells", sa.JSON(), nullable=False),
+ sa.Column("deletable", sa.Boolean(), default=True),
+ sa.Column("created_at", sa.DateTime(), default=lambda: datetime.now(timezone.utc)),
+ )
+
+
+def downgrade() -> None:
+ bind = op.get_bind()
+ inspector = sa.inspect(bind)
+
+ if "notebooks" in inspector.get_table_names():
+ op.drop_table("notebooks")
diff --git a/alembic/versions/8057ae7329c2_initial_migration.py b/alembic/versions/8057ae7329c2_initial_migration.py
index 48e795327..aa0ecd4b8 100644
--- a/alembic/versions/8057ae7329c2_initial_migration.py
+++ b/alembic/versions/8057ae7329c2_initial_migration.py
@@ -19,6 +19,7 @@ depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
db_engine = get_relational_engine()
+ # we might want to delete this
await_only(db_engine.create_database())
diff --git a/assets/cognee-new-ui.webp b/assets/cognee-new-ui.webp
new file mode 100644
index 000000000..81e408de3
Binary files /dev/null and b/assets/cognee-new-ui.webp differ
diff --git a/cognee-frontend/public/next.svg b/cognee-frontend/public/next.svg
deleted file mode 100644
index 5174b28c5..000000000
--- a/cognee-frontend/public/next.svg
+++ /dev/null
@@ -1 +0,0 @@
-
\ No newline at end of file
diff --git a/cognee-frontend/public/vercel.svg b/cognee-frontend/public/vercel.svg
deleted file mode 100644
index d2f842227..000000000
--- a/cognee-frontend/public/vercel.svg
+++ /dev/null
@@ -1 +0,0 @@
-
\ No newline at end of file
diff --git a/cognee-frontend/public/videos/background-video-blur.mp4 b/cognee-frontend/public/videos/background-video-blur.mp4
new file mode 100644
index 000000000..23d7ecdba
Binary files /dev/null and b/cognee-frontend/public/videos/background-video-blur.mp4 differ
diff --git a/cognee-frontend/src/app/(graph)/CogneeAddWidget.tsx b/cognee-frontend/src/app/(graph)/CogneeAddWidget.tsx
index 434ceb0c6..3efdb47af 100644
--- a/cognee-frontend/src/app/(graph)/CogneeAddWidget.tsx
+++ b/cognee-frontend/src/app/(graph)/CogneeAddWidget.tsx
@@ -26,9 +26,10 @@ export interface NodesAndEdges {
interface CogneeAddWidgetProps {
onData: (data: NodesAndLinks) => void;
+ useCloud?: boolean;
}
-export default function CogneeAddWidget({ onData }: CogneeAddWidgetProps) {
+export default function CogneeAddWidget({ onData, useCloud = false }: CogneeAddWidgetProps) {
const {
datasets,
refreshDatasets,
@@ -76,17 +77,18 @@ export default function CogneeAddWidget({ onData }: CogneeAddWidgetProps) {
return addData(dataset, files)
.then(() => {
- const onUpdate = (data: NodesAndEdges) => {
- onData({
- nodes: data.nodes,
- links: data.edges,
- });
- setProcessingFilesDone();
- };
+ // const onUpdate = (data: NodesAndEdges) => {
+ // onData({
+ // nodes: data.nodes,
+ // links: data.edges,
+ // });
+ // setProcessingFilesDone();
+ // };
- return cognifyDataset(dataset, onUpdate)
+ return cognifyDataset(dataset, useCloud)
.then(() => {
refreshDatasets();
+ setProcessingFilesDone();
});
});
};
diff --git a/cognee-frontend/src/app/(graph)/GraphVisualization.tsx b/cognee-frontend/src/app/(graph)/GraphVisualization.tsx
index 67d6458f8..4e2d1e642 100644
--- a/cognee-frontend/src/app/(graph)/GraphVisualization.tsx
+++ b/cognee-frontend/src/app/(graph)/GraphVisualization.tsx
@@ -1,5 +1,6 @@
"use client";
+import classNames from "classnames";
import { MutableRefObject, useEffect, useImperativeHandle, useRef, useState, useCallback } from "react";
import { forceCollide, forceManyBody } from "d3-force-3d";
import ForceGraph, { ForceGraphMethods, GraphData, LinkObject, NodeObject } from "react-force-graph-2d";
@@ -10,6 +11,7 @@ interface GraphVisuzaliationProps {
ref: MutableRefObject