Refactor spotify api to include image downloads

app.py

@@ -8,16 +8,28 @@ from pathlib import Path
 import gradio as gr
 from moviepy.editor import AudioFileClip
 
+
+import opengraph
+import pathlib
+
+import requests
+
 output_dir = Path("temp/").absolute()
 output_dir.mkdir(exist_ok=True, parents=True)
 
-
 class SpotifyApi:
-    spotify_directory = Path("spotify")
-    final_directory = output_dir
+    spotify_directory = Path(output_dir / "spotify")
 
-    def __init__(self):
+    def __init__(self, url):
         self.setup_spotify()
+        self.url = url
+        self.opengraph = opengraph.OpenGraph(url=url)
+        self.random_string = str(uuid.uuid4())[:8]
+        self.perma_output_path = Path(output_dir / self.random_string)
+        self.temp_output_path = Path(
+            self.spotify_directory / self.random_string
+        ).resolve()
+        self.folder_dir: Path = None
 
     def setup_spotify(self) -> None:
         # Check if the credentials file exists
@@ -28,32 +40,30 @@ class SpotifyApi:
                 )
         subprocess.call(["spodcast", "-l", "spotify.rc"])
 
-    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
+    def download_episode(self) -> str:
+        out_path = self.temp_output_path.resolve()
+        subprocess.call(["spodcast", "--root-path", out_path, self.url])
         mp3_path = self.get_final_mp3()
         assert mp3_path is not None
         return mp3_path
 
+    def download_image(self):
+        image = self.opengraph["image"]
+        r = requests.get(image, allow_redirects=True)
+        path = self.perma_output_path.with_suffix(".jpg").absolute()
+        open(path, "wb").write(r.content)
+
+    def get_title(self) -> str:
+        return self.opengraph["title"]
+
+    # Move output file in the temp mp3 folder to the final output folder that we'll store the video in
     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).resolve()
-        ):
+        for root, dirs, files in os.walk(self.temp_output_path.resolve()):
             for file in files:
                 if file.endswith(".mp3"):
-                    final_mp3 = (
-                        Path(self.final_directory / self.foldername)
-                        .with_suffix(".mp3")
-                        .absolute()
-                    )
+                    final_mp3 = self.perma_output_path.with_suffix(".mp3").absolute()
                     shutil.copy(os.path.join(root, file), final_mp3)
-                    shutil.rmtree(
-                        Path(self.spotify_directory / self.foldername).absolute()
-                    )
+                    shutil.rmtree(self.temp_output_path.absolute())
                     return final_mp3.as_posix()
 
 

requirements.txt

@@ -18,4 +18,5 @@ spodcast
 moviepy
 huggingface-hub
 opencv-python
-image_tools
+image_tools
+podiant-opengraph

test.ipynb

@@ -0,0 +1,90 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "9f0e1e4b-e36a-4dfe-adfd-6f99fb1d39f6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import os\n",
+    "from app import animate_images, AudioInput, SpotifyApi"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "778fa4aa-cc12-4c33-a866-47ddcedcaf5f",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "WARNING:spodcast.podcast:Downloaded Hello Internet: Nine Ladies Dancing\n"
+     ]
+    }
+   ],
+   "source": [
+    "def test_spotify_downloads():\n",
+    "    url = \"https://open.spotify.com/episode/3Ev4jlIjT8IVih1Oxidhqf?si=54a95e8fb07640c5\"\n",
+    "    api = SpotifyApi(url=url)\n",
+    "    api.download_episode()\n",
+    "    api.download_image()\n",
+    "    api.get_title()\n",
+    "\n",
+    "test_spotify_downloads()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "de461d9b-bfbb-4d2d-b35e-fc3889494991",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def get_test_image_paths():\n",
+    "    gallery_dir = \"temp/images\"\n",
+    "    return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)][:2]\n",
+    "\n",
+    "get_test_image_paths()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "99df13fd-bd5d-4964-9300-016fcf7625e6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def get_test_audio_paths():\n",
+    "    audio_path = \"temp/audio.mp3\"\n",
+    "    audio_input = AudioInput(audio_path, 10, 10)\n",
+    "    return audio_input\n",
+    "\n",
+    "get_test_audio_paths()"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.9.13"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

test.py

@@ -1,17 +0,0 @@
-import os
-from app import animate_images, AudioInput
-
-def get_test_image_paths():
-    gallery_dir = "temp/images"
-    return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)][:2]
-    
-def get_test_audio_paths():
-    audio_path = "temp/audio.mp3"
-    audio_input = AudioInput(audio_path, 10, 10)
-    return audio_input
-
-
-image_paths = get_test_image_paths()
-audio_input = get_test_audio_paths()
-
-animate_images(image_paths, audio_input)