lyrics-transcriber 0.12.8__py3-none-any.whl → 0.12.9__py3-none-any.whl

lyrics_transcriber/llm_prompts/promptfooconfig.yaml

@@ -1,39 +1,61 @@
 # This configuration runs each prompt through a series of example inputs and checks if they meet requirements.
 # Learn more: https://promptfoo.dev/docs/configuration/guide
 
+description: Song lyric corrector for a karaoke video studio, responsible for reading lyrics inputs, correcting them and generating JSON-based responses containing the corrected lyrics according to predefined criteria.
+providers:
+  - id: openai:gpt-3.5-turbo-1106
+    config:
+      temperature: 0
+  # - id: openai:gpt-4-1106-preview
+  #   config:
+  #     temperature: 0
 prompts:
-  - file://llm_prompt_lyrics_correction_*.txt
-providers: [openai:gpt-3.5-turbo-0613, openai:gpt-4-1106-preview]
+  - file://llm_prompt_lyrics_correction_andrew_handwritten_20231118.txt
+
+defaultTest:
+  assert:
+    - type: is-json
+      value:
+        required: [id, text, words]
+        type: object
+        properties:
+          id:
+            type: number
+          text:
+            type: string
+          words:
+            type: array
+            items:
+              type: object
+              properties:
+                text:
+                  type: string
+                start:
+                  type: number
+                end:
+                  type: number
+                confidence:
+                  type: number
+
 tests:
-  - description: First test case - automatic review
+  - description: ABBA - Under Attack (segment 0)
     vars:
-      var1: first variable's value
-      var2: another value
-      var3: some other value
-    # For more information on assertions, see https://promptfoo.dev/docs/configuration/expected-outputs
+      reference_lyrics: file://test_data/ABBA-UnderAttack-Genius.txt
+      previous_two_corrected_lines: 
+      upcoming_two_uncorrected_lines: 
+      segment_input: |
+        {"id": 0, "start": 17.46, "end": 21.3, "confidence": 0.792, "text": " Don't know how to take it, don't know where to go", "words": [{"text": "Don't", "start": 17.46, "end": 18.2, "confidence": 0.278}, {"text": "know", "start": 18.2, "end": 18.42, "confidence": 0.965}, {"text": "how", "start": 18.42, "end": 18.66, "confidence": 0.865}, {"text": "to", "start": 18.66, "end": 18.88, "confidence": 0.994}, {"text": "take", "start": 18.88, "end": 19.2, "confidence": 0.992}, {"text": "it,", "start": 19.2, "end": 19.44, "confidence": 0.974}, {"text": "don't", "start": 19.56, "end": 19.8, "confidence": 0.917}, {"text": "know", "start": 19.8, "end": 20.02, "confidence": 0.989}, {"text": "where", "start": 20.02, "end": 20.46, "confidence": 0.963}, {"text": "to", "start": 20.46, "end": 20.76, "confidence": 0.983}, {"text": "go", "start": 20.76, "end": 21.3, "confidence": 0.982}]}
     assert:
-      - type: equals
-        value: expected LLM output goes here
       - type: contains
-        value: some text
-      - type: javascript
-        value: 1 / (output.length + 1)  # prefer shorter outputs
+        value: "Don't know how to take it, don't know where to go"
 
-  - description: Second test case - manual review
-    # Test cases don't need assertions if you prefer to manually review the output
+  - description: ABBA - Under Attack (segment 1)
     vars:
-      var1: new value
-      var2: another value
-      var3: third value
-
-  - description: Third test case - other types of automatic review
-    vars:
-      var1: yet another value
-      var2: and another
-      var3: dear llm, please output your response in json format
+      reference_lyrics: file://test_data/ABBA-UnderAttack-Genius.txt
+      previous_two_corrected_lines: 
+      upcoming_two_uncorrected_lines: 
+      segment_input: |
+        {"id": 1, "start": 22.04, "end": 27.84, "confidence": 0.763, "text": " My resistance running low And every day the hole is getting tighter", "words": [{"text": "My", "start": 22.04, "end": 22.32, "confidence": 0.535}, {"text": "resistance", "start": 22.32, "end": 22.94, "confidence": 0.936}, {"text": "running", "start": 22.94, "end": 23.66, "confidence": 0.89}, {"text": "low", "start": 23.66, "end": 24.36, "confidence": 0.999}, {"text": "And", "start": 24.36, "end": 25.14, "confidence": 0.485}, {"text": "every", "start": 25.14, "end": 25.56, "confidence": 0.568}, {"text": "day", "start": 25.56, "end": 25.88, "confidence": 0.997}, {"text": "the", "start": 25.88, "end": 26.1, "confidence": 0.959}, {"text": "hole", "start": 26.1, "end": 26.48, "confidence": 0.361}, {"text": "is", "start": 26.48, "end": 26.68, "confidence": 0.947}, {"text": "getting", "start": 26.68, "end": 27.08, "confidence": 0.996}, {"text": "tighter", "start": 27.08, "end": 27.84, "confidence": 0.975}]}
     assert:
