Add python input option - yay!

.gitignore

@@ -167,3 +167,5 @@ cython_debug/
 #.idea/
 
 *.DS_STORE
+spotify.rc
+spotify/

Pipfile

@@ -0,0 +1,11 @@
+[[source]]
+url = "https://pypi.org/simple"
+verify_ssl = true
+name = "pypi"
+
+[packages]
+
+[dev-packages]
+
+[requires]
+python_version = "3.10"

app.py

@@ -1,19 +1,60 @@
 import gradio as gr
-import torch
 import os
-from torch import autocast
-from diffusers import StableDiffusionPipeline
 from moviepy.editor import AudioFileClip, ImageClip
 from pathlib import Path
-
+import subprocess
+import uuid
+import shutil
 
 output_dir = Path("temp/")
 output_dir.mkdir(exist_ok=True, parents=True)
 os.chdir(output_dir)
 
 
-def process_inputs(prompt, audio):
+class SpotifyApi:
+    spotify_directory = Path(output_dir / "spotify")
+    final_directory = output_dir
+
+    def __init__(self):
+        self.setup_spotipy
+
+    def setup_spotipy(self):
+        # Check if the credentials file exists
+        if not os.path.exists("spotif.rc"):
+            with open("spotify.rc", "w") as f:
+                f.write(
+                    f"{os.environ['SPOTIFY_USERNAME']} {os.environ['SPOTIFY_PASSWORD']}"
+                )
+            subprocess.call(["spodcast", "-l", "spotify.rc"])
+        else:
+            pass
+
+    def download_episode(self, episode_url):
+        # 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()
+
+    def get_final_mp3(self):
+        # 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 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)
+                    shutil.rmtree(Path(self.spotify_directory / self.foldername))
+                    return final_mp3.as_posix()
+
+
+def process_inputs(prompt, audio, spotify_url):
     image = get_stable_diffusion_image(prompt)
+    if spotify_url:
+        spotify = SpotifyApi()
+        audio = spotify.download_episode(spotify_url)
     video = add_static_image_to_audio(image, audio)
     return video
 
@@ -47,6 +88,14 @@ def get_stable_diffusion_image(prompt):
 
 
 iface = gr.Interface(
-    fn=process_inputs, inputs=["text", gr.Audio(type="filepath")], outputs="video"
+    fn=process_inputs,
+    inputs=[
+        gr.Textbox(label="Describe your podcast clip"),
+        gr.Audio(type="filepath", label="Upload an mp3"),
+        gr.Textbox(label="Or Paste a spotify episode link"),
+    ],
+    outputs="video",
 )
+
+
 iface.launch()

requirements.txt

@@ -4,4 +4,5 @@ nvidia-ml-py3
 ftfy
 --extra-index-url https://download.pytorch.org/whl/cu113 torch
 moviepy
-gradio
+gradio
+spodcast