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)