videopython 0.1.41__py3-none-any.whl → 0.2.0__py3-none-any.whl

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/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/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/utils/image.py

@@ -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.41.dist-info → videopython-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: videopython
-Version: 0.1.41
+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.41.dist-info → videopython-0.2.0.dist-info}/RECORD

@@ -2,18 +2,19 @@ videopython/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,
 videopython/base/compose.py,sha256=pti12VY3Yg7TZZiENPF6veM8POWssfsK8ePDdGlhAhA,1968
 videopython/base/effects.py,sha256=ZFUWrgVWTn4uWpxPfTQSQQKEZN5ns4btMofOZNHCeQc,7540
 videopython/base/exceptions.py,sha256=68_16lUPOR9_zhWdeBGS8_NFI32VbrcoDbN5KHHg0_w,44
-videopython/base/transforms.py,sha256=yDtM1uZOacB0OLPQnSGHl5upoNNeN0dPWXe0hJPeV-I,6004
+videopython/base/transforms.py,sha256=VP8SGArokfXN-IE7tk-3i0oMM3HV5zpJa-GLj7BoeRo,5856
 videopython/base/transitions.py,sha256=zYsxIgiVfN9P-CoGWUWRYFBr_0inX1sAJ02gyIEQ678,3694
 videopython/base/video.py,sha256=kG-juKN-da5NzV89YxZl5JkyMTJFkgPceh4yuAUnsQs,11099
 videopython/generation/__init__.py,sha256=Qse024UgiS9OxXzbbInyZ-9cpfI4enR2Dcds4lLDpNA,201
 videopython/generation/audio.py,sha256=BTc-3vJ5e6D0lt2OPo2hfOcUqhNXIcvRLNoo2oQ470M,777
 videopython/generation/image.py,sha256=i8zJm0WXn_Pykby9Urw1kzDcla6ArYhRgG-ueRdoAJ0,675
-videopython/generation/video.py,sha256=WMFKKUSfIkQmxL6xhUb-MeAiHU6uOF_oFpmf69H8V8g,1827
+videopython/generation/pipeline.py,sha256=v8GHkGNLErnQBjzNA8oem7fRv7YOx_NdduEC47kQkf0,773
+videopython/generation/video.py,sha256=206YON_XjPTYyjIJ3j5uBgd_yHmCDg7SqbkIU9GzEgw,1831
 videopython/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 videopython/utils/common.py,sha256=F-30YoKUwWDI7HiJUWw0gRFUguhShSVaxT0aFfvpifg,936
-videopython/utils/image.py,sha256=8m1uzyfrj5Kdbw7IZyqwj-6NXK4KH-szJxtN_EQva4s,12084
-videopython-0.1.41.dist-info/LICENSE,sha256=nJL9jVOt2MSW7swNDq4Y6oD_n9bLI0B0afr8ougtZ6s,10832
-videopython-0.1.41.dist-info/METADATA,sha256=Hu511G-WnKEwOn9gG_53ywH1NDAH2pA5d_LzFUuYmfM,15659
-videopython-0.1.41.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-videopython-0.1.41.dist-info/top_level.txt,sha256=OikTGG8Swfw_syz--1atAn5KQ4GH9Pye17eATGred-Q,12
-videopython-0.1.41.dist-info/RECORD,,
+videopython/utils/image.py,sha256=gng1by8ieYRVs1DlxHPMqYIPxKc1WWwBm8W5oASBKfY,12084
+videopython-0.2.0.dist-info/LICENSE,sha256=nJL9jVOt2MSW7swNDq4Y6oD_n9bLI0B0afr8ougtZ6s,10832
+videopython-0.2.0.dist-info/METADATA,sha256=W8FecvXP3eT7BSeUNjDKiykVPIrcwXsademTs9gl3MU,16531
+videopython-0.2.0.dist-info/WHEEL,sha256=mguMlWGMX-VHnMpKOjjQidIo1ssRlCFu4a4mBpz1s2M,91
+videopython-0.2.0.dist-info/top_level.txt,sha256=OikTGG8Swfw_syz--1atAn5KQ4GH9Pye17eATGred-Q,12
+videopython-0.2.0.dist-info/RECORD,,

{videopython-0.1.41.dist-info → videopython-0.2.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (70.1.1)
 Root-Is-Purelib: true
 Tag: py3-none-any