Update chatgpt-ad-maker.py

chatgpt-ad-maker.py

@@ -70,10 +70,10 @@ def process_video(video_path, dot_size=10, spacing=2, invert=False):
         frame_width = int(frame_width * scale)
         frame_height = max_height
     
-    # Create temporary output file
+    # Create temporary output file with MP4V codec
     output_path = "temp_output.mp4"
-    fourcc = cv2.VideoWriter_fourcc(*'avc1')
-    out = cv2.VideoWriter(output_path, fourcc, fps, (frame_width, frame_height), False)
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Changed codec to MP4V
+    out = cv2.VideoWriter(output_path, fourcc, fps, (frame_width, frame_height), True)  # Set isColor=True
 
     try:
         while cap.isOpened():
@@ -88,11 +88,14 @@ def process_video(video_path, dot_size=10, spacing=2, invert=False):
             # Convert BGR to RGB for processing
             frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
             
-            # Apply dot effect
+            # Apply dot effect (returns grayscale)
             dotted_frame = create_dot_effect(frame_rgb, dot_size, spacing, invert)
             
+            # Convert grayscale to BGR for video writer
+            dotted_frame_bgr = cv2.cvtColor(dotted_frame, cv2.COLOR_GRAY2BGR)
+            
             # Write the frame
-            out.write(dotted_frame)
+            out.write(dotted_frame_bgr)
 
     finally:
         # Ensure resources are released
@@ -134,4 +137,4 @@ with gr.Blocks(title="ChatGPT Ad Maker") as iface:
         )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()