-      - type: contains-json
-      - type: similar
-        value: ensures that output is semantically similar to this text
-      - type: model-graded-closedqa
-        value: ensure that output contains a reference to X
+      - type: contains
+        value: "My resistance running low And every day the hold is getting tighter"

lyrics_transcriber/llm_prompts/test_data/ABBA-UnderAttack-Genius.txt

@@ -0,0 +1,48 @@
+Don't know how to take it, don't know where to go
+My resistance running low
+And every day the hold is getting tighter and it troubles me so
+(You know that I'm nobody's fool)
+I'm nobody's fool and yet it's clear to me
+I don't have a strategy
+It's just like taking candy from a baby and I think I must be
+
+Under attack, I'm being taken
+About to crack, defences breaking
+Won't somebody please have a heart
+Come and rescue me now 'cause I'm falling apart
+Under attack, I'm taking cover
+He's on my track, my chasing lover
+Thinking nothing can stop him now
+Should I want to, I'm not sure I would know how
+
+This is getting crazy, I should tell him so
+Really let my anger show
+Persuade him that the answer to his questions is a definite no
+(I'm kind of flattered I suppose)
+Guess I'm kind of flattered but I'm scared as well
+Something like a magic spell
+I hardly dare to think of what would happen, where I'd be if I fell
+
+Under attack, I'm being taken
+About to crack, defences breaking
+Won't somebody please have a heart
+Come and rescue me now 'cause I'm falling apart
+Under attack, I'm taking cover
+He's on my track, my chasing lover
+Thinking nothing's gonna stop him now
+Should I want to, I'm not sure I won't know how
+
+Under attack, I'm being taken
+About to crack, defences breaking
+Won't somebody see and save a heart
+Come and rescue me now 'cause I'm falling apart
+Under attack, I'm taking cover
+He's on my track, my chasing lover
+Thinking nothing can stop him now
+Should I want to, I'm not sure I would know how
+
+Under attack, I'm being taken
+About to crack, defences breaking
+Won't somebody please have a heart
+Come and rescue me now 'cause I'm falling apart
+Under attack, I'm taking cover

lyrics_transcriber/transcriber.py

@@ -303,14 +303,13 @@ class LyricsTranscriber:
                 previous_two_corrected_lines = ""
                 upcoming_two_uncorrected_lines = ""
 
-                if segment["id"] > 2:
-                    for previous_segment in corrected_lyrics_dict["segments"]:
-                        if previous_segment["id"] in (segment["id"] - 2, segment["id"] - 1):
-                            previous_two_corrected_lines += previous_segment["text"].strip() + "\n"
-
-                    for next_segment in self.outputs["transcription_data_dict"]["segments"]:
-                        if next_segment["id"] in (segment["id"] + 1, segment["id"] + 2):
-                            upcoming_two_uncorrected_lines += next_segment["text"].strip() + "\n"
+                for previous_segment in corrected_lyrics_dict["segments"]:
+                    if previous_segment["id"] in (segment["id"] - 2, segment["id"] - 1):
+                        previous_two_corrected_lines += previous_segment["text"].strip() + "\n"
+
+                for next_segment in self.outputs["transcription_data_dict"]["segments"]:
+                    if next_segment["id"] in (segment["id"] + 1, segment["id"] + 2):
+                        upcoming_two_uncorrected_lines += next_segment["text"].strip() + "\n"
 
                 llm_transcript_segment += f"--- Segment {segment['id']} / {total_segments} ---\n"
                 llm_transcript_segment += f"Previous two corrected lines:\n\n{previous_two_corrected_lines}\nUpcoming two uncorrected lines:\n\n{upcoming_two_uncorrected_lines}\nData input:\n\n{segment_input}\n"

{lyrics_transcriber-0.12.8.dist-info → lyrics_transcriber-0.12.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lyrics-transcriber
-Version: 0.12.8
+Version: 0.12.9
 Summary: Automatically create synchronised lyrics files in ASS and MidiCo LRC formats with word-level timestamps, using Whisper and lyrics from Genius and Spotify
 Home-page: https://github.com/karaokenerds/python-lyrics-transcriber
 License: MIT

{lyrics_transcriber-0.12.8.dist-info → lyrics_transcriber-0.12.9.dist-info}/RECORD

@@ -3,14 +3,15 @@ lyrics_transcriber/llm_prompts/README.md,sha256=DPAGRDVGt9ZNcQAAoQGFhwesLY3D6hD8
 lyrics_transcriber/llm_prompts/llm_prompt_lyrics_correction_andrew_handwritten_20231118.txt,sha256=a3XjAYfyhWt1uCKKqm_n2Pc0STdmBdiHHtJ7ODP99Nk,4046
 lyrics_transcriber/llm_prompts/llm_prompt_lyrics_correction_gpt_optimised_20231119.txt,sha256=r6HN3DD_3gwh3B_JPd2R0I4lDXuB5iy7B90J9agOxbQ,2369
 lyrics_transcriber/llm_prompts/llm_prompt_lyrics_matching_andrew_handwritten_20231118.txt,sha256=hvk2Vs3M3Q4zGQsiQnXvnpd8wXWfwsudYeqN5qFyNWs,1754
-lyrics_transcriber/llm_prompts/promptfooconfig.yaml,sha256=Qm2GsQh2AoKJfskC3V92E5jtDg8zT9yKVgChUqeE9yw,1416
-lyrics_transcriber/transcriber.py,sha256=c6F6-BWnzq3KDtsMHshYYstWAkQDyYseoxdFgsYAo6U,38748
+lyrics_transcriber/llm_prompts/promptfooconfig.yaml,sha256=O4YxlLV7XSUiSw_1Q9G7ELC2VAbrYUV_N5QxrPbd1jE,3735
+lyrics_transcriber/llm_prompts/test_data/ABBA-UnderAttack-Genius.txt,sha256=8d-RvZtyINKUlpQLwMi-VD--Y59J-epPt7SZSqjFbPI,1690
+lyrics_transcriber/transcriber.py,sha256=ykodPGx-bDmT3BBGY5MywfMfmUGUrUsQw7C-k-PzYS0,38686
 lyrics_transcriber/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lyrics_transcriber/utils/ass.py,sha256=b8lnjgXGD1OD1ld_b1xxUmSOf4nSEfz9BpgSkh16R4g,90291
 lyrics_transcriber/utils/cli.py,sha256=-xzokp8RZMh8DDCr-PZbh4yHqlnJP70Ts62-VnjZq48,6398
 lyrics_transcriber/utils/subtitles.py,sha256=GPLbo9oX3WJIY4wRO40k0RZF-UqvKjr2LFnXVOS0RHI,10762
-lyrics_transcriber-0.12.8.dist-info/LICENSE,sha256=BiPihPDxhxIPEx6yAxVfAljD5Bhm_XG2teCbPEj_m0Y,1069
-lyrics_transcriber-0.12.8.dist-info/METADATA,sha256=NHYdhOYcWQjEHBVQgIsMUIro0wFCq2XG0_rxKDOl_RM,5051
-lyrics_transcriber-0.12.8.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-lyrics_transcriber-0.12.8.dist-info/entry_points.txt,sha256=lh6L-iR5CGELaNcouDK94X78eS5Ua_tK9lI4UEkza-k,72
-lyrics_transcriber-0.12.8.dist-info/RECORD,,
+lyrics_transcriber-0.12.9.dist-info/LICENSE,sha256=BiPihPDxhxIPEx6yAxVfAljD5Bhm_XG2teCbPEj_m0Y,1069
+lyrics_transcriber-0.12.9.dist-info/METADATA,sha256=et_XjNqSkHONMdJ8KlO9olAK3KkAQ4Wz7HIR8rw0YP0,5051
+lyrics_transcriber-0.12.9.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+lyrics_transcriber-0.12.9.dist-info/entry_points.txt,sha256=lh6L-iR5CGELaNcouDK94X78eS5Ua_tK9lI4UEkza-k,72
+lyrics_transcriber-0.12.9.dist-info/RECORD,,