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"],

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,

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: