Create app.py

app.py

@@ -0,0 +1,63 @@
+import zipfile
+import os
+import cv2
+from gradio_client import Client, handle_file
+import shutil
+
+# Function to process the video and apply the color transformation to each frame
+def process_video(input_zip_path, style_image_path, output_video_path):
+    # Initialize the Hugging Face Client for the color transformation API
+    client = Client("Abu1998/Image_Color_Transfer_Video")
+
+    # Prepare the zip file containing images
+    with zipfile.ZipFile(input_zip_path, 'r') as zip_ref:
+        zip_ref.extractall('input_images')  # Extract images to 'input_images' folder
+
+    # Process each image
+    image_files = [f for f in os.listdir('input_images') if f.endswith('.png') or f.endswith('.jpg')]
+    
+    # Prepare output folder
+    output_folder = 'output_images'
+    os.makedirs(output_folder, exist_ok=True)
+
+    # Apply color transformation to each image
+    for image_file in image_files:
+        input_image_path = os.path.join('input_images', image_file)
+        
+        # Use the style image and the current image for transformation
+        result = client.predict(
+            img1=handle_file(style_image_path),
+            img2=handle_file(input_image_path),
+            api_name="/_transfer_style"
+        )
+        
+        # Save the transformed image
+        transformed_image_path = os.path.join(output_folder, f"transformed_{image_file}")
+        with open(transformed_image_path, 'wb') as f:
+            f.write(result[0]['data'])
+
+    # Create a video from the transformed images
+    images = [os.path.join(output_folder, f) for f in sorted(os.listdir(output_folder))]
+    frame = cv2.imread(images[0])
+    height, width, layers = frame.shape
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    video_writer = cv2.VideoWriter(output_video_path, fourcc, 30.0, (width, height))
+
+    for image in images:
+        frame = cv2.imread(image)
+        video_writer.write(frame)
+    
+    video_writer.release()
+    
+    # Clean up temporary files
+    shutil.rmtree('input_images')
+    shutil.rmtree('output_images')
+
+    print(f"Video saved to {output_video_path}")
+
+# Example usage
+input_zip_path = 'path_to_zip_of_images.zip'  # Path to the zip file of images from the video
+style_image_path = 'path_to_style_image.jpg'  # Path to the style image for color transformation
+output_video_path = 'output_video.mp4'  # Path where the final video will be saved
+
+process_video(input_zip_path, style_image_path, output_video_path)