sync ms

app.py

@@ -5,10 +5,18 @@ import requests
 import gradio as gr
 from bs4 import BeautifulSoup
 from pydub import AudioSegment
+from datetime import datetime, timedelta
 
 TMP_DIR = "./__pycache__"
 
 
+def get_prev_day(date_str):
+    date_format = "%Y/%m/%d"
+    date_obj = datetime.strptime(date_str, date_format)
+    previous_day = date_obj - timedelta(days=1)
+    return previous_day.strftime(date_format)
+
+
 def remove_end_seconds(input_file: str, output_file: str, seconds: float):
     audio = AudioSegment.from_file(input_file)
     remove_ms = seconds * 1000
@@ -43,9 +51,16 @@ def download_mp3(url: str, local_filename: str):
                 f.write(response.content)
 
             print(f"Successfully downloaded: {local_filename}")
+            remove_end_seconds(local_filename, local_filename, 3.1)
             return True
 
         else:
+            if response.status_code == 404:
+                bad_date = "/".join(url.split("/audio/")[-1].split("/")[:-1])
+                fixed_date = get_prev_day(bad_date)
+                fixed_url = url.replace(bad_date, fixed_date)
+                return download_mp3(fixed_url, local_filename)
+
             print(f"Error: {response.status_code}, {response.text}")
             return False
 
@@ -85,7 +100,6 @@ def infer(page_url: str, date: str):
     outpath = f"{TMP_DIR}/{sound_id}.mp3"
     create_dir()
     if download_mp3(mp3_url, outpath):
-        remove_end_seconds(outpath, outpath, 3.1)
         return outpath
 
     else: