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):
# Note: replace jpwfm2z1g587733t with your link (see your portal)
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)