Add types

app.py

@@ -6,9 +6,11 @@ import subprocess
 import uuid
 import shutil
 
+
 output_dir = Path("temp/")
 output_dir.mkdir(exist_ok=True, parents=True)
-os.chdir(output_dir)
+
+os.chdir(output_dir) # change working directory to output_dir because the hf spaces model has no option to specify output directory ¯\_(ツ)_/¯
 
 
 class SpotifyApi:
@@ -18,7 +20,7 @@ class SpotifyApi:
     def __init__(self):
         self.setup_spotipy
 
-    def setup_spotipy(self):
+    def setup_spotipy(self) -> None:
         # Check if the credentials file exists
         if not os.path.exists("spotif.rc"):
             with open("spotify.rc", "w") as f:
@@ -29,24 +31,28 @@ class SpotifyApi:
         else:
             pass
 
-    def download_episode(self, episode_url):
+    def download_episode(self, episode_url) -> str:
         # Generate a 8 character random string
         foldername = str(uuid.uuid4())[:8]
         out_path = (self.spotify_directory / foldername).resolve()
         subprocess.call(["spodcast", "--root-path", out_path, episode_url])
         self.foldername = foldername
-        return self.get_final_mp3()
+        mp3_path = self.get_final_mp3()
+        assert mp3_path is not None
+        return mp3_path
 
-    def get_final_mp3(self):
+    def get_final_mp3(self) -> str | None:
         # Look in all the subdirectories of spotify for the mp3 file
-        for root, dirs, files in os.walk(Path(self.spotify_directory / self.foldername)):
+        for root, dirs, files in os.walk(
+            Path(self.spotify_directory / self.foldername)
+        ):
             for file in files:
                 if file.endswith(".mp3"):
-                    final_mp3 = Path(self.final_directory / self.foldername).with_suffix(
-                        ".mp3"
-                    )
-                    shutil.copy(os.path.join(root, file), final_mp3)
+                    final_mp3 = Path(
+                        self.final_directory / self.foldername
+                    ).with_suffix(".mp3")
                     shutil.rmtree(Path(self.spotify_directory / self.foldername))
+                    shutil.copy(os.path.join(root, file), final_mp3)
                     return final_mp3.as_posix()
 
 
@@ -59,31 +65,23 @@ def process_inputs(prompt, audio, spotify_url):
     return video
 
 
-def add_static_image_to_audio(image, audio):
+def add_static_image_to_audio(image, audio) -> 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`"""
-    # create the audio clip object
     audio_clip = AudioFileClip(audio)
-    # create the image clip object
     image_clip = ImageClip(image)
-    # use set_audio method from image clip to combine the audio with the image
     video_clip = image_clip.set_audio(audio_clip)
-    # specify the duration of the new clip to be the duration of the audio clip
     video_clip.duration = audio_clip.duration
-    # set the FPS to 1
     video_clip.fps = 1
-    # write the resuling video clip
-    path = "out.mp4"
+    path = Path(output_dir / "output.mp4")
     video_clip.write_videofile(path)
-    return path
+    return path.as_posix()
 
 
-def get_stable_diffusion_image(prompt):
-    path = "temp/image_out.png"
+def get_stable_diffusion_image(prompt) -> str:
     stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
     gallery_dir = stable_diffusion(prompt, fn_index=2)
-    # Rename gallery dir to sdout
     return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)][0]