Merge commit '2f5efee9c7f81cadc22f033f9ba20bb5cbc2d66e' into develop

2026-06-15 18:01:52 +02:00 · 2025-11-24 14:41:10 +01:00
parent e73a2a8203 2f5efee9c7
commit 34d644d7d7
9 changed files with 12757 additions and 12 deletions
@@ -0,0 +1,138 @@
+const fs = require("fs");
+const path = require("path");
+
+// Prepare output directory (always storage/transcriptionSummaries under project root)
+const outputDir = `${__dirname}/../../../storage/transcriptionSummaries`;
+if (!fs.existsSync(outputDir)) {
+    fs.mkdirSync(outputDir, { recursive: true });
+}
+
+//Speaker, ALL-Sentences, Start, End
+
+module.exports = {
+    name: "summarize-transcription", // Unique name for our function that will later be used to get the function from the map via "mapFunctions.get("example").function()"
+    type: "summarizer", // value used to differentiate each module to order them in the UI
+    displayname: "Summarizer",  // The displayname used within the UI
+    async function(args) {
+        let inputJson = args.json;
+
+        //JSON Path
+        if (args.jsonPath) {
+            try {
+                const raw = fs.readFileSync(args.jsonPath, "utf-8");
+                inputJson = JSON.parse(raw);
+            } catch (e) {
+                console.error("Failed to load JSON from file:", e);
+                return { error: "Could not read JSON from file path." };
+            }
+        }
+        // JSON parsen
+        if (typeof inputJson === "string") {
+            try {
+                inputJson = JSON.parse(inputJson);
+            } catch (e) {
+                console.log("Invalid JSON in summarize-transcription");
+                return { error: "Invalid JSON" };
+            }
+        }
+
+        const words = inputJson.words;
+        if (!Array.isArray(words)) {
+            return { error: "No words Array found" };
+        }
+
+        const ENDINGS = [".", "!", "?"]; // '...' auch als Satzende ?
+        const ABBREVIATIONS = new Set(["z.B.", "bzw.", "u.a.", "Dr.", "Mr.", "Mrs.", "Prof.", "etc."]); //TODO weitere Ergaenzen
+
+        const result = [];
+        let currentSentence = "";
+        let currentSpeaker = null;
+        let startTime = null;
+        let endTime = null;
+
+        for (const w of words) {
+            if (!currentSpeaker) currentSpeaker = w.speaker;
+            if (startTime === null) startTime = w.start;
+            endTime = w.end;
+    
+            //speaker changing
+            if (currentSpeaker !== w.speaker && currentSentence) { 
+                const lastEntry = result[result.length - 1];
+                if (lastEntry && lastEntry.speaker === currentSpeaker) {
+                    lastEntry.sentence += " " + currentSentence;
+                    lastEntry.end = endTime;
+                } else {
+                    result.push({
+                        speaker: currentSpeaker,
+                        sentence: currentSentence,
+                        start: startTime,
+                        end: endTime
+                    });
+                }
+                currentSentence = "";
+                startTime = w.start;
+            }
+            currentSpeaker = w.speaker;
+            currentSentence += (currentSentence ? " " : "") + w.text; //sentence beginning or not
+            const lastWord = w.text.trim();
+            const lastChar = lastWord.slice(-1);
+            const isAbbreviation = ABBREVIATIONS.has(lastWord);
+
+            //sentence ending
+            if (ENDINGS.includes(lastChar) && !isAbbreviation) {
+                const lastEntry = result[result.length - 1];
+                if (lastEntry && lastEntry.speaker === currentSpeaker) {
+                    lastEntry.sentence += " " + currentSentence;
+                    lastEntry.end = endTime;
+                } else {
+                    result.push({
+                        speaker: currentSpeaker,
+                        sentence: currentSentence,
+                        start: startTime,
+                        end: endTime
+                    });
+                }
+                currentSentence = "";
+                startTime = null;
+                endTime = null;
+                currentSpeaker = null;
+            }
+        }
+
+        // safe last sentence 
+        if (currentSentence) {
+            const lastEntry = result[result.length - 1];
+            if (lastEntry && lastEntry.speaker === currentSpeaker) {
+                lastEntry.sentence += " " + currentSentence;
+                lastEntry.end = endTime;
+            } else {
+                result.push({
+                    speaker: currentSpeaker,
+                    sentence: currentSentence,
+                    start: startTime,
+                    end: endTime
+                });
+            }
+        }
+
+        // Output as Text
+        const output = result.map(r => 
+            `Sprecher ${r.speaker} [${r.start.toFixed(2)} - ${r.end.toFixed(2)}]: ${r.sentence}`
+        );
+
+        // Output on cosole
+        //console.log("\n------------\nMerged Transcription Result:\n", output, "\n------------\n");
+
+        try {
+            const jsonPath = path.join(outputDir, "transcription_result.json");
+            fs.writeFileSync(jsonPath, JSON.stringify(result, null, 2), "utf-8");
+
+            const txtPath = path.join(outputDir, "transcription_result.txt");
+            fs.writeFileSync(txtPath, output.join("\n"), "utf-8");
+
+            console.log(`Summary successfully saved:\n- ${jsonPath}\n- ${txtPath}`);
+        } catch (err) {
+            console.error("Error saving Summary:", err);
+        }
+    }
+}
@@ -0,0 +1,121 @@
+const fs = require("fs");
+const path = require("path");
+
+// Prepare output directory (always storage/transcriptionSummaries under project root)
+const outputDir = `${__dirname}/../../../storage/transcriptionSummaries`;
+if (!fs.existsSync(outputDir)) {
+    fs.mkdirSync(outputDir, { recursive: true });
+}
+
+//Speaker, Sentence, Start, End
+
+module.exports = {
+    name: "summarize-transcription2", // Unique name for our function that will later be used to get the function from the map via "mapFunctions.get("example").function()"
+    type: "summarizer", // value used to differentiate each module to order them in the UI
+    displayname: "Summarizer",  // The displayname used within the UI
+    async function(args) {
+        let inputJson = args.json;
+
+        //JSON Path
+        if (args.jsonPath) {
+            try {
+                const raw = fs.readFileSync(args.jsonPath, "utf-8");
+                inputJson = JSON.parse(raw);
+            } catch (e) {
+                console.error("Failed to load JSON from file:", e);
+                return { error: "Could not read JSON from file path." };
+            }
+        }
+        // JSON parsen
+        if (typeof inputJson === "string") {
+            try {
+                inputJson = JSON.parse(inputJson);
+            } catch (e) {
+                console.log("Invalid JSON in summarize-transcription");
+                return { error: "Invalid JSON" };
+            }
+        }
+
+        const words = inputJson.words;
+        if (!Array.isArray(words)) {
+            return { error: "No words Array found" };
+        }
+
+        const ENDINGS = [".", "!", "?"]; // '...' auch als Satzende ?
+        const ABBREVIATIONS = new Set(["z.B.", "bzw.", "u.a.", "Dr.", "Mr.", "Mrs.", "Prof.", "etc."]); //TODO weitere Ergaenzen
+
+        const result = [];
+        let currentSentence = "";
+        let currentSpeaker = null;
+        let startTime = null;
+        let endTime = null;
+
+        for (const w of words) {
+            if (!currentSpeaker) currentSpeaker = w.speaker;
+            if (startTime === null) startTime = w.start;
+            endTime = w.end;
+    
+            //speaker changing
+             if (currentSpeaker !== w.speaker && currentSentence) {
+                result.push({
+                    speaker: currentSpeaker,
+                    sentence: currentSentence,
+                    start: startTime,
+                    end: endTime
+                });
+                currentSentence = "";
+                startTime = w.start;
+            }
+            currentSpeaker = w.speaker;
+            currentSentence += (currentSentence ? " " : "") + w.text; //sentence beginning or not
+            const lastWord = w.text.trim();
+            const lastChar = lastWord.slice(-1);
+            const isAbbreviation = ABBREVIATIONS.has(lastWord);
+
+            //sentence ending
+            if (ENDINGS.includes(lastChar) && !isAbbreviation) {
+                result.push({
+                    speaker: currentSpeaker,
+                    sentence: currentSentence,
+                    start: startTime,
+                    end: endTime
+                });
+                currentSentence = "";
+                startTime = null;
+                endTime = null;
+                currentSpeaker = null;
+            }
+        }
+
+        // safe last sentence 
+        if (currentSentence) {
+            result.push({
+                speaker: currentSpeaker,
+                sentence: currentSentence,
+                start: startTime,
+                end: endTime
+            });
+        }
+
+
+        // Output as Text
+        const output = result.map(r => 
+            `Sprecher ${r.speaker} [${r.start.toFixed(2)} - ${r.end.toFixed(2)}]: ${r.sentence}`
+        );
+
+        // Output on cosole
+        //console.log("\n------------\nMerged Transcription Result:\n", output, "\n------------\n");
+
+        try {
+            const jsonPath = path.join(outputDir, "transcription_result.json");
+            fs.writeFileSync(jsonPath, JSON.stringify(result, null, 2), "utf-8");
+
+            const txtPath = path.join(outputDir, "transcription_result.txt");
+            fs.writeFileSync(txtPath, output.join("\n"), "utf-8");
+
+            console.log(`Summary successfully saved:\n- ${jsonPath}\n- ${txtPath}`);
+        } catch (err) {
+            console.error("Error saving Summary:", err);
+        }
+    }
+}
@@ -1,8 +1,123 @@
-module.exports = {
-    name:"assembly", // Unique name for our function that will later be used to get the function from the map via "mapFunctions.get("example").function()"
-    type:"transcription", // value used to differentiate each module to order them in the UI
-    displayname:"Assembly", // The displayname used within the UI
-    async function(parameter){
-        // TODO add code to actually process the audio file
+require('dotenv').config();
+
+const API_KEY = process.env.API_KEY;
+const BASE_URL = 'https://api.assemblyai.com/v2';
+
+//---------------------------------------------------Upload audio---------------------------------------------------
+
+async function uploadAudio(audioPath) {
+  const audioData = fs.readFileSync(audioPath);
+
+  const response = await axios.post(`${BASE_URL}/upload`, audioData, {
+    headers: {
+      authorization: API_KEY,
+      'content-type': 'application/octet-stream'
    }
-}
+  });
+
+  return response.data.upload_url;
+}
+
+////---------------------------------------------------Extract session id---------------------------------------------------
+
+function getSessionId(inputPath) {
+  try {
+    const parsed = new URL(inputPath);
+    const base = path.basename(parsed.pathname);
+    return base.replace(/\.[^.]+$/, '');
+  } catch {
+    return path.basename(inputPath, path.extname(inputPath));
+  }
+}
+
+//---------------------------------------------------Create transcript---------------------------------------------------
+
+async function createTranscript(audioUrl) {
+  const response = await axios.post(
+    `${BASE_URL}/transcript`,
+    {
+      audio_url: audioUrl,
+      speaker_labels: true,
+      language_detection: true
+    },
+    {
+      headers: {
+        authorization: API_KEY,
+        'content-type': 'application/json'
+      }
+    }
+  );
+
+  return response.data.id;
+}
+
+//---------------------------------------------------Poll transcript---------------------------------------------------
+
+async function pollTranscript(transcriptId) {
+  while (true) {
+    const response = await axios.get(`${BASE_URL}/transcript/${transcriptId}`, {
+      headers: { authorization: API_KEY }
+    });
+
+    const status = response.data.status;
+
+    if (status === 'completed') return response.data;
+    if (status === 'error') throw new Error(`Transcription failed: ${response.data.error}`);
+
+    await new Promise(res => setTimeout(res, 3000));
+  }
+}
+
+//---------------------------------------------------Save transcript---------------------------------------------------
+
+function saveTranscript(transcript, sessionId) {
+  const outputDir = path.join(__dirname, '../../../storage/transcripts');
+
+  if (!fs.existsSync(outputDir)) {
+    fs.mkdirSync(outputDir, { recursive: true });
+  }
+
+  const outputPath = path.join(outputDir, `${sessionId}.json`);
+  fs.writeFileSync(outputPath, JSON.stringify(transcript, null, 2));
+
+  console.log(`Transcript saved: ${outputPath}`);
+}
+
+//---------------------------------------------------Modul---------------------------------------------------
+
+module.exports = {
+  name: 'assembly',
+  type: 'transcription',
+  displayname: 'AssemblyAI',
+
+  async function(audioFileName) {
+    try {
+      // audioFileName ist nur "datei.mp3"
+      const audioPath = path.join(
+        __dirname,
+        '../../../storage/audio',
+        audioFileName
+      );
+
+      let audioUrl;
+
+      if (/^https?:\/\//i.test(audioFileName)) {
+        audioUrl = audioFileName;
+      } else {
+        if (!fs.existsSync(audioPath)) {
+          throw new Error(`Audio file not found: ${audioPath}`);
+        }
+        audioUrl = await uploadAudio(audioPath);
+      }
+
+      const transcriptId = await createTranscript(audioUrl);
+      const transcript = await pollTranscript(transcriptId);
+
+      const sessionId = getSessionId(audioFileName);
+      saveTranscript(transcript, sessionId);
+
+    } catch (error) {
+      console.error('Transcription error:', error.message);
+    }
+  }
+};
@@ -6,6 +6,13 @@ module.exports = {
        // We are now calling the example function from the example folder
        mapFunctions.get("example").function("Startup")

+       let transcript = await mapFunctions.get("assembly").function('../../storage/audio/IMG_2978.wav');
+
+       let summary = await mapFunctions.get("summarize-transcription").function({jsonPath:'/Users/santa/Proj25/video2document/storage/transcripts/IMG_2978.json'});
+
+
+        
+
        // mapFunctions.get("extraction-video-to-audio").function({inputVideoPath:"./a.mp4", outputType:"wav"})
        // mapFunctions.get("extraction-video-to-audio").function({inputVideoPath:"./b.mp4", outputType:"wav"})
        // mapFunctions.get("extraction-video-to-audio").function({inputVideoPath:"./b.mp4", outputType:"flac"})