From 96e8c22c446992f15750b39b30f8e798728450e6 Mon Sep 17 00:00:00 2001 From: Johnny Huynh <27847622+johnnyhuy@users.noreply.github.com> Date: Mon, 21 Oct 2024 08:55:52 +1100 Subject: [PATCH] feat: Refactor transcription options and key phrase extraction - Refactor transcription options in `transcribe_me/audio/transcription.py` - Update key phrase extraction section in `transcribe_me/audio/transcription.py` --- transcribe_me/audio/transcription.py | 8 -------- 1 file changed, 8 deletions(-) diff --git a/transcribe_me/audio/transcription.py b/transcribe_me/audio/transcription.py index 355c2e1..f78cefc 100644 --- a/transcribe_me/audio/transcription.py +++ b/transcribe_me/audio/transcription.py @@ -86,7 +86,6 @@ def transcribe_with_assemblyai( speaker_labels=True, summarization=True, sentiment_analysis=True, - auto_highlights=True, iab_categories=True, ) transcriber = aai.Transcriber() @@ -116,13 +115,6 @@ def transcribe_with_assemblyai( file.write(f"Confidence: {result.confidence}\n") file.write(f"Timestamp: {result.start} - {result.end}\n\n") - # Key Phrases - with open(f"{base_name}_key_phrases.txt", "w", encoding="utf-8") as file: - for phrase in transcript.auto_highlights_result.results: - file.write(f"Highlight: {phrase.text}\n") - file.write(f"Count: {phrase.count}\n") - file.write(f"Rank: {phrase.rank}\n") - # Topic Detection if transcript.iab_categories: with open(f"{base_name}_topics.txt", "w", encoding="utf-8") as file: