Comparar idiomas por codigo para que auto funcione con el subtitulado

app.py

@@ -408,10 +408,25 @@ def subtitle_video(original_video_path, source_languaje, target_languaje):
     print('*'*NUMBER)
     print("Subtitle video")
 
-    if source_languaje != target_languaje:
+    # Con "auto" el idioma no se conoce hasta haber transcrito, así que se lee
+    # el que quedó guardado en la transcripción
+    language_dict = get_language_dict()
+    source_code = None
+    if source_languaje == AUTO_LANGUAGE:
+        transcription_file = f"transcriptions/transcription_{source_languaje}.json"
+        if os.path.exists(transcription_file):
+            with open(transcription_file, encoding="utf-8") as f:
+                source_code = json.load(f).get("language")
+    elif source_languaje in language_dict:
+        source_code = language_dict[source_languaje]["transcriber"]
+
+    target_code = language_dict.get(target_languaje, {}).get("transcriber")
+
+    if source_code != target_code:
         return [
             gr.update(value=None, visible=False),
-            gr.update(value="Translation pipeline not implemented yet")
+            gr.update(value=f"Translation pipeline not implemented yet "
+                            f"({source_code} -> {target_code})")
         ]
 
     merged_path = "merged_transcription_diarization/merged.json"

transcribe.py

@@ -124,6 +124,10 @@ def transcribe(audio_file, language, device, chunk_length_s=30, stride_length_s=
 
     print(f"Time to transcribe: {time() - t0:.2f} seconds")
 
+    # Se guarda el idioma realmente usado: con autodetección es el único sitio
+    # donde se sabe, y hace falta después para decidir si hay que traducir
+    result["language"] = language
+
     # Both engines already return the normalised shape, so the second value is
     # kept for the callers that expect the old two-value contract
     return result, result