videopython 0.1.4__tar.gz → 0.2.0__tar.gz

{videopython-0.1.4 → videopython-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: videopython
-Version: 0.1.4
+Version: 0.2.0
 Summary: Minimal video generation and processing library.
 Author-email: Bartosz Wójtowicz <bartoszwojtowicz@outlook.com>, Bartosz Rudnikowicz <bartoszrudnikowicz840@gmail.com>, Piotr Pukisz <piotr.pukisz@gmail.com>
 License:                                  Apache License
@@ -250,8 +250,41 @@ 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

{videopython-0.1.4 → videopython-0.2.0}/README.md

@@ -20,8 +20,41 @@ 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

{videopython-0.1.4 → videopython-0.2.0}/pyproject.toml

@@ -11,7 +11,7 @@ include = ["videopython.*"]
 
 [project]
 name = "videopython"
-version = "0.1.4"
+version = "0.2.0"
 description = "Minimal video generation and processing library."
 readme = "README.md"
 requires-python = ">=3.10"

videopython-0.2.0/src/videopython/base/exceptions.py

@@ -0,0 +1,2 @@
+class OutOfBoundsError(Exception):
+    pass

{videopython-0.1.4 → videopython-0.2.0}/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
 

videopython-0.2.0/src/videopython/generation/pipeline.py

@@ -0,0 +1,32 @@
+import cv2
+import numpy as np
+from PIL import Image
+
+from videopython.base.transforms import Resize
+from videopython.generation import ImageToVideo, TextToImage
+
+N_ITERATIONS = 11
+PRMOPT = "Sunset at the sea, cimenatic view"
+
+
+def main():
+    text_to_image = TextToImage()
+    image_to_video = ImageToVideo()
+
+    target_height = 576
+    target_width = 1024
+
+    base_image = text_to_image.generate_image(PRMOPT)
+    image = cv2.resize(np.asarray(base_image), (target_width, target_height))
+
+    video = image_to_video.generate_video(image)
+
+    for i in range(N_ITERATIONS - 1):
+        print(f"Generating {i+2}/{N_ITERATIONS}...")
+        video += image_to_video.generate_video(Image.fromarray(video.frames[-1]))
+
+    video.save()
+
+
+if __name__ == "__main__":
+    main()

{videopython-0.1.4 → videopython-0.2.0}/src/videopython/generation/video.py

@@ -6,7 +6,7 @@ from PIL.Image import Image
 from videopython.base.video import Video
 
 TEXT_TO_VIDEO_MODEL = "cerspense/zeroscope_v2_576w"
-IMAGE_TO_VIDEO_MODEL = "stabilityai/stable-video-diffusion-img2vid-xt"
+IMAGE_TO_VIDEO_MODEL = "stabilityai/stable-video-diffusion-img2vid-xt-1-1"
 
 
 class TextToVideo:

{videopython-0.1.4 → videopython-0.2.0}/src/videopython/utils/image.py

@@ -4,8 +4,8 @@ import cv2
 import numpy as np
 from PIL import Image, ImageDraw, ImageFont
 
+from videopython.base.exceptions import OutOfBoundsError
 from videopython.base.video import Video
-from videopython.exceptions import OutOfBoundsError
 
 
 class ImageText:
@@ -247,7 +247,7 @@ class SlideOverImage:
         self.fps = fps
         self.length_seconds = length_seconds
 
-    def slide(self, image: np.ndarray) -> Video:
+    def apply(self, image: np.ndarray) -> Video:
         image = self._resize(image)
         max_offset = image.shape[1] - self.video_width
         frame_count = round(self.fps * self.length_seconds)

{videopython-0.1.4 → videopython-0.2.0}/src/videopython.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: videopython
-Version: 0.1.4
+Version: 0.2.0
 Summary: Minimal video generation and processing library.
 Author-email: Bartosz Wójtowicz <bartoszwojtowicz@outlook.com>, Bartosz Rudnikowicz <bartoszrudnikowicz840@gmail.com>, Piotr Pukisz <piotr.pukisz@gmail.com>
 License:                                  Apache License
@@ -250,8 +250,41 @@ 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

{videopython-0.1.4 → videopython-0.2.0}/src/videopython.egg-info/SOURCES.txt

@@ -12,12 +12,14 @@ src/videopython.egg-info/top_level.txt
 src/videopython/base/__init__.py
 src/videopython/base/compose.py
 src/videopython/base/effects.py
+src/videopython/base/exceptions.py
 src/videopython/base/transforms.py
 src/videopython/base/transitions.py
 src/videopython/base/video.py
 src/videopython/generation/__init__.py
 src/videopython/generation/audio.py
 src/videopython/generation/image.py
+src/videopython/generation/pipeline.py
 src/videopython/generation/video.py
 src/videopython/utils/__init__.py
 src/videopython/utils/common.py

{videopython-0.1.4 → videopython-0.2.0}/tests/test_transforms.py

@@ -7,7 +7,7 @@ from videopython.base.transforms import CutFrames, CutSeconds, Resize
 
 @pytest.mark.parametrize("start, end", [(0, 100), (100, 101), (100, 120)])
 def test_cut_frames(start, end, small_video):
-    cut_frames = CutFrames(start_frame=start, end_frame=end)
+    cut_frames = CutFrames(start=start, end=end)
     start_frame = small_video.frames[start].copy()
     transformed = cut_frames.apply(small_video)
     assert len(transformed.frames) == (end - start)
@@ -16,7 +16,7 @@ def test_cut_frames(start, end, small_video):
 
 @pytest.mark.parametrize("start, end", [(0, 0.5), (0, 1), (0.5, 1.5)])
 def test_cut_seconds(start, end, small_video):
-    cut_seconds = CutSeconds(start_second=start, end_second=end)
+    cut_seconds = CutSeconds(start=start, end=end)
     start_frame = small_video.frames[round(start * small_video.fps)].copy()
     transformed = cut_seconds.apply(small_video)
     assert len(transformed.frames) == round((end - start) * small_video.fps)

{videopython-0.1.4 → videopython-0.2.0}/tests/test_utils.py

@@ -3,7 +3,7 @@ from videopython.utils.image import SlideOverImage
 
 def test_slide_over_image(small_image):
     slide = SlideOverImage(direction="left", video_shape=(150, small_image.shape[0]), fps=24.0, length_seconds=1.0)
-    video = slide.slide(small_image)
+    video = slide.apply(small_image)
 
     assert video.frames.shape[0] == 24
     assert video.frames.shape[1] == small_image.shape[0]