diff --git a/.github/workflows/ci-cd.yaml b/.github/workflows/ci-cd.yaml
new file mode 100644
index 0000000..591ea9d
--- /dev/null
+++ b/.github/workflows/ci-cd.yaml
@@ -0,0 +1,33 @@
+# Copyright Jiaqi Liu
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+---
+name: CI/CD
+
+on:
+ pull_request:
+ push:
+ branches: [master]
+
+jobs:
+ sync-to-huggingface-space:
+ runs-on: ubuntu-latest
+ steps:
+ - uses: actions/checkout@v3
+ with:
+ fetch-depth: 0
+ lfs: true
+ - name: Push to hub
+ run: git push https://QubitPi:$HF_TOKEN@huggingface.co/spaces/QubitPi/tiger master:main -f
+ env:
+ HF_TOKEN: ${{ secrets.HF_TOKEN }}
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..d868f45
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,4 @@
+.venv
+.idea/
+.DS_Store
+__pycache__
diff --git "a/01_\360\237\216\245_Input_YouTube_Link.py" "b/01_\360\237\216\245_Input_YouTube_Link.py"
new file mode 100644
index 0000000..ab467ae
--- /dev/null
+++ "b/01_\360\237\216\245_Input_YouTube_Link.py"
@@ -0,0 +1,258 @@
+import whisper
+from pytubefix import YouTube
+from pytubefix.cli import on_progress
+import requests
+import time
+import streamlit as st
+from streamlit_lottie import st_lottie
+import numpy as np
+import os
+from typing import Iterator
+from io import StringIO
+from utils import write_vtt, write_srt
+import ffmpeg
+from languages import LANGUAGES
+import torch
+from zipfile import ZipFile
+from io import BytesIO
+import base64
+import pathlib
+import re
+
+st.set_page_config(page_title="Auto Subtitled Video Generator", page_icon=":movie_camera:", layout="wide")
+
+torch.cuda.is_available()
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+# Model options: tiny, base, small, medium, large
+loaded_model = whisper.load_model("small", device=DEVICE)
+current_size = "None"
+
+
+
+# Define a function that we can use to load lottie files from a link.
+def load_lottieurl(url: str):
+ r = requests.get(url)
+ if r.status_code != 200:
+ return None
+ return r.json()
+
+APP_DIR = pathlib.Path(__file__).parent.absolute()
+
+LOCAL_DIR = APP_DIR / "local_youtube"
+LOCAL_DIR.mkdir(exist_ok=True)
+save_dir = LOCAL_DIR / "output"
+save_dir.mkdir(exist_ok=True)
+
+
+
+col1, col2 = st.columns([1, 3])
+with col1:
+ lottie = load_lottieurl("https://assets8.lottiefiles.com/packages/lf20_jh9gfdye.json")
+ st_lottie(lottie)
+
+with col2:
+ st.write("""
+ ## Auto Subtitled Video Generator
+ ##### Input a YouTube video link and get a video with subtitles.
+ ###### ➠ If you want to transcribe the video in its original language, select the task as "Transcribe"
+ ###### ➠ If you want to translate the subtitles to English, select the task as "Translate"
+ ###### I recommend starting with the base model and then experimenting with the larger models, the small and medium models often work well. """)
+
+
+def download_video(link):
+ yt = YouTube(link, on_progress_callback=on_progress)
+ ys = yt.streams.get_highest_resolution()
+ video = ys.download(filename=f"{save_dir}/youtube_video.mp4")
+ return video
+
+
+def convert(seconds):
+ return time.strftime("%H:%M:%S", time.gmtime(seconds))
+
+
+def change_model(current_size, size):
+ if current_size != size:
+ loaded_model = whisper.load_model(size)
+ return loaded_model
+ else:
+ raise Exception("Model size is the same as the current size.")
+
+
+def inference(link, loaded_model, task):
+ yt = YouTube(link, on_progress_callback=on_progress)
+ ys = yt.streams.get_audio_only()
+ path = ys.download(filename=f"{save_dir}/audio.mp3", mp3=True)
+ if task == "Transcribe":
+ options = dict(task="transcribe", best_of=5)
+ results = loaded_model.transcribe(path, **options)
+ vtt = getSubs(results["segments"], "vtt", 80)
+ srt = getSubs(results["segments"], "srt", 80)
+ lang = results["language"]
+ return results["text"], vtt, srt, lang
+ elif task == "Translate":
+ options = dict(task="translate", best_of=5)
+ results = loaded_model.transcribe(path, **options)
+ vtt = getSubs(results["segments"], "vtt", 80)
+ srt = getSubs(results["segments"], "srt", 80)
+ lang = results["language"]
+ return results["text"], vtt, srt, lang
+ else:
+ raise ValueError("Task not supported")
+
+
+def getSubs(segments: Iterator[dict], format: str, maxLineWidth: int) -> str:
+ segmentStream = StringIO()
+
+ if format == 'vtt':
+ write_vtt(segments, file=segmentStream, maxLineWidth=maxLineWidth)
+ elif format == 'srt':
+ write_srt(segments, file=segmentStream, maxLineWidth=maxLineWidth)
+ else:
+ raise Exception("Unknown format " + format)
+
+ segmentStream.seek(0)
+ return segmentStream.read()
+
+
+def get_language_code(language):
+ if language in LANGUAGES.keys():
+ detected_language = LANGUAGES[language]
+ return detected_language
+ else:
+ raise ValueError("Language not supported")
+
+
+def generate_subtitled_video(video, audio, transcript):
+ video_file = ffmpeg.input(video)
+ audio_file = ffmpeg.input(audio)
+ ffmpeg.concat(video_file.filter("subtitles", transcript), audio_file, v=1, a=1).output("youtube_sub.mp4").run(quiet=True, overwrite_output=True)
+ video_with_subs = open("youtube_sub.mp4", "rb")
+ return video_with_subs
+
+
+def main():
+ size = st.selectbox("Select Model Size (The larger the model, the more accurate the transcription will be, but it will take longer)", ["tiny", "base", "small", "medium", "large-v3"], index=1)
+ loaded_model = change_model(current_size, size)
+ st.write(f"Model is {'multilingual' if loaded_model.is_multilingual else 'English-only'} "
+ f"and has {sum(np.prod(p.shape) for p in loaded_model.parameters()):,} parameters.")
+ link = st.text_input("YouTube Link (The longer the video, the longer the processing time)", placeholder="Input YouTube link and press enter")
+ task = st.selectbox("Select Task", ["Transcribe", "Translate"], index=0)
+ if task == "Transcribe":
+ if st.button("Transcribe"):
+ with st.spinner("Transcribing the video..."):
+ results = inference(link, loaded_model, task)
+ video = download_video(link)
+ lang = results[3]
+ detected_language = get_language_code(lang)
+
+ col3, col4 = st.columns(2)
+ with col3:
+ st.video(video)
+
+ # Split result["text"] on !,? and . , but save the punctuation
+ sentences = re.split("([!?.])", results[0])
+ # Join the punctuation back to the sentences
+ sentences = ["".join(i) for i in zip(sentences[0::2], sentences[1::2])]
+ text = "\n\n".join(sentences)
+ with open("transcript.txt", "w+", encoding='utf8') as f:
+ f.writelines(text)
+ f.close()
+ with open(os.path.join(os.getcwd(), "transcript.txt"), "rb") as f:
+ datatxt = f.read()
+
+ with open("transcript.vtt", "w+",encoding='utf8') as f:
+ f.writelines(results[1])
+ f.close()
+ with open(os.path.join(os.getcwd(), "transcript.vtt"), "rb") as f:
+ datavtt = f.read()
+
+ with open("transcript.srt", "w+",encoding='utf8') as f:
+ f.writelines(results[2])
+ f.close()
+ with open(os.path.join(os.getcwd(), "transcript.srt"), "rb") as f:
+ datasrt = f.read()
+
+ with col4:
+ with st.spinner("Generating Subtitled Video"):
+ video_with_subs = generate_subtitled_video(video, f"{save_dir}/audio.mp3", "transcript.srt")
+ st.video(video_with_subs)
+ st.balloons()
+
+ zipObj = ZipFile("YouTube_transcripts_and_video.zip", "w")
+ zipObj.write("transcript.txt")
+ zipObj.write("transcript.vtt")
+ zipObj.write("transcript.srt")
+ zipObj.write("youtube_sub.mp4")
+ zipObj.close()
+ ZipfileDotZip = "YouTube_transcripts_and_video.zip"
+ with open(ZipfileDotZip, "rb") as f:
+ datazip = f.read()
+ b64 = base64.b64encode(datazip).decode()
+ href = f"\
+ Download Transcripts and Video\
+ "
+ st.markdown(href, unsafe_allow_html=True)
+
+ elif task == "Translate":
+ if st.button("Translate to English"):
+ with st.spinner("Translating to English..."):
+ results = inference(link, loaded_model, task)
+ video = download_video(link)
+ lang = results[3]
+ detected_language = get_language_code(lang)
+
+ col3, col4 = st.columns(2)
+ with col3:
+ st.video(video)
+
+ # Split result["text"] on !,? and . , but save the punctuation
+ sentences = re.split("([!?.])", results[0])
+ # Join the punctuation back to the sentences
+ sentences = ["".join(i) for i in zip(sentences[0::2], sentences[1::2])]
+ text = "\n\n".join(sentences)
+ with open("transcript.txt", "w+", encoding='utf8') as f:
+ f.writelines(text)
+ f.close()
+ with open(os.path.join(os.getcwd(), "transcript.txt"), "rb") as f:
+ datatxt = f.read()
+
+ with open("transcript.vtt", "w+",encoding='utf8') as f:
+ f.writelines(results[1])
+ f.close()
+ with open(os.path.join(os.getcwd(), "transcript.vtt"), "rb") as f:
+ datavtt = f.read()
+
+ with open("transcript.srt", "w+",encoding='utf8') as f:
+ f.writelines(results[2])
+ f.close()
+ with open(os.path.join(os.getcwd(), "transcript.srt"), "rb") as f:
+ datasrt = f.read()
+
+ with col4:
+ with st.spinner("Generating Subtitled Video"):
+ video_with_subs = generate_subtitled_video(video, f"{save_dir}/audio.mp3", "transcript.srt")
+ st.video(video_with_subs)
+ st.balloons()
+
+ zipObj = ZipFile("YouTube_transcripts_and_video.zip", "w")
+ zipObj.write("transcript.txt")
+ zipObj.write("transcript.vtt")
+ zipObj.write("transcript.srt")
+ zipObj.write("youtube_sub.mp4")
+ zipObj.close()
+ ZipfileDotZip = "YouTube_transcripts_and_video.zip"
+ with open(ZipfileDotZip, "rb") as f:
+ datazip = f.read()
+ b64 = base64.b64encode(datazip).decode()
+ href = f"\
+ Download Transcripts and Video\
+ "
+ st.markdown(href, unsafe_allow_html=True)
+
+ else:
+ st.info("Please select a task.")
+
+
+if __name__ == "__main__":
+ main()
+
\ No newline at end of file
diff --git a/LICENSE b/LICENSE
new file mode 100644
index 0000000..7d7429c
--- /dev/null
+++ b/LICENSE
@@ -0,0 +1,201 @@
+ Apache License
+ Version 2.0, January 2004
+ http://www.apache.org/licenses/
+
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+ 1. Definitions.
+
+ "License" shall mean the terms and conditions for use, reproduction,
+ and distribution as defined by Sections 1 through 9 of this document.
+
+ "Licensor" shall mean the copyright owner or entity authorized by
+ the copyright owner that is granting the License.
+
+ "Legal Entity" shall mean the union of the acting entity and all
+ other entities that control, are controlled by, or are under common
+ control with that entity. For the purposes of this definition,
+ "control" means (i) the power, direct or indirect, to cause the
+ direction or management of such entity, whether by contract or
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
+ outstanding shares, or (iii) beneficial ownership of such entity.
+
+ "You" (or "Your") shall mean an individual or Legal Entity
+ exercising permissions granted by this License.
+
+ "Source" form shall mean the preferred form for making modifications,
+ including but not limited to software source code, documentation
+ source, and configuration files.
+
+ "Object" form shall mean any form resulting from mechanical
+ transformation or translation of a Source form, including but
+ not limited to compiled object code, generated documentation,
+ and conversions to other media types.
+
+ "Work" shall mean the work of authorship, whether in Source or
+ Object form, made available under the License, as indicated by a
+ copyright notice that is included in or attached to the work
+ (an example is provided in the Appendix below).
+
+ "Derivative Works" shall mean any work, whether in Source or Object
+ form, that is based on (or derived from) the Work and for which the
+ editorial revisions, annotations, elaborations, or other modifications
+ represent, as a whole, an original work of authorship. For the purposes
+ of this License, Derivative Works shall not include works that remain
+ separable from, or merely link (or bind by name) to the interfaces of,
+ the Work and Derivative Works thereof.
+
+ "Contribution" shall mean any work of authorship, including
+ the original version of the Work and any modifications or additions
+ to that Work or Derivative Works thereof, that is intentionally
+ submitted to Licensor for inclusion in the Work by the copyright owner
+ or by an individual or Legal Entity authorized to submit on behalf of
+ the copyright owner. For the purposes of this definition, "submitted"
+ means any form of electronic, verbal, or written communication sent
+ to the Licensor or its representatives, including but not limited to
+ communication on electronic mailing lists, source code control systems,
+ and issue tracking systems that are managed by, or on behalf of, the
+ Licensor for the purpose of discussing and improving the Work, but
+ excluding communication that is conspicuously marked or otherwise
+ designated in writing by the copyright owner as "Not a Contribution."
+
+ "Contributor" shall mean Licensor and any individual or Legal Entity
+ on behalf of whom a Contribution has been received by Licensor and
+ subsequently incorporated within the Work.
+
+ 2. Grant of Copyright License. Subject to the terms and conditions of
+ this License, each Contributor hereby grants to You a perpetual,
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+ copyright license to reproduce, prepare Derivative Works of,
+ publicly display, publicly perform, sublicense, and distribute the
+ Work and such Derivative Works in Source or Object form.
+
+ 3. Grant of Patent License. Subject to the terms and conditions of
+ this License, each Contributor hereby grants to You a perpetual,
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+ (except as stated in this section) patent license to make, have made,
+ use, offer to sell, sell, import, and otherwise transfer the Work,
+ where such license applies only to those patent claims licensable
+ by such Contributor that are necessarily infringed by their
+ Contribution(s) alone or by combination of their Contribution(s)
+ with the Work to which such Contribution(s) was submitted. If You
+ institute patent litigation against any entity (including a
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
+ or a Contribution incorporated within the Work constitutes direct
+ or contributory patent infringement, then any patent licenses
+ granted to You under this License for that Work shall terminate
+ as of the date such litigation is filed.
+
+ 4. Redistribution. You may reproduce and distribute copies of the
+ Work or Derivative Works thereof in any medium, with or without
+ modifications, and in Source or Object form, provided that You
+ meet the following conditions:
+
+ (a) You must give any other recipients of the Work or
+ Derivative Works a copy of this License; and
+
+ (b) You must cause any modified files to carry prominent notices
+ stating that You changed the files; and
+
+ (c) You must retain, in the Source form of any Derivative Works
+ that You distribute, all copyright, patent, trademark, and
+ attribution notices from the Source form of the Work,
+ excluding those notices that do not pertain to any part of
+ the Derivative Works; and
+
+ (d) If the Work includes a "NOTICE" text file as part of its
+ distribution, then any Derivative Works that You distribute must
+ include a readable copy of the attribution notices contained
+ within such NOTICE file, excluding those notices that do not
+ pertain to any part of the Derivative Works, in at least one
+ of the following places: within a NOTICE text file distributed
+ as part of the Derivative Works; within the Source form or
+ documentation, if provided along with the Derivative Works; or,
+ within a display generated by the Derivative Works, if and
+ wherever such third-party notices normally appear. The contents
+ of the NOTICE file are for informational purposes only and
+ do not modify the License. You may add Your own attribution
+ notices within Derivative Works that You distribute, alongside
+ or as an addendum to the NOTICE text from the Work, provided
+ that such additional attribution notices cannot be construed
+ as modifying the License.
+
+ You may add Your own copyright statement to Your modifications and
+ may provide additional or different license terms and conditions
+ for use, reproduction, or distribution of Your modifications, or
+ for any such Derivative Works as a whole, provided Your use,
+ reproduction, and distribution of the Work otherwise complies with
+ the conditions stated in this License.
+
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
+ any Contribution intentionally submitted for inclusion in the Work
+ by You to the Licensor shall be under the terms and conditions of
+ this License, without any additional terms or conditions.
+ Notwithstanding the above, nothing herein shall supersede or modify
+ the terms of any separate license agreement you may have executed
+ with Licensor regarding such Contributions.
+
+ 6. Trademarks. This License does not grant permission to use the trade
+ names, trademarks, service marks, or product names of the Licensor,
+ except as required for reasonable and customary use in describing the
+ origin of the Work and reproducing the content of the NOTICE file.
+
+ 7. Disclaimer of Warranty. Unless required by applicable law or
+ agreed to in writing, Licensor provides the Work (and each
+ Contributor provides its Contributions) on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+ implied, including, without limitation, any warranties or conditions
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+ PARTICULAR PURPOSE. You are solely responsible for determining the
+ appropriateness of using or redistributing the Work and assume any
+ risks associated with Your exercise of permissions under this License.
+
+ 8. Limitation of Liability. In no event and under no legal theory,
+ whether in tort (including negligence), contract, or otherwise,
+ unless required by applicable law (such as deliberate and grossly
+ negligent acts) or agreed to in writing, shall any Contributor be
+ liable to You for damages, including any direct, indirect, special,
+ incidental, or consequential damages of any character arising as a
+ result of this License or out of the use or inability to use the
+ Work (including but not limited to damages for loss of goodwill,
+ work stoppage, computer failure or malfunction, or any and all
+ other commercial damages or losses), even if such Contributor
+ has been advised of the possibility of such damages.
+
+ 9. Accepting Warranty or Additional Liability. While redistributing
+ the Work or Derivative Works thereof, You may choose to offer,
+ and charge a fee for, acceptance of support, warranty, indemnity,
+ or other liability obligations and/or rights consistent with this
+ License. However, in accepting such obligations, You may act only
+ on Your own behalf and on Your sole responsibility, not on behalf
+ of any other Contributor, and only if You agree to indemnify,
+ defend, and hold each Contributor harmless for any liability
+ incurred by, or claims asserted against, such Contributor by reason
+ of your accepting any such warranty or additional liability.
+
+ END OF TERMS AND CONDITIONS
+
+ APPENDIX: How to apply the Apache License to your work.
+
+ To apply the Apache License to your work, attach the following
+ boilerplate notice, with the fields enclosed by brackets "[]"
+ replaced with your own identifying information. (Don't include
+ the brackets!) The text should be enclosed in the appropriate
+ comment syntax for the file format. We also recommend that a
+ file or class name and description of purpose be included on the
+ same "printed page" as the copyright notice for easier
+ identification within third-party archives.
+
+ Copyright 2024 Jiaqi Liu
+
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+ http://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
diff --git a/README.md b/README.md
new file mode 100644
index 0000000..8876be4
--- /dev/null
+++ b/README.md
@@ -0,0 +1,6 @@
+Tiger
+=====
+
+![Python Version Badge][Python Version Badge]
+
+Tiger is a Hugging Face space that helps language learner to extract movie subtitles
diff --git a/languages.py b/languages.py
new file mode 100644
index 0000000..0551511
--- /dev/null
+++ b/languages.py
@@ -0,0 +1,101 @@
+LANGUAGES = {
+ "en": "eng",
+ "zh": "zho",
+ "de": "deu",
+ "es": "spa",
+ "ru": "rus",
+ "ko": "kor",
+ "fr": "fra",
+ "ja": "jpn",
+ "pt": "por",
+ "tr": "tur",
+ "pl": "pol",
+ "ca": "cat",
+ "nl": "nld",
+ "ar": "ara",
+ "sv": "swe",
+ "it": "ita",
+ "id": "ind",
+ "hi": "hin",
+ "fi": "fin",
+ "vi": "vie",
+ "iw": "heb",
+ "uk": "ukr",
+ "el": "ell",
+ "ms": "msa",
+ "cs": "ces",
+ "ro": "ron",
+ "da": "dan",
+ "hu": "hun",
+ "ta": "tam",
+ "no": "nor",
+ "th": "tha",
+ "ur": "urd",
+ "hr": "hrv",
+ "bg": "bul",
+ "lt": "lit",
+ "la": "lat",
+ "mi": "mri",
+ "ml": "mal",
+ "cy": "cym",
+ "sk": "slk",
+ "te": "tel",
+ "fa": "fas",
+ "lv": "lav",
+ "bn": "ben",
+ "sr": "srp",
+ "az": "aze",
+ "sl": "slv",
+ "kn": "kan",
+ "et": "est",
+ "mk": "mkd",
+ "br": "bre",
+ "eu": "eus",
+ "is": "isl",
+ "hy": "hye",
+ "ne": "nep",
+ "mn": "mon",
+ "bs": "bos",
+ "kk": "kaz",
+ "sq": "sqi",
+ "sw": "swa",
+ "gl": "glg",
+ "mr": "mar",
+ "pa": "pan",
+ "si": "sin",
+ "km": "khm",
+ "sn": "sna",
+ "yo": "yor",
+ "so": "som",
+ "af": "afr",
+ "oc": "oci",
+ "ka": "kat",
+ "be": "bel",
+ "tg": "tgk",
+ "sd": "snd",
+ "gu": "guj",
+ "am": "amh",
+ "yi": "yid",
+ "lo": "lao",
+ "uz": "uzb",
+ "fo": "fao",
+ "ht": "hat",
+ "ps": "pus",
+ "tk": "tuk",
+ "nn": "nno",
+ "mt": "mlt",
+ "sa": "san",
+ "lb": "ltz",
+ "my": "mya",
+ "bo": "bod",
+ "tl": "tgl",
+ "mg": "mlg",
+ "as": "asm",
+ "tt": "tat",
+ "haw": "haw",
+ "ln": "lin",
+ "ha": "hau",
+ "ba": "bak",
+ "jw": "jav",
+ "su": "sun",
+}
\ No newline at end of file
diff --git a/packages.txt b/packages.txt
new file mode 100644
index 0000000..20645e6
--- /dev/null
+++ b/packages.txt
@@ -0,0 +1 @@
+ffmpeg
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..17f7a3f
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,10 @@
+ffmpeg==1.4
+ffmpeg_python==0.2.0
+numpy==1.23.3
+pytubefix==6.13.0
+requests==2.32.3
+streamlit==1.37.1
+streamlit_lottie==0.0.3
+torch==2.4.0
+transformers==4.44.0
+openai-whisper
\ No newline at end of file