qqwjq1981 commited on
Commit
7435cc9
·
verified ·
1 Parent(s): 306be63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -47,6 +47,14 @@ import webrtcvad
47
  from pydub import AudioSegment
48
  from pydub.silence import split_on_silence
49
  import soundfile as sf
 
 
 
 
 
 
 
 
50
 
51
  logger = logging.getLogger(__name__)
52
 
@@ -380,7 +388,7 @@ def transcribe_segments_with_scribe(full_audio_path, segments):
380
  logger.info(f"Segment {i+1} discarded: cleaned text is empty.")
381
 
382
  if "language_code" in scribe_result and detected_language == "unknown":
383
- detected_language = scribe_result["language_code"]
384
 
385
  except Exception as e:
386
  logger.error(f"Error processing segment {i+1}: {e}")
@@ -1509,4 +1517,4 @@ def build_interface():
1509
  tts_model = None
1510
  # Launch the Gradio interface
1511
  demo = build_interface()
1512
- demo.launch()
 
47
  from pydub import AudioSegment
48
  from pydub.silence import split_on_silence
49
  import soundfile as sf
50
+ import langcodes
51
+
52
+ # ISO 639-3 → ISO 639-1
53
+ def iso_639_3_to_1(code3):
54
+ try:
55
+ return langcodes.Language.get(code3).language
56
+ except:
57
+ return 'en'
58
 
59
  logger = logging.getLogger(__name__)
60
 
 
388
  logger.info(f"Segment {i+1} discarded: cleaned text is empty.")
389
 
390
  if "language_code" in scribe_result and detected_language == "unknown":
391
+ detected_language = iso_639_3_to_1(scribe_result["language_code"])
392
 
393
  except Exception as e:
394
  logger.error(f"Error processing segment {i+1}: {e}")
 
1517
  tts_model = None
1518
  # Launch the Gradio interface
1519
  demo = build_interface()
1520
+ demo.launch()