Just set up mongodb to sotre day trancriptions

Author: AryanFP

bun.lock

@@ -5,6 +5,7 @@
   "scripts": {
     "start": "bun src/server/index.ts",
     "dev": "concurrently \"npm run dev:backend\" \"npm run dev:frontend\"",
+    "dev:d": "doppler run -- bun run dev",
     "dev:backend": "bun --hot src/server/index.ts",
     "dev:frontend": "cd src/frontend && npm run dev",
     "build:frontend": "cd src/frontend && npm run build",
@@ -17,7 +18,7 @@
   "description": "Example app for taking photos with MentraOS",
   "dependencies": {
     "@mentra/react": "^2.0.2",
-    "@mentra/sdk": "^3.0.0-hono.2",
+    "@mentra/sdk": "^2.1.29",
     "ejs": "^3.1.10",
     "express": "^4.21.2",
     "http-proxy-middleware": "^3.0.5",

package.json

@@ -0,0 +1,9 @@
+export class DisplayManager {
+ 
+
+private test : string;
+
+constructor () {
+    this.test = "test";
+  }
+}

src/server/modules/transcription.ts

@@ -1,4 +1,5 @@
 import { TranscriptionData } from "@mentra/sdk";
+import { Transcript as TranscriptModel } from "../schema/transcript.schema";
 
 export interface TranscriptSegment {
   speakerId: string;
@@ -11,8 +12,16 @@ export interface TranscriptSegment {
 
 export class Transcript {
   private segments: TranscriptSegment[] = [];
+  private userEmail: string;
+  private directionizationId: string;
 
-  addSegment(data: TranscriptionData): void {
+  constructor(userEmail: string, directionizationId: string) {
+    this.userEmail = userEmail;
+    this.directionizationId = directionizationId;
+  }
+
+  // Converts incoming transcription data into a segment and adds it to the transcript if finalized
+  async addSegment(data: TranscriptionData): Promise<void> {
     const segment: TranscriptSegment = {
       speakerId: data.speakerId ?? "unknown",
       text: data.text,
@@ -24,17 +33,31 @@ export class Transcript {
 
     if (data.isFinal) {
       this.segments.push(segment);
+
+      // Save to MongoDB when finalized
+      try {
+        await TranscriptModel.create({
+          userEmail: this.userEmail,
+          directionizationId: segment.speakerId,
+          content: segment.text,
+        });
+      } catch (error) {
+        console.error("Failed to save transcript to MongoDB:", error);
+      }
     }
   }
 
+  // Returns a copy of all transcript segments
   getSegments(): TranscriptSegment[] {
     return [...this.segments];
   }
 
+  // Combines all segment text into a single string with spaces between segments
   getFullText(): string {
     return this.segments.map((s) => s.text).join(" ");
   }
 
+  // Clears all transcript segments
   clear(): void {
     this.segments = [];
   }

src/shared/class/DisplayManger.ts

@@ -1,20 +1,22 @@
-import { AppSession, TranscriptionData } from "@mentra/sdk";
+import { AppSession, TranscriptionData,  } from "@mentra/sdk";
 import { Transcript } from "./Transcript";
 import { sendTranscription } from "../../server/stream/sse";
-
+import { DisplayManager } from "./DisplayManger";
 export class User {
   private static readonly sessions: Map<string, User> = new Map();
 
   readonly userId: string;
   readonly session: AppSession;
   readonly transcript: Transcript;
-
+  readonly display: DisplayManager;
+  
   private transcriptionCleanup: (() => void) | null = null;
 
   constructor(session: AppSession) {
     this.userId = session.userId;
     this.session = session;
-    this.transcript = new Transcript();
+    this.transcript = new Transcript(session.userId, session.userId);
+    this.display = new DisplayManager();
 
     User.sessions.set(this.userId, this);
   }
@@ -28,11 +30,11 @@ export class User {
     }
 
     this.transcriptionCleanup = this.session.events.onTranscription(
-      (data: TranscriptionData) => {
+      async (data: TranscriptionData) => {
         const speaker = data.speakerId ?? "unknown";
         console.log(` ${this.userId}, [${speaker}]: ${data.text} (final: ${data.isFinal})`);
 
-        this.transcript.addSegment(data);
+        await this.transcript.addSegment(data);
         sendTranscription(this.userId, data);
       }
     );

src/shared/class/Transcript.ts

@@ -0,0 +1,32 @@
+import { Schema, model } from "mongoose";
+import { getFormatedDate } from "./util";
+
+const transcriptChunkSchema = new Schema({
+  userEmail: {
+    type: String,
+    required: true,
+    index: true,
+  },
+
+  directionizationId: {
+    type: String,
+    required: true,
+  },
+
+  content: {
+    type: String,
+    required: true,
+  },
+
+  createdAt: {
+    type: Date,
+    default: Date.now,
+  },
+
+  date: {
+    type: String,
+    default: () => getFormatedDate(),
+  },
+});
+
+export const Transcript = model("Transcript", transcriptChunkSchema);