From 7ddd625bf85613405e82b9ac6f7ce15ed7fda6ea Mon Sep 17 00:00:00 2001
From: Dylan De Faoite <dylanseandefaoite@gmail.com>
Date: Sun, 1 Mar 2026 16:40:00 +0000
Subject: [PATCH] fix: database schema missing type column

---
 db/database.py | 4 +++-
 db/schema.sql  | 1 +
 server/app.py  | 3 +--
 3 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/db/database.py b/db/database.py
index 9f1a776..80cf5b3 100644
--- a/db/database.py
+++ b/db/database.py
@@ -64,6 +64,7 @@ class PostgresConnector:
         query = """
             INSERT INTO events (
                 dataset_id,
+                type,
                 parent_id,
                 author,
                 content,
@@ -87,7 +88,7 @@ class PostgresConnector:
                 %s, %s, %s, %s, %s,
                 %s, %s, %s, %s, %s,
                 %s, %s, %s, %s, %s,
-                %s, %s, %s, %s
+                %s, %s, %s, %s, %s
             )
         """
 
@@ -96,6 +97,7 @@ class PostgresConnector:
         for _, row in event_data.iterrows():
             values.append((
                 dataset_id,
+                row["type"],
                 row["parent_id"],
                 row["author"],
                 row["content"],
diff --git a/db/schema.sql b/db/schema.sql
index 4ad2ac6..693f821 100644
--- a/db/schema.sql
+++ b/db/schema.sql
@@ -20,6 +20,7 @@ CREATE TABLE events (
     /* Required Fields */
     id SERIAL PRIMARY KEY,
     dataset_id INTEGER NOT NULL,
+    type VARCHAR(255) NOT NULL,
 
     author VARCHAR(255) NOT NULL,
     content TEXT NOT NULL,
diff --git a/server/app.py b/server/app.py
index 2f8a9e8..b3830a3 100644
--- a/server/app.py
+++ b/server/app.py
@@ -136,7 +136,7 @@ def upload_data():
         db.save_dataset_content(dataset_id, enriched_df)
 
         return jsonify(
-            {"message": "File uploaded successfully", "event_count": len(enriched_df)}
+            {"message": "File uploaded successfully", "event_count": len(enriched_df), "dataset_id": dataset_id}
         ), 200
     except ValueError as e:
         return jsonify({"error": f"Failed to read JSONL file: {str(e)}"}), 400
@@ -171,7 +171,6 @@ def content_endpoint(dataset_id):
         return jsonify({"error": "Unauthorized access to dataset"}), 403
 
     dataset_content = db.get_dataset_content(dataset_id)
-
     try:
         return jsonify(stat_gen.get_content_analysis(dataset_content)), 200
     except ValueError as e: