Added subtitle download and transcript download

Home.py

@@ -51,13 +51,29 @@ def app():
         status_text.text('Subtitling audio...')
         subtitles = mw.get_prediction_with_timelines(transcript, file.name + '.srt')
         progress_bar.progress(90)
-        status_text.text('Saving transcript...')
-        ms.save_subtitles(subtitles, file.name + '.srt')
+        
         status_text.text('Saving subtitles...')
+        st.session_state['subtitles'] = subtitles
+        ms.save_subtitles(subtitles, file.name + '.srt')
+
+        status_text.text('Saving transcript...')
+        st.session_state['transcript'] = transcript
         mt.save_transcript(transcript, file.name + '.txt')
         status_text.text('Done!')
         progress_bar.progress(100)
 
+        st.download_button(
+            label="Download Transcript",
+            data=st.session_state['transcript'],
+            file_name=file.name + '.txt',
+            mime='text/plain'
+        )
+        st.download_button(
+            label="Download Subtitles",
+            data=st.session_state['subtitles'],
+            file_name=file.name + '.srt',
+            mime='text/plain'
+        )
 
 if __name__ == '__main__':
     app()