Update app.py

app.py

@@ -29,54 +29,7 @@ def get_frame_count_in_duration(filepath):
     video.release()
     return gr.update(maximum=frame_count)
 
-# Function to process a video chunk and return the processed video
-def process_video_chunk(chunk, index):
-    # Count the frame number of the video chunk
-    frame_count = len(chunk)
 
-    # Export the processed chunk to a file with the index in the file name
-    processed_chunk_filename = f'processed_chunk_{index}.mp4'
-    processed_chunk.write_videofile(processed_chunk_filename)
-
-    return processed_chunk_filename, frame_count
-
-# Function to split video into chunks
-def split_video_into_chunks(video_path, chunk_size):
-    # Load the video clip
-    video = VideoFileClip(video_path)
-
-    # Calculate the total number of frames
-    total_frames = int(video.duration * video.fps)
-
-    # Calculate the number of chunks needed
-    num_chunks = math.ceil(total_frames / chunk_size)
-
-    # Create a list to store the chunks and their frame counts
-    chunks = []
-
-    # Split the video into chunks
-    for i in range(num_chunks):
-        # Calculate the start and end frame for the chunk
-        start_frame = i * chunk_size
-        end_frame = min((i + 1) * chunk_size, total_frames)
-
-        # Extract the chunk from the video
-        chunk = video.subclip(start_frame / video.fps, end_frame / video.fps)
-
-        # Calculate the frame count for the chunk
-        frame_count = end_frame - start_frame
-
-        # Add the chunk and its frame count to the list
-        chunks.append((chunk, frame_count))
-
-    # If the last chunk is smaller than the chunk size
-    if len(chunks) > 0 and chunks[-1][1] < chunk_size:
-        # Adjust the end frame of the last chunk to the total frames
-        last_chunk = chunks[-1][0]
-        last_frame_count = chunks[-1][1]
-        chunks[-1] = (video.subclip(last_chunk.t_start, video.duration), last_frame_count)
-
-    return chunks
 
 
 
@@ -87,52 +40,23 @@ def run_inference(prompt, video_path, condition, video_length):
     os.makedirs(output_path, exist_ok=True)
 
     # Construct the final video path
-    video_path_output = 'final_video.mp4'
+    video_path_output = os.path.join(output_path, f"{prompt}.mp4"
 
     # Check if the file already exists
     if os.path.exists(video_path_output):
         # Delete the existing file
         os.remove(video_path_output)
 
-    # Specify the path to your video file
-    video_path = video_path
-
-    # Specify the maximum number of frames per chunk
-    chunk_size = 12
-
-    # Split the video into chunks
-    video_chunks = split_video_into_chunks(video_path, chunk_size)
-
-    # Process each chunk and store the processed chunk filenames and frame counts
-    processed_chunk_info = []
-    for i, (chunk, _) in enumerate(video_chunks):
-        processed_chunk_filename, frame_count = process_video_chunk(chunk, i)
-        # Count the frame number of the video chunk
-        #frame_count = len(chunk)
-        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{chunk}' --output_path '{output_path}' --video_length {frame_count} --smoother_steps 19 20"
-        subprocess.run(command, shell=True)
-
-        # Construct the video path
-        video_path_output = os.path.join(output_path, f"{prompt}_{i}.mp4")
-
-        processed_chunk_filename = video_path_output
-        #processed_chunk_filenames.append(processed_chunk_filename)
-        processed_chunk_info.append((processed_chunk_filename, frame_count))
-
-    # Load the processed video chunks
-    processed_chunks = [VideoFileClip(filename) for filename, _ in processed_chunk_info]
-
-    # Concatenate the processed video chunks into a final video
-    final_video = concatenate_videoclips(processed_chunks)
+    
+    command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{chunk}' --output_path '{output_path}' --video_length {video_length} --smoother_steps 19 20"
+    subprocess.run(command, shell=True)
 
-    # Export the final video to a file
-    final_video.write_videofile('final_video.mp4')
+    # Construct the video path
+    video_path_output = os.path.join(output_path, f"{prompt}.mp4")
 
-    # Clean up the temporary processed chunk files (optional)
-    for filename, _ in processed_chunk_info:
-        os.remove(filename)
+        
 
-    return "done", 'final_video.mp4' 
+    return "done", video_path_output 
 
 
 with gr.Blocks() as demo: