Video python snippets

Get up to speed with the video API

Generate 20 captions

import requests
import json

X_RapidAPI_Key = '4116843d07msh704be0dfa337826p108b3fj43524'
X_RapidAPI_Host = 'colorize-video.p.rapidapi.com'

headers = {
    'X-RapidAPI-Key': '4116843d07msh704be0dfa337826p108b3fjsn7ab402961f9f',
    'X-RapidAPI-Host': 'colorize-video.p.rapidapi.com',
}

url = "https://" + X_RapidAPI_Host + "/generate_video_frame_prompt"

files = {
    'image': ('test.jpg', open('./test.jpg', 'rb'), 'image/jpeg')   
}

data = {
  "standard_filter_id": "0", # 1-20, 0 is disabled 
  "artistic_filter_id": "0", # 1-100, 0 is disabled
  "all_standard_filters": "true", # return all 20 prompts
  "all_artistic_filters": "false", # return all 100 prompts
  "raw_captions": "false" # remove pre and post fix to the caption
}

response = requests.post(url, files=files, data=data, headers=headers)
prompts = data = json.loads(response.text)

Create 20 versions of one frame

import requests

X_RapidAPI_Key = '4116843d07msh704be0dfa337826p108b3fj43524'
X_RapidAPI_Host = 'colorize-video.p.rapidapi.com'

headers = {
    'X-RapidAPI-Key': '4116843d07msh704be0dfa337826p108b3fjsn7ab402961f9f',
    'X-RapidAPI-Host': 'colorize-video.p.rapidapi.com',
}

url = "https://" + X_RapidAPI_Host + "/colorize_video_frame_with_prompt"

data = {
      "resolution": "sd", #sd, full-hd, or 4k
      "prompt": "", # The prompt to colorize an image
      "auto_color": "false", # Apply automatic color balancing
      "white_balance": "false", # Apply automatic white balancing
      "temperature": "0.0", # -1.0 - 1.0, Apply color temperature
      "saturation": "1.0" # 0.0 - 2.0, Change saturation 
    }

for i, prompt in enumerate(prompts):
    
    files = {
        'image': ('test.jpg', open('./test.jpg', 'rb'), 'image/jpeg')   
    }

    data['prompt'] = prompt

    response = requests.post(url, files=files, data=data, headers=headers)
    f = open(str(i) + ".jpg", 'wb')
    f.write(response.content)
    f.close()

Extract all the frames from a video

import cv2
import os

def extract_images_from_video(video_path, output_folder):
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    # Open the video file
    video = cv2.VideoCapture(video_path)

    # Check if the video was opened successfully
    if not video.isOpened():
        print("Error: Could not open the video file.")
        return

    # Extract images from the video
    frame_number = 0
    while True:
        ret, frame = video.read()

        # Break the loop if we have reached the end of the video
        if not ret:
            break

        # Save the current frame as an image
        output_filename = os.path.join(output_folder, f"frame{frame_number}.png")
        cv2.imwrite(output_filename, frame)
        frame_number += 1

    # Release the video object and close all windows
    video.release()
    cv2.destroyAllWindows()
    print(f"Images extracted and saved in the folder: {output_folder}")

Colorize all the frames with the same prompt

import os
import time
import requests

X_RapidAPI_Key = '4116843d07msh704be0dfa337826p108b3fj43524'
X_RapidAPI_Host = 'colorize-video.p.rapidapi.com'

headers = {
    'X-RapidAPI-Key': '4116843d07msh704be0dfa337826p108b3fjsn7ab402961f9f',
    'X-RapidAPI-Host': 'colorize-video.p.rapidapi.com',
}

def colorize_image(input_file, output_file, prompt):
  
    url = "https://" + X_RapidAPI_Host + "/colorize_video_frame_with_prompt"
    
    files = {
        'image': (input_file, open(input_file, 'rb'), 'image/jpeg')
    }
    data = {
        "resolution": "sd",
        "prompt": prompt,
        "auto_color": "false",
        "white_balance": "false",
        "temperature": "0.0",
        "saturation": "1.0"
    }
    response = requests.post(url, files=files, data=data, headers=headers)
    
    with open(output_file, 'wb') as f:
        f.write(response.content)

