videopython 0.3.0__tar.gz → 0.4.0__tar.gz

{videopython-0.3.0 → videopython-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: videopython
-Version: 0.3.0
+Version: 0.4.0
 Summary: Minimal video generation and processing library.
 Project-URL: Homepage, https://github.com/bartwojtowicz/videopython/
 Project-URL: Repository, https://github.com/bartwojtowicz/videopython/
@@ -19,7 +19,7 @@ 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: soundpython>=0.1.9
+Requires-Dist: soundpython>=0.1.11
 Requires-Dist: tqdm>=4.66.3
 Description-Content-Type: text/markdown
 
@@ -39,10 +39,10 @@ sudo apt-get install ffmpeg
 
 ### Install with pip
 ```bash
-pip install videopython[generation]
+pip install videopython[ai]
 ```
-> You can install without `[generation]` dependencies for basic video handling and processing. 
-> The funcionalities found in `videopython.generation` won't work.
+> You can install without `[ai]` dependencies for basic video handling and processing. 
+> The funcionalities found in `videopython.ai` won't work.
 
 ## Basic Usage
 
@@ -52,10 +52,10 @@ pip install videopython[generation]
 from videopython.base.video import Video
 
 # Load videos and print metadata
-video1 = Video.from_path("tests/test_data/fast_benchmark.mp4")
+video1 = Video.from_path("tests/test_data/small_video.mp4")
 print(video1)
 
-video2 = Video.from_path("tests/test_data/slow_benchmark.mp4")
+video2 = Video.from_path("tests/test_data/big_video.mp4")
 print(video2)
 
 # Define the transformations
@@ -79,18 +79,18 @@ savepath = video.save()
 
 ### Video Generation
 
-> Using Nvidia A40 or better is recommended for the `videopython.generation` module.
+> Using Nvidia A40 or better is recommended for the `videopython.ai` module.
 ```python
 # Generate image and animate it
-from videopython.generation import ImageToVideo
-from videopython.generation import TextToImage
-from videopython.generation import TextToMusic
+from videopython.ai.generation import ImageToVideo
+from videopython.ai.generation import TextToImage
+from videopython.ai.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
+from videopython.ai.generation import TextToVideo
 video_gen = TextToVideo()
 video = video_gen.generate_video("Dogs playing in the snow")
 for _ in range(10):

{videopython-0.3.0 → videopython-0.4.0}/README.md

@@ -14,10 +14,10 @@ sudo apt-get install ffmpeg
 
 ### Install with pip
 ```bash
-pip install videopython[generation]
+pip install videopython[ai]
 ```
-> You can install without `[generation]` dependencies for basic video handling and processing. 
-> The funcionalities found in `videopython.generation` won't work.
+> You can install without `[ai]` dependencies for basic video handling and processing. 
+> The funcionalities found in `videopython.ai` won't work.
 
 ## Basic Usage
 
@@ -27,10 +27,10 @@ pip install videopython[generation]
 from videopython.base.video import Video
 
 # Load videos and print metadata
-video1 = Video.from_path("tests/test_data/fast_benchmark.mp4")
+video1 = Video.from_path("tests/test_data/small_video.mp4")
 print(video1)
 
-video2 = Video.from_path("tests/test_data/slow_benchmark.mp4")
+video2 = Video.from_path("tests/test_data/big_video.mp4")
 print(video2)
 
 # Define the transformations
@@ -54,18 +54,18 @@ savepath = video.save()
 
 ### Video Generation
 
-> Using Nvidia A40 or better is recommended for the `videopython.generation` module.
+> Using Nvidia A40 or better is recommended for the `videopython.ai` module.
 ```python
 # Generate image and animate it
-from videopython.generation import ImageToVideo
-from videopython.generation import TextToImage
-from videopython.generation import TextToMusic
+from videopython.ai.generation import ImageToVideo
+from videopython.ai.generation import TextToImage
+from videopython.ai.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
+from videopython.ai.generation import TextToVideo
 video_gen = TextToVideo()
 video = video_gen.generate_video("Dogs playing in the snow")
 for _ in range(10):

{videopython-0.3.0 → videopython-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "videopython"
-version = "0.3.0"
+version = "0.4.0"
 description = "Minimal video generation and processing library."
 authors = [
     { name = "Bartosz Wójtowicz", email = "bartoszwojtowicz@outlook.com" },
@@ -24,24 +24,27 @@ dependencies = [
     "opencv-python>=4.9.0.80",
     "pillow>=10.3.0",
     "pydub>=0.25.1",
-    "soundpython>=0.1.9",
+    "soundpython>=0.1.11",
     "tqdm>=4.66.3",
+    
 ]
 
 [dependency-groups]
 dev = [
-    "black>=24.3.0",
-    "isort>=5.12.0",
+    "ruff>=0.1.14",
     "mypy>=1.8.0",
     "pytest>=7.4.0",
     "types-Pillow>=10.2.0.20240213",
     "types-tqdm>=4.66.0.20240106",
+    "pytest-cov>=6.1.1",
 ]
-generation = [
+ai = [
     "accelerate>=0.29.2",
     "diffusers>=0.26.3",
     "torch>=2.1.0",
     "transformers>=4.38.1",
+    "openai-whisper>=20240930",
+    "numba>=0.61.0",
 ]
 
 [project.urls]
@@ -67,3 +70,19 @@ pythonpath = [".src/"]
 testpaths = ["src/tests"]
 python_files = ["test_*.py"]
 addopts = "-v --tb=short"
+
+[tool.ruff]
+line-length = 120
+target-version = "py310"
+
+[tool.ruff.lint]
+select = [
+    "E",   # pycodestyle errors
+    "F",   # pyflakes
+    "I",   # isort
+]
+isort.known-first-party = ["videopython"]
+
+[tool.ruff.format]
+indent-style = "space"
+quote-style = "double"

{videopython-0.3.0/src/videopython → videopython-0.4.0/src/videopython/ai}/generation/image.py

@@ -1,6 +1,3 @@
-import io
-import os
-
 import torch
 from diffusers import DiffusionPipeline
 from PIL import Image

videopython-0.4.0/src/videopython/ai/understanding/transcribe.py

@@ -0,0 +1,37 @@
+from typing import Literal
+
+import whisper
+
+from videopython.base.transcription import Transcription, TranscriptionSegment
+from videopython.base.video import Video
+
+
+class VideoTranscription:
+    def __init__(self, model_name: Literal["tiny", "base", "small", "medium", "large", "turbo"] = "small") -> None:
+        self.model = whisper.load_model(name=model_name)
+
+    def transcribe_video(self, video: Video) -> Transcription:
+        """Transcribes video to text.
+
+        Args:
+            video: Video to transcribe.
+
+        Returns:
+            List of dictionaries with segments of text and their start and end times.
+        """
+        if video.audio.is_silent:
+            return Transcription(segments=[])
+
+        audio = video.audio.to_mono()
+        audio = audio.resample(whisper.audio.SAMPLE_RATE)
+        audio_data = audio.data
+
+        transcription = self.model.transcribe(audio=audio_data, word_timestamps=True)
+
+        transcription_segments = [
+            TranscriptionSegment(start=segment["start"], end=segment["end"], text=segment["text"])
+            for segment in transcription["segments"]
+        ]
+        result = Transcription(segments=transcription_segments)
+
+        return result

{videopython-0.3.0 → videopython-0.4.0}/src/videopython/base/effects.py

@@ -156,13 +156,13 @@ class Zoom(Effect):
 
         width = video.metadata.width
         height = video.metadata.height
-        crop_sizes_w, crop_sizes_h = np.linspace(width // self.zoom_factor, width, n_frames), np.linspace(
-            height // self.zoom_factor, height, n_frames
+        crop_sizes_w, crop_sizes_h = (
+            np.linspace(width // self.zoom_factor, width, n_frames),
+            np.linspace(height // self.zoom_factor, height, n_frames),
         )
 
         if self.mode == "in":
             for frame, w, h in tqdm(zip(video.frames, reversed(crop_sizes_w), reversed(crop_sizes_h))):
-
                 x = width / 2 - w / 2
                 y = height / 2 - h / 2
 

videopython-0.4.0/src/videopython/base/transcription.py

@@ -0,0 +1,13 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class TranscriptionSegment:
+    start: float
+    end: float
+    text: str
+
+
+@dataclass
+class Transcription:
+    segments: list[TranscriptionSegment]

{videopython-0.3.0 → videopython-0.4.0}/src/videopython/base/transforms.py

@@ -1,7 +1,6 @@
 from abc import ABC, abstractmethod
 from enum import Enum
 from multiprocessing import Pool
-from typing import Literal
 
 import cv2
 import numpy as np
@@ -154,7 +153,6 @@ class CropMode(Enum):
 
 
 class Crop(Transformation):
-
     def __init__(self, width: int, height: int, mode: CropMode = CropMode.CENTER):
         self.width = width
         self.height = height