Update app.py

app.py

@@ -50,8 +50,16 @@ 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":
-        srt_file = open("transcript.srt", "w", encoding="utf-8")
-        srt_file.write(transcript)
-        srt_file.close()
-        st.download_button("Download SRT", "transcription.srt")
+        st.download_button(
+            "Download .srt file",
+            data=transcript,
+            file_name="transcript.srt",
+            mime="text/plain"
+        )
+