Update app.py

app.py

@@ -35,6 +35,11 @@ def format_timestamp(t):
     mi = (t - int(t))*1000
     return f"{int(hh):02d}:{int(mm):02d}:{int(ss):02d},{int(mi):03d}"
 
+def save_srt(transcript):
+    with open("transcript.srt", "w") as f:
+        f.write(transcript)
+    return True
+
 langs = ["None"] + sorted(list(whisper.tokenizer.LANGUAGES.values()))
 model_size = list(whisper._MODELS.keys())
 
@@ -49,17 +54,12 @@ st.markdown("Larger models are more accurate, but slower. For 1min video, it'll
 if st.button("Transcribe"):
     transcript = get_transcript(url, model_size, lang, format)
     st.text_area("Transcription of the video", transcript)
-
-if transcript:
-    with st.spinner("Saving transcript..."):
-        with open("transcript.srt", "w", encoding="utf-8") as f:
-            f.write(transcript)
-    st.success("Transcript saved as transcript.srt")
     if format == ".srt":
-        st.download_button(
-            "Download .srt file",
-            data=transcript,
-            file_name="transcript.srt",
-            mime="text/plain"
-        )
+        if save_srt(transcript):
+            st.markdown(get_srt_download_link("transcript.srt"), unsafe_allow_html=True)
 
+def get_srt_download_link(file_path):
+    with open(file_path, "rb") as f:
+        srt = f.read().decode("utf-8")
+    href = f"<a href='data:text/plain;charset=utf-8,{srt}' download='transcript.srt'>Download .srt file</a>"
+    return href