openreplay/frontend/app/mstore/aiSummaryStore.ts

import { aiService } from 'App/services';
import { makeAutoObservable } from 'mobx';

export default class AiSummaryStore {
  text = '';

  constructor() {
    makeAutoObservable(this);
  }

  setText(text: string) {
    this.text = text;
  }

  appendText(text: string) {
    this.text += text;
  }

  getSummary = async (sessionId: string) => {
    this.setText('');
    const respBody = await aiService.getSummary(sessionId);
    if (!respBody) return;

    const reader = respBody.getReader();

    let lastIncompleteWord = '';

    const processTextChunk = (textChunk: string) => {
      textChunk = lastIncompleteWord + textChunk;
      const words = textChunk.split(' ');

      lastIncompleteWord = words.pop() || '';

      words.forEach((word) => {
        if(word) this.appendText(word + ' ');
      });
    };

    try {
      while (true) {
        const { done, value } = await reader.read();
        if (done) {
          // Processing any remaining incomplete word at the end of the stream
          if (lastIncompleteWord) {
            this.appendText(lastIncompleteWord + ' ');
          }
          break;
        }
        let textChunk = new TextDecoder().decode(value, { stream: true });
        if (this.text === '') {
          textChunk = textChunk.trimStart()
        }
        processTextChunk(textChunk);
      }
    } catch (error) {
      console.log(error);
    }
  };
}