Update chatgpt-ad-maker.py

chatgpt-ad-maker.py

@@ -49,9 +49,10 @@ def process_video(video_path, dot_size=10, spacing=2, invert=False, fps=30):
     orig_fps = int(cap.get(cv2.CAP_PROP_FPS))
     orig_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     orig_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
     
-    # Use user-defined FPS or original FPS if not provided
-    output_fps = fps if fps > 0 else orig_fps
+    # Ensure output FPS does not exceed input FPS
+    output_fps = min(fps, orig_fps)
     
     max_height = 720
     if orig_height > max_height:
@@ -71,11 +72,28 @@ def process_video(video_path, dot_size=10, spacing=2, invert=False, fps=30):
     out = cv2.VideoWriter(output_path, fourcc, output_fps, (frame_width, frame_height), True)
 
     try:
+        frame_counter = 0
         while cap.isOpened():
             ret, frame = cap.read()
             if not ret:
                 break
             
+            # Skip or duplicate frames based on FPS ratio
+            if output_fps < orig_fps:
+                # Skip frames to match the output FPS
+                if frame_counter % (orig_fps // output_fps) != 0:
+                    frame_counter += 1
+                    continue
+            elif output_fps > orig_fps:
+                # Duplicate frames to match the output FPS
+                for _ in range(output_fps // orig_fps):
+                    if frame.shape[0] != frame_height:
+                        frame = cv2.resize(frame, (frame_width, frame_height))
+                    frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+                    dotted_frame = create_dot_effect(frame_rgb, dot_size, spacing, invert)
+                    dotted_frame_bgr = cv2.cvtColor(dotted_frame, cv2.COLOR_GRAY2BGR)
+                    out.write(dotted_frame_bgr)
+            
             if frame.shape[0] != frame_height:
                 frame = cv2.resize(frame, (frame_width, frame_height))
             
@@ -83,6 +101,8 @@ def process_video(video_path, dot_size=10, spacing=2, invert=False, fps=30):
             dotted_frame = create_dot_effect(frame_rgb, dot_size, spacing, invert)
             dotted_frame_bgr = cv2.cvtColor(dotted_frame, cv2.COLOR_GRAY2BGR)
             out.write(dotted_frame_bgr)
+            
+            frame_counter += 1
 
     finally:
         cap.release()
@@ -90,6 +110,14 @@ def process_video(video_path, dot_size=10, spacing=2, invert=False, fps=30):
     
     return output_path
 
+def get_video_fps(video_path):
+    cap = cv2.VideoCapture(video_path)
+    if not cap.isOpened():
+        return 30  # Default FPS if video cannot be opened
+    fps = int(cap.get(cv2.CAP_PROP_FPS))
+    cap.release()
+    return fps
+
 with gr.Blocks(title="Dot Effect Generator") as iface:
     gr.Markdown("# 🎨 Dot Effect Generator")
     gr.Markdown("Transform media into stylized dot patterns")
@@ -115,6 +143,16 @@ with gr.Blocks(title="Dot Effect Generator") as iface:
         with gr.Row():
             vid_fps = gr.Slider(1, 60, 30, step=1, label="Output FPS")
         vid_btn = gr.Button("Process Video", variant="primary")
+        
+        # Update FPS slider max value based on input video
+        def update_fps_slider(video_path):
+            if video_path is None:
+                return gr.Slider.update(maximum=60, value=30)  # Default if no video
+            fps = get_video_fps(video_path)
+            return gr.Slider.update(maximum=fps, value=min(fps, 30))
+        
+        vid_input.change(update_fps_slider, vid_input, vid_fps)
+        
         vid_btn.click(process_video, [vid_input, vid_dot_size, vid_spacing, vid_invert, vid_fps], vid_output)
 
 if __name__ == "__main__":