full transcription output

Signed-off-by: Raphael <[email protected]>

app.py

@@ -66,8 +66,8 @@ def demo(url: str, translate: bool):
     LOG.info("Base directory %s", basedir)
     video_path, video = download(url, os.path.join(basedir, 'video.mp4'))
     audio_clips(video, basedir)
-    srt_file, summary = process_video(basedir, video.duration, translate)
-    return summary, srt_file, [video_path, srt_file]
+    srt_file, full_transcription, summary = process_video(basedir, video.duration, translate)
+    return summary, srt_file, [video_path, srt_file], full_transcription
 
 
 def download(url, dst):
@@ -118,7 +118,7 @@ def process_video(basedir: str, duration, translate: bool):
     subs = translation(transcriptions, translate)
     srt_file = build_srt_clips(subs, basedir)
     summary = summarize(transcriptions, translate)
-    return srt_file, summary
+    return srt_file, ' '.join(subs).strip(), summary
 
 
 def transcription(audio_dir: str, duration):
@@ -247,6 +247,7 @@ iface = gr.Interface(
         gr.Text(label="Video summary"),
         gr.File(label="SRT file"),
         gr.Video(label="Video with subtitles"),
+        gr.Text(label="Full transcription")
     ])
 
 iface.launch()