import gradio as gr
import cv2
import requests
import os

def download_video(url):
    # Set up headers to mimic a browser request
    try:
        # Send a GET request to the URL
        response = requests.get(url, stream=True)

        # Check if the request was successful (status code 200)
        if response.status_code == 200:
            # Define the file path for saving the video; filename can be adjusted as needed
            filename = 'video.mp4'
            with open(filename, 'wb') as f:
                # Write the content in chunks to handle large files efficiently
                for chunk in response.iter_content(chunk_size=8192):
                    if chunk:
                        f.write(chunk)
                        # Optional: Add progress indicator or logging here
            print(f"Video saved as {filename}")
            return filename
        else:
            print(f"Download failed with status code {response.status_code}")

    except requests.exceptions.RequestException as e:
        print(f"An error occurred: {e}")
    


    
def extract_frames(video_path):
    video_path = download_video(video_path)
    # Open the video file
    cap = cv2.VideoCapture(video_path)
    if not cap.isOpened():
        # If video cannot be opened, return four None values
        return [None] * 4
    
    # Get the total number of frames
    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
    if total_frames == 0:
        # If video has no frames, release and return None values
        cap.release()
        return [None] * 4
    
    # Calculate frame indices at 0%, 25%, 50%, and 75%
    indices = [int(total_frames * i / 4) for i in range(4)]
    frames = []
    
    # Extract frames at the calculated indices
    for idx in indices:
        cap.set(cv2.CAP_PROP_POS_FRAMES, idx)
        ret, frame = cap.read()
        if ret:
            # Convert from BGR (OpenCV format) to RGB (Gradio format)
            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
            frames.append(frame)
        else:
            # If frame extraction fails, append None
            frames.append(None)
    
    # Release the video capture object
    cap.release()
    
    # Return the four frames as a tuple for Gradio outputs
    return tuple(frames)

# Create the Gradio interface
iface = gr.Interface(
    fn=extract_frames,
    inputs=gr.Text(label="Upload a video"),
    outputs=[gr.Image(label=f"Frame at {i*25}%") for i in range(4)],
    title="Video Frame Extractor",
    description="Upload a video to extract four frames from different parts of the video."
)

# Launch the web app
iface.launch()