Update app.py

app.py

@@ -55,10 +55,16 @@ if st.button("Transcribe"):
     transcript = get_transcript(url, model_size, lang, format)
     st.text_area("Transcription of the video", transcript)
     if format == ".srt":
-        if st.button("Download transcript"):
-            if save_srt(transcript):
-                st.success("Transcript downloaded successfully!")
-            else:
-                st.error("Failed to download transcript.")
+        save_srt(transcript)
+        if st.button("Download Transcript"):
+            file_ = open("transcript.srt", "rb")
+            contents = file_.read()
+            file_.close()
+            st.download_button(
+                label="Download Transcript",
+                data=contents,
+                file_name="done.srt",
+                mime="text/srt"
+            )