Add support for subtitles

app.py

@@ -1,18 +1,28 @@
 import gradio as gr
 import whisper
+from whisper.utils import write_srt
 
 model = whisper.load_model("base")
 
 def transcribe(file):
     result = model.transcribe(file)
-    text = result['text']
-    f = open("transcript_base.txt", "w")
-    f.write(text)
-    f.close()
-    return "transcript_base.txt"
+
+    #transcript
+    with open(str(file) + "-transcript.txt", "w") as f:
+        f.write(result['text'])
+
+    #subtitles
+    with open(str(file) + "-subs.srt", "w") as srt:
+        write_srt(result["segments"], file=srt)
+
+    download = []
+    download.append(str(file) + "-subs.srt");
+    download.append(str(file) + "-transcript.txt");
+
+    return download
 
 iface = gr.Interface(
-    title = 'Whisper transcription from file', 
+    title = 'Whisper transcription and subtitles from file', 
     fn=transcribe, 
     inputs=[
         gr.inputs.Audio(source="upload", type="filepath", label="Upload Audio")