videopython 0.1.41__tar.gz → 0.2.1__tar.gz

videopython-0.2.1/.gitignore

@@ -0,0 +1,140 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# type checker
+.pyre/
+.mypy_cache/
+
+# Random shit
+*.ipynb
+.vscode
+*.csv
+
+# Data directories
+data/downloaded/*.mp4
+data/exported/*.mp4
+!data/exported/example.mp4

videopython-0.2.1/PKG-INFO

@@ -0,0 +1,130 @@
+Metadata-Version: 2.3
+Name: videopython
+Version: 0.2.1
+Summary: Minimal video generation and processing library.
+Project-URL: Homepage, https://github.com/bartwojtowicz/videopython/
+Project-URL: Repository, https://github.com/bartwojtowicz/videopython/
+Project-URL: Documentation, https://github.com/bartwojtowicz/videopython/
+Author-email: Bartosz Wójtowicz <bartoszwojtowicz@outlook.com>, Bartosz Rudnikowicz <bartoszrudnikowicz840@gmail.com>, Piotr Pukisz <piotr.pukisz@gmail.com>
+License: Apache-2.0
+License-File: LICENSE
+Keywords: editing,generation,movie,opencv,python,video,videopython
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: <3.13,>=3.10
+Requires-Dist: numpy>=1.25.2
+Requires-Dist: opencv-python>=4.9.0.80
+Requires-Dist: pillow>=10.3.0
+Requires-Dist: pydub>=0.25.1
+Requires-Dist: tqdm>=4.66.3
+Provides-Extra: dev
+Requires-Dist: black==24.3.0; extra == 'dev'
+Requires-Dist: isort==5.12.0; extra == 'dev'
+Requires-Dist: mypy==1.8.0; extra == 'dev'
+Requires-Dist: pydub-stubs==0.25.1.1; extra == 'dev'
+Requires-Dist: pytest==7.4.0; extra == 'dev'
+Requires-Dist: types-pillow==10.2.0.20240213; extra == 'dev'
+Requires-Dist: types-tqdm==4.66.0.20240106; extra == 'dev'
+Provides-Extra: generation
+Requires-Dist: accelerate>=0.29.2; extra == 'generation'
+Requires-Dist: diffusers>=0.26.3; extra == 'generation'
+Requires-Dist: torch>=2.1.0; extra == 'generation'
+Requires-Dist: transformers>=4.38.1; extra == 'generation'
+Description-Content-Type: text/markdown
+
+# About
+
+Minimal video generation and processing library.
+
+## Setup 
+
+### Install ffmpeg
+```bash
+# Install with brew for MacOS:
+brew install ffmpeg
+# Install with apt-get for Ubuntu:
+sudo apt-get install ffmpeg
+```
+
+### Install with pip
+```bash
+pip install videopython[generation]
+```
+> You can install without `[generation]` dependencies for basic video handling and processing. 
+> The funcionalities found in `videopython.generation` won't work.
+
+## Basic Usage
+
+### Video handling
+
+```python
+from videopython.base.video import Video
+
+# Load videos and print metadata
+video1 = Video.from_path("tests/test_data/fast_benchmark.mp4")
+print(video1)
+
+video2 = Video.from_path("tests/test_data/slow_benchmark.mp4")
+print(video2)
+
+# Define the transformations
+from videopython.base.transforms import CutSeconds, ResampleFPS, Resize, TransformationPipeline
+
+pipeline = TransformationPipeline(
+    [CutSeconds(start=1.5, end=6.5), ResampleFPS(fps=30), Resize(width=1000, height=1000)]
+)
+video1 = pipeline.run(video1)
+video2 = pipeline.run(video2)
+
+# Combine videos, add audio and save
+from videopython.base.transitions import FadeTransition
+
+fade = FadeTransition(effect_time_seconds=3.0)
+video = fade.apply(videos=(video1, video2))
+video.add_audio_from_file("tests/test_data/test_audio.mp3")
+
+savepath = video.save()
+```
+
+### Video Generation
+
+> Using Nvidia A40 or better is recommended for the `videopython.generation` module.
+```python
+# Generate image and animate it
+from videopython.generation import ImageToVideo
+from videopython.generation import TextToImage
+from videopython.generation import TextToMusic
+
+image = TextToImage().generate_image(prompt="Golden Retriever playing in the park")
+video = ImageToVideo().generate_video(image=image, fps=24)
+
+# Video generation directly from prompt
+from videopython.generation import TextToVideo
+video_gen = TextToVideo()
+video = video_gen.generate_video("Dogs playing in the snow")
+for _ in range(10):
+    video += video_gen.generate_video("Dogs playing in the snow")
+
+# Cut the first 2 seconds
+from videopython.base.transforms import CutSeconds
+transformed_video = CutSeconds(start_second=0, end_second=2).apply(video.copy())
+
+# Upsample to 30 FPS
+from videopython.base.transforms import ResampleFPS
+transformed_video = ResampleFPS(new_fps=30).apply(transformed_video)
+
+# Resize to 1000x1000
+from videopython.base.transforms import Resize
+transformed_video = Resize(width=1000, height=1000).apply(transformed_video)
+
+# Add generated music
+# MusicGen cannot generate more than 1503 tokens (~30seconds of audio)
+text_to_music = TextToMusic()
+audio = text_to_music.generate_audio("Happy dogs playing together in a park", max_new_tokens=256)
+transformed_video.add_audio(audio=audio)
+
+filepath = transformed_video.save()
+```

{videopython-0.1.41 → videopython-0.2.1}/README.md

@@ -20,12 +20,46 @@ pip install videopython[generation]
 > The funcionalities found in `videopython.generation` won't work.
 
 ## Basic Usage
-> Using Nvidia A40 or better is recommended for the `videopython.generation` module.
+
+### Video handling
 
+```python
+from videopython.base.video import Video
+
+# Load videos and print metadata
+video1 = Video.from_path("tests/test_data/fast_benchmark.mp4")
+print(video1)
+
+video2 = Video.from_path("tests/test_data/slow_benchmark.mp4")
+print(video2)
+
+# Define the transformations
+from videopython.base.transforms import CutSeconds, ResampleFPS, Resize, TransformationPipeline
+
+pipeline = TransformationPipeline(
+    [CutSeconds(start=1.5, end=6.5), ResampleFPS(fps=30), Resize(width=1000, height=1000)]
+)
+video1 = pipeline.run(video1)
+video2 = pipeline.run(video2)
+
+# Combine videos, add audio and save
+from videopython.base.transitions import FadeTransition
+
+fade = FadeTransition(effect_time_seconds=3.0)
+video = fade.apply(videos=(video1, video2))
+video.add_audio_from_file("tests/test_data/test_audio.mp3")
+
+savepath = video.save()
+```
+
+### Video Generation
+
+> Using Nvidia A40 or better is recommended for the `videopython.generation` module.
 ```python
 # Generate image and animate it
 from videopython.generation import ImageToVideo
 from videopython.generation import TextToImage
+from videopython.generation import TextToMusic
 
 image = TextToImage().generate_image(prompt="Golden Retriever playing in the park")
 video = ImageToVideo().generate_video(image=image, fps=24)
@@ -49,5 +83,11 @@ transformed_video = ResampleFPS(new_fps=30).apply(transformed_video)
 from videopython.base.transforms import Resize
 transformed_video = Resize(width=1000, height=1000).apply(transformed_video)
 
+# Add generated music
+# MusicGen cannot generate more than 1503 tokens (~30seconds of audio)
+text_to_music = TextToMusic()
+audio = text_to_music.generate_audio("Happy dogs playing together in a park", max_new_tokens=256)
+transformed_video.add_audio(audio=audio)
+
 filepath = transformed_video.save()
 ```

videopython-0.2.1/pyproject.toml

@@ -0,0 +1,88 @@
+[project]
+name = "videopython"
+version = "0.2.1"
+description = "Minimal video generation and processing library."
+authors = [
+    { name = "Bartosz Wójtowicz", email = "bartoszwojtowicz@outlook.com" },
+    { name = "Bartosz Rudnikowicz", email = "bartoszrudnikowicz840@gmail.com" },
+    { name = "Piotr Pukisz", email = "piotr.pukisz@gmail.com" }
+]
+license = { text = "Apache-2.0" }
+readme = "README.md"
+requires-python = ">=3.10, <3.13"
+keywords = ["python", "videopython", "video", "movie", "opencv", "generation", "editing"]
+classifiers = [
+    "License :: OSI Approved :: Apache Software License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Operating System :: OS Independent",
+]
+
+dependencies = [
+    "numpy>=1.25.2",
+    "opencv-python>=4.9.0.80",
+    "pillow>=10.3.0",
+    "pydub>=0.25.1",
+    "tqdm>=4.66.3",
+]
+
+[project.optional-dependencies]
+dev = [
+    "black==24.3.0",
+    "isort==5.12.0",
+    "mypy==1.8.0",
+    "pytest==7.4.0",
+    "types-Pillow==10.2.0.20240213",
+    "types-tqdm==4.66.0.20240106",
+    "pydub-stubs==0.25.1.1",
+]
+generation = [
+    "accelerate>=0.29.2",
+    "diffusers>=0.26.3",
+    "torch>=2.1.0",
+    "transformers>=4.38.1",
+]
+
+[project.urls]
+Homepage = "https://github.com/bartwojtowicz/videopython/"
+Repository = "https://github.com/bartwojtowicz/videopython/"
+Documentation = "https://github.com/bartwojtowicz/videopython/"
+
+[tool.rye]
+managed = true
+dev-dependencies = [
+    "black==24.3.0",
+    "isort==5.12.0",
+    "mypy==1.8.0",
+    "pytest==7.4.0",
+    "types-Pillow==10.2.0.20240213",
+    "types-tqdm==4.66.0.20240106",
+    "pydub-stubs==0.25.1.1",
+]
+
+[tool.rye.scripts]
+test-unit = "pytest"
+test-type = "mypy src"
+test-static = { chain = [
+    "black src -l 120 --check",
+    "isort src --profile black --check"
+]}
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/videopython"]
+
+[tool.hatch.build.targets.sdist]
+include = ["src/videopython", "src/videopython/py.typed"]
+
+[tool.mypy]
+mypy_path = "stubs"
+
+[tool.pytest]
+testpaths = ["src/tests"]
+python_files = ["test_*.py"]
+addopts = "-v --tb=short"

{videopython-0.1.41 → videopython-0.2.1}/src/videopython/base/transforms.py

@@ -53,22 +53,22 @@ class TransformationPipeline:
 
 
 class CutFrames(Transformation):
-    def __init__(self, start_frame: int, end_frame: int):
-        self.start_frame = start_frame
-        self.end_frame = end_frame
+    def __init__(self, start: int, end: int):
+        self.start = start
+        self.end = end
 
     def apply(self, video: Video) -> Video:
-        video = video[self.start_frame : self.end_frame]
+        video = video[self.start : self.end]
         return video
 
 
 class CutSeconds(Transformation):
-    def __init__(self, start_second: float | int, end_second: float | int):
-        self.start_second = start_second
-        self.end_second = end_second
+    def __init__(self, start: float | int, end: float | int):
+        self.start = start
+        self.end = end
 
     def apply(self, video: Video) -> Video:
-        video = video[round(self.start_second * video.fps) : round(self.end_second * video.fps)]
+        video = video[round(self.start * video.fps) : round(self.end * video.fps)]
         return video
 
 
@@ -112,18 +112,18 @@ class Resize(Transformation):
 
 
 class ResampleFPS(Transformation):
-    def __init__(self, new_fps: int | float):
-        self.new_fps = float(new_fps)
+    def __init__(self, fps: int | float):
+        self.fps = float(fps)
 
     def _downsample(self, video: Video) -> Video:
-        target_frame_count = int(len(video.frames) * (self.new_fps / video.fps))
+        target_frame_count = int(len(video.frames) * (self.fps / video.fps))
         new_frame_indices = np.round(np.linspace(0, len(video.frames) - 1, target_frame_count)).astype(int)
         video.frames = video.frames[new_frame_indices]
-        video.fps = self.new_fps
+        video.fps = self.fps
         return video
 
     def _upsample(self, video: Video) -> Video:
-        target_frame_count = int(len(video.frames) * (self.new_fps / video.fps))
+        target_frame_count = int(len(video.frames) * (self.fps / video.fps))
         new_frame_indices = np.linspace(0, len(video.frames) - 1, target_frame_count)
         new_frames = []
         for i in tqdm(range(len(new_frame_indices) - 1)):
@@ -134,17 +134,17 @@ class ResampleFPS(Transformation):
             ]
             new_frames.append(new_frame.astype(np.uint8))
         video.frames = np.array(new_frames, dtype=np.uint8)
-        video.fps = self.new_fps
+        video.fps = self.fps
         return video
 
     def apply(self, video: Video) -> Video:
-        if video.fps == self.new_fps:
+        if video.fps == self.fps:
             return video
-        elif video.fps > self.new_fps:
-            print(f"Downsampling video from {video.fps} to {self.new_fps} FPS.")
+        elif video.fps > self.fps:
+            print(f"Downsampling video from {video.fps} to {self.fps} FPS.")
             video = self._downsample(video)
         else:
-            print(f"Upsampling video from {video.fps} to {self.new_fps} FPS.")
+            print(f"Upsampling video from {video.fps} to {self.fps} FPS.")
             video = self._upsample(video)
         return video