make path os independent;add devcontainer

.devcontainer/devcontainer.json

@@ -0,0 +1,22 @@
+// For format details, see https://aka.ms/devcontainer.json. For config options, see the
+// README at: https://github.com/devcontainers/templates/tree/main/src/python
+{
+	"name": "Python 3",
+	// Or use a Dockerfile or Docker Compose file. More info: https://containers.dev/guide/dockerfile
+	"image": "mcr.microsoft.com/devcontainers/python:1-3.11-bookworm"
+
+	// Features to add to the dev container. More info: https://containers.dev/features.
+	// "features": {},
+
+	// Use 'forwardPorts' to make a list of ports inside the container available locally.
+	// "forwardPorts": [],
+
+	// Use 'postCreateCommand' to run commands after the container is created.
+	// "postCreateCommand": "pip3 install --user -r requirements.txt",
+
+	// Configure tool-specific properties.
+	// "customizations": {},
+
+	// Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
+	// "remoteUser": "root"
+}

.gitignore

@@ -1,8 +1,10 @@
+data/*
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]
 *$py.class
 data/*
+Python-3.11.6/
 
 # C extensions
 *.so

Pipfile

@@ -13,4 +13,3 @@ tqdm = "*"
 
 [requires]
 python_version = "3.11"
-python_full_version = "3.11.6"

Pipfile.lock

@@ -1,11 +1,10 @@
 {
     "_meta": {
         "hash": {
-            "sha256": "f36698f021c6ce2b4d6740d104a40e1aaaea72469d2dee5c85f55d0ab5dcfdf1"
+            "sha256": "27098db4432ef9d9cfe405af23f5980c4af71c5d79ef8b9f67f922b6b4b6a5ed"
         },
         "pipfile-spec": 6,
         "requires": {
-            "python_full_version": "3.11.6",
             "python_version": "3.11"
         },
         "sources": [
@@ -170,14 +169,6 @@
             "markers": "python_full_version >= '3.7.0'",
             "version": "==3.3.2"
         },
-        "colorama": {
-            "hashes": [
-                "sha256:08695f5cb7ed6e0531a20572697297273c47b8cae5a63ffc6d6ed5c201be6e44",
-                "sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6"
-            ],
-            "markers": "platform_system == 'Windows'",
-            "version": "==0.4.6"
-        },
         "coloredlogs": {
             "hashes": [
                 "sha256:612ee75c546f53e92e70049c9dbfcc18c935a2b9a53b66085ce9ef6a6e5c0934",
@@ -229,6 +220,7 @@
                 "sha256:591809328b93c8e4594d52097ec6352a270a81fbb7b956254967f28700f7e4da"
             ],
             "index": "pypi",
+            "markers": "python_version >= '3.8'",
             "version": "==0.10.0"
         },
         "filelock": {
@@ -248,11 +240,11 @@
         },
         "fsspec": {
             "hashes": [
-                "sha256:6271f1d3075a378bfe432f6f42bf7e1d2a6ba74f78dd9b512385474c579146a0",
-                "sha256:c4da01a35ac65c853f833e43f67802c25213f560820d54ddf248f92eddd5e990"
+                "sha256:8548d39e8810b59c38014934f6b31e57f40c1b20f911f4cc2b85389c7e9bf0cb",
+                "sha256:d800d87f72189a745fa3d6b033b9dc4a34ad069f60ca60b943a63599f5501960"
             ],
             "markers": "python_version >= '3.8'",
-            "version": "==2023.12.1"
+            "version": "==2023.12.2"
         },
         "huggingface-hub": {
             "hashes": [
@@ -280,11 +272,11 @@
         },
         "imageio": {
             "hashes": [
-                "sha256:39999d05eb500089e60be467dd7d618f56e142229b44c3961c2b420eeb538d7e",
-                "sha256:d580d6576d0ae39c459a444a23f6f61fe72123a3df2264f5fce8c87784a4be2e"
+                "sha256:78722d40b137bd98f5ec7312119f8aea9ad2049f76f434748eb306b6937cc1ce",
+                "sha256:c5094c48ccf6b2e6da8b4061cd95e1209380afafcbeae4a4e280938cce227e1d"
             ],
             "markers": "python_version >= '3.4'",
-            "version": "==2.33.0"
+            "version": "==2.33.1"
         },
         "imageio-ffmpeg": {
             "hashes": [
@@ -475,20 +467,13 @@
             "markers": "python_version >= '3.8'",
             "version": "==4.25.1"
         },
-        "pyreadline3": {
-            "hashes": [
-                "sha256:6f3d1f7b8a31ba32b73917cefc1f28cc660562f39aea8646d30bd6eff21f7bae",
-                "sha256:b0efb6516fd4fb07b45949053826a62fa4cb353db5be2bbb4a7aa1fdd1e345fb"
-            ],
-            "markers": "python_version >= '3.8' and sys_platform == 'win32'",
-            "version": "==3.4.1"
-        },
         "pytube": {
             "hashes": [
                 "sha256:076052efe76f390dfa24b1194ff821d4e86c17d41cb5562f3a276a8bcbfc9d1d",
                 "sha256:07b9904749e213485780d7eb606e5e5b8e4341aa4dccf699160876da00e12d78"
             ],
             "index": "pypi",
+            "markers": "python_version >= '3.7'",
             "version": "==15.0.0"
         },
         "pyyaml": {
@@ -681,15 +666,16 @@
                 "sha256:d88e651f9db8d8551a62556d3cff9e3034274ca5d66e93197cf2490e2dcb69c7"
             ],
             "index": "pypi",
+            "markers": "python_version >= '3.7'",
             "version": "==4.66.1"
         },
         "typing-extensions": {
             "hashes": [
-                "sha256:8f92fc8806f9a6b641eaa5318da32b44d401efaac0f6678c9bc448ba3605faa0",
-                "sha256:df8e4339e9cb77357558cbdbceca33c303714cf861d1eef15e1070055ae8b7ef"
+                "sha256:23478f88c37f27d76ac8aee6c905017a143b0b1b886c3c9f66bc2fd94f9f5783",
+                "sha256:af72aea155e91adfc61c3ae9e0e342dbc0cba726d6cba4b6c72c1f34e47291cd"
             ],
             "markers": "python_version >= '3.8'",
-            "version": "==4.8.0"
+            "version": "==4.9.0"
         },
         "urllib3": {
             "hashes": [

main.py

@@ -29,7 +29,7 @@ def main(video_url,
             pbar.update(33.34)
             return
         if not os.path.exists(srt_path):
-            transcriber(stream_title, srt_path)
+            transcriber(os.path.join(invideo_dir,invideo_filename), srt_path)
         pbar.update(66.66)
         subtitler(os.path.join(invideo_dir,invideo_filename), srt_path, outvideo_path, fontsize,bg_color)
         pbar.update(33.34)
@@ -39,7 +39,7 @@ if __name__ == '__main__':
     parser.add_argument('--invideo_filename', required=True, type=str, help='filename and extension of ')
     parser.add_argument('--outvideo_path', required=True, help='path to the output video')
     parser.add_argument('--video_url', required=False, default=None, type=str, help='A video file to be subtitled (Optional)')
-    parser.add_argument('--srt_path', required=False, default="./data/audio.srt", type=str, help='path to the srt file (default: data/audio.srt)')
+    parser.add_argument('--srt_path', required=False, default="data/audio.srt", type=str, help='path to the srt file (default: data/audio.srt)')
     parser.add_argument('--fontsize', required=False, default=32, type=int, help='Font size for captions (int)')
     parser.add_argument('--bg_color', required=False, default="#070a13b3", type=str, help='Hex color value for caption background colour.')
     args = parser.parse_args()

utils/subtitler.py

@@ -1,4 +1,5 @@
 from moviepy.editor import VideoFileClip, CompositeVideoClip, TextClip
+import os
 
 def parse_srt(srt_file):
     """Parse the SRT file and return a list of (start, end, text) for each subtitle."""
@@ -23,6 +24,10 @@ def parse_srt(srt_file):
 
 def subtitler(video_file, srt_file, output_file,fontsize, bg_color):
     """Add subtitles from an SRT file to a video."""
+    video_file = os.path.abspath(video_file)
+    srt_file = os.path.abspath(srt_file)
+    output_file = os.path.abspath(output_file)
+    
     clip = VideoFileClip(video_file)
     subtitles = parse_srt(srt_file)
 

utils/transcriber.py

@@ -1,5 +1,5 @@
 from faster_whisper import WhisperModel
-import logging
+import logging, os
 
 logging.basicConfig(filename='main.log',
                 encoding='utf-8',
@@ -15,6 +15,7 @@ def write_srt(segments, srt_path):
             file.write(f"{i+1}\n{segment.start} --> {segment.end}\n{segment.text}\n\n")
 
 def transcriber(input_path:str, srt_path:str):
+    str_path = os.path.abspath(srt_path)
     model_size = "large-v3"
 
     # Run on GPU with FP16