Add start time and end time

app.py

@@ -1,11 +1,11 @@
-import gradio as gr
 import os
-from moviepy.editor import AudioFileClip, ImageClip
-from pathlib import Path
+import shutil
 import subprocess
 import uuid
-import shutil
+from pathlib import Path
 
+import gradio as gr
+from moviepy.editor import AudioFileClip, ImageClip
 
 output_dir = Path("temp/").absolute()
 output_dir.mkdir(exist_ok=True, parents=True)
@@ -14,6 +14,8 @@ os.chdir(
     output_dir
 )  # change working directory to output_dir because the hf spaces model has no option to specify output directory ¯\_(ツ)_/¯
 
+# TODO: Add an if statement that checks if a gpu is available, if one is then do weird stable diffusion stuff, if one isn't, then just use the regular hugging face api
+
 
 class SpotifyApi:
     spotify_directory = Path("spotify")
@@ -62,16 +64,26 @@ class SpotifyApi:
                     return final_mp3.as_posix()
 
 
-def process_inputs(prompt, audio, spotify_url) -> str:
+class AudioInput:
+    def __init__(self, path: str, start_time: int, run_for: int):
+        self.path = path
+        self.start_time = start_time
+        self.run_for = run_for
+
+
+def process_inputs(
+    prompt: str, audio_path: str, spotify_url: str, start_time: int, run_for: int
+) -> str:
+    audio_input = AudioInput(audio_path, start_time, run_for)
     if spotify_url:
         spotify = SpotifyApi()
-        audio = spotify.download_episode(spotify_url)
+        audio_input.path = spotify.download_episode(spotify_url)
     image = get_stable_diffusion_image(prompt)
-    video = add_static_image_to_audio(image, audio)
+    video = add_static_image_to_audio(image, audio_input)
     return video
 
 
-def add_static_image_to_audio(image, audio) -> str:
+def add_static_image_to_audio(image, audio_input) -> str:
     """Create and save a video file to `output_path` after
     combining a static image that is located in `image_path`
     with an audio file in `audio_path`"""
@@ -80,10 +92,16 @@ def add_static_image_to_audio(image, audio) -> str:
     vid_output_dir = Path(output_dir / foldername)
     vid_output_dir.mkdir(exist_ok=True, parents=True)
     os.chdir(vid_output_dir)
-    audio_clip = AudioFileClip(audio)
+    audio_clip = AudioFileClip(audio_input.path)
+    # Make the audio clip start at the specified time and set the duration to the specified duration
+    audio_clip = audio_clip.subclip(
+        audio_input.start_time, audio_input.start_time + audio_input.run_for
+    )
     image_clip = ImageClip(image)
     video_clip = image_clip.set_audio(audio_clip)
-    video_clip.duration = audio_clip.duration
+    video_clip.duration = (
+        audio_clip.duration
+    )  # The duration here is the cut duration from above
     video_clip.fps = 1
     path = Path("output.mp4").as_posix()
     video_clip.write_videofile(path)
@@ -102,6 +120,8 @@ iface = gr.Interface(
         gr.Textbox(label="Describe your podcast clip"),
         gr.Audio(type="filepath", label="Upload an mp3"),
         gr.Textbox(label="Or Paste a spotify episode link"),
+        gr.Number(label="Start time (in seconds)"),
+        gr.Number(label="Run for (in seconds)"),
     ],
     outputs="video",
 )