Sure! Pl

vosk/test_files/test_vosk_transcription.py

@@ -0,0 +1,39 @@
+import requests
+import difflib
+import sys
+
+# Usage: python test_vosk_transcription.py <audio_file> <reference_text>
+
+API_URL = 'http://localhost:5000/transcribe'
+
+
+def similarity(a, b):
+    return difflib.SequenceMatcher(None, a, b).ratio()
+
+
+def main():
+    if len(sys.argv) != 3:
+        print("Usage: python test_vosk_transcription.py <audio_file> <reference_text>")
+        sys.exit(1)
+    audio_path = sys.argv[1]
+    reference_text = sys.argv[2]
+    with open(audio_path, 'rb') as f:
+        files = {'audio': f}
+        response = requests.post(API_URL, files=files)
+    if response.status_code != 200:
+        print(f"API error: {response.text}")
+        sys.exit(1)
+    transcription = response.json().get('transcription', '')
+    sim = similarity(transcription, reference_text)
+    print(f"Transcription: {transcription}")
+    print(f"Reference: {reference_text}")
+    print(f"Similarity: {sim:.2f}")
+    if sim > 0.2:
+        print("Test PASSED: Similarity above threshold.")
+        sys.exit(0)
+    else:
+        print("Test FAILED: Similarity below threshold.")
+        sys.exit(1)
+
+if __name__ == '__main__':
+    main()