def process_files(input_folder, output_folder, prompt):
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    for filename in os.listdir(input_folder):
        input_file = os.path.join(input_folder, filename)
        output_file = os.path.join(output_folder, f'colorized_{filename}')
        success = False
        retries = 0

        while not success and retries < 10:
            try:
                colorize_image(input_file, output_file, prompt)
                success = True
            except Exception as e:
                print(f"Error processing {input_file}: {e}")
                time.sleep(60)
                retries += 1

input_folder = './input'
output_folder = './output'
prompt = "Your selected prompt"
process_files(input_folder, output_folder, prompt)

Colorize all the images with the same reference image

import os
import time
import requests
from pathlib import Path

X_RapidAPI_Key = '4116843d07msh704be0dfa337826p108b3fj43524'
X_RapidAPI_Host = 'colorize-video.p.rapidapi.com'

headers = {
    'X-RapidAPI-Key': '4116843d07msh704be0dfa337826p108b3fjsn7ab402961f9f',
    'X-RapidAPI-Host': 'colorize-video.p.rapidapi.com',
}

def colorize_files(input_folder, output_folder, ref_image_path):
    
    url = "https://" + X_RapidAPI_Host + "/colorize_video_frame_with_prompt"
   
    data = {
        "resolution": "sd",
        "auto_color": "false",
        "white_balance": "false",
        "temperature": "0.0",
        "saturation": "1.0"
    }
    
    Path(output_folder).mkdir(parents=True, exist_ok=True)
    
    for file_name in os.listdir(input_folder):
        if file_name.lower().endswith(('.jpg', '.jpeg')):
            input_file_path = os.path.join(input_folder, file_name)
            output_file_path = os.path.join(output_folder, f'colorized_{file_name}')
            
            for _ in range(10):
                try:
                    files = {
                        'image': (file_name, open(input_file_path, 'rb'), 'image/jpeg'),
                        'image_ref': ('ref_image.jpg', open(ref_image_path, 'rb'), 'image/jpeg')
                    }
                    response = requests.post(url, files=files, data=data, headers=headers)
                    response.raise_for_status()
                    
                    with open(output_file_path, 'wb') as f:
                        f.write(response.content)
                    
                    print(f"Colorized {file_name} and saved as {output_file_path}")
                    break
                except Exception as e:
                    print(f"Error while processing {file_name}: {e}")
                    time.sleep(60)

input_folder = './input_folder'
output_folder = './output_folder'
ref_image_path = 'ref_image.jpg'

colorize_files(input_folder, output_folder, ref_image_path)

Use the previous frame to colorize one after each other

import os
import time
import requests
from pathlib import Path

X_RapidAPI_Key = '4116843d07msh704be0dfa337826p108b3fj43524'
X_RapidAPI_Host = 'colorize-video.p.rapidapi.com'

headers = {
    'X-RapidAPI-Key': '4116843d07msh704be0dfa337826p108b3fjsn7ab402961f9f',
    'X-RapidAPI-Host': 'colorize-video.p.rapidapi.com',
}

def colorize_images(folder_path):
    
    url = "https://" + X_RapidAPI_Host + "/colorize_video_frame_with_prompt"
   
    data = {
        "resolution": "sd",
        "auto_color": "false",
        "white_balance": "false",
        "temperature": "0.0",
        "saturation": "1.0"
    }
    
    output_folder = Path(folder_path) / "colorized"
    output_folder.mkdir(parents=True, exist_ok=True)

    reference_image = Path(folder_path) / "the_first_reference.jpg"

    for file in os.listdir(folder_path):
        if file.endswith(".jpg") or file.endswith(".jpeg"):
            input_file = Path(folder_path) / file

            retries = 10
            while retries > 0:
                try:
                    files = {
                        'image': (file, open(input_file, 'rb'), 'image/jpeg'),
                        'image_ref': ('ref_image.jpg', open(reference_image, 'rb'), 'image/jpeg')
                    }

                    response = requests.post(url, files=files, data=data, headers=headers)
                    response.raise_for_status()

                    colorized_file = output_folder / f'colorized_{file}'
                    with open(colorized_file, 'wb') as f:
                        f.write(response.content)

                    reference_image = colorized_file
                    break

                except requests.exceptions.RequestException:
                    retries -= 1
                    time.sleep(60)
                    
colorize_images("path/to/your/folder")