Update app.py

app.py

@@ -92,41 +92,30 @@ def export_to_video_file(video_frames, output_video_path=None, fps=10):
 # those are way too slow for a AiTube which needs things to be as fast as possible
 # -----------------------------------------------------------------------------------
 
-def interpolate_video_frames(input_file_path, output_file_path, output_fps=10, desired_duration=1):
-    """
-    Interpolates frames in a video file to adjust frame rate and duration using ffmpeg's minterpolate.
-    
-    Parameters:
-        input_file_path (str): Path to the input video file.
-        output_file_path (str): Path to the output video file.
-        output_fps (int): Target frames per second for the output video.
-        desired_duration (int): Desired duration of the video in seconds.
-    
-    Returns:
-        str: The file path of the modified video.
-    """
-    
-    # Construct the ffmpeg command for interpolation
+def interpolate_video_frames(input_file_path, output_file_path, output_fps=10, desired_duration=1.6, original_duration=1.6):
+    scale_factor = original_duration / desired_duration
+    interpolation_filter = f'minterpolate=fps={output_fps},setpts={scale_factor}*PTS'
+
     cmd = [
         'ffmpeg',
-        '-i', input_file_path,  # input file
-        '-filter:v', f'minterpolate=fps={output_fps}',  # minterpolate filter options
-        '-r', str(output_fps),  # output frame rate
-        output_file_path  # Output file
+        '-i', input_file_path,
+        '-filter:v', interpolation_filter,
+        '-r', str(output_fps),
+        output_file_path
     ]
 
+    # Logging for debugging
     print("output_fps:", output_fps)
     print("desired_duration:", desired_duration)
-    print("cmd:", cmd)    
+    print("original_duration:", original_duration)
+    print("cmd:", cmd)
 
-    # Execute the command
     try:
         subprocess.run(cmd, check=True)
-        print("Video interpolation successful.")
-        return input_file_path
+        return output_file_path
     except subprocess.CalledProcessError as e:
         print("Failed to interpolate video. Error:", e)
-        return output_file_path
+        return input_file_path  # In case of error, return original path
 
 def generate_image(secret_token, prompt, base, width, height, motion, step, desired_duration, desired_fps):
     if secret_token != SECRET_TOKEN:
@@ -269,7 +258,7 @@ with gr.Blocks() as demo:
                     ('8-Step', 8)],
                 value=4,
             )
-            duration_slider = gr.Slider(label="Desired Duration (seconds)", min_value=1, max_value=30, value=1, step=1)
+            duration_slider = gr.Slider(label="Desired Duration (seconds)", min_value=1, max_value=120, value=1.6, step=0.1)
             fps_slider = gr.Slider(label="Desired Frames Per Second", min_value=10, max_value=60, value=10, step=1)
     
             submit = gr.Button()