Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from speechbrain.inference import EncoderClassifier
 import torch
+import requests
 import subprocess
 import os
 import uuid
@@ -14,16 +15,86 @@ def get_model():
         model = EncoderClassifier.from_hparams("Jzuluaga/accent-id-commonaccent_ecapa")
     return model
 
-def download_video_from_url(url):
-    out_path = f"video_{uuid.uuid4()}.mp4"
-    ydl_opts = {
-        'format': 'bestaudio/best',
-        'outtmpl': out_path,
-        'quiet': True,
+def extract_id_from_url(url):
+    urlRet = url.split("/")[-1]
+    if '?' in urlRet:
+        urlRet = urlRet.split("?")[0]
+    return urlRet
+
+def fetch_loom_download_url(id):
+    response = requests.post(url=f"https://www.loom.com/api/campaigns/sessions/{id}/transcoded-url")
+    if response.status_code == 200:
+        return response.json()["url"]
+    else:
+        print("Error while retrieving response: ", response.status_code)
+        exit
+
+
+
+def download_loom_video(url, filename):
+    headers = {
+        "User-Agent": "Mozilla/5.0"
     }
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        ydl.download([url])
-    return out_path
+
+    try:
+        with requests.get(url, headers=headers, stream=True) as response:
+            response.raise_for_status()
+
+            with open(filename, "wb") as f:
+                for chunk in response.iter_content(chunk_size=8192):
+                    if chunk:
+                        f.write(chunk)
+
+        print(f"Downloaded video to {filename}")
+        return filename
+
+    except requests.exceptions.RequestException as e:
+        print(f"Failed to download Loom video: {e}")
+        return None
+
+def download_direct_mp4(url, filename):
+    try:
+        response = requests.get(url, stream=True)
+        response.raise_for_status()
+        with open(filename, "wb") as f:
+            for chunk in response.iter_content(chunk_size=8192):
+                if chunk:
+                    f.write(chunk)
+        return filename
+    except Exception as e:
+        print(f"Error downloading direct mp4: {e}")
+        return None
+        
+
+def download_video_from_url(url):
+    if "loom.com" in url:
+        video_id = extract_id_from_url(url)
+        print(video_id)
+        direct_url = fetch_loom_download_url(video_id)
+        print(direct_url)
+        filename = f"LoomVideo_{video_id}.mp4"
+        success = download_loom_video(direct_url, filename)
+        print(success)
+        return filename if success else None
+
+    elif url.endswith(".mp4"):
+        filename = f"video_{uuid.uuid4()}.mp4"
+        result = download_direct_mp4(url, filename)
+        return result
+
+    else:
+        # fallback to yt_dlp for youtube, vimeo, etc.
+        out_path = f"video_{uuid.uuid4()}.mp4"
+        ydl_opts = {
+            'format': 'bestaudio/best',
+            'outtmpl': out_path,
+            'quiet': True,
+        }
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            ydl.download([url])
+        return out_path
+
+
 
 def extract_audio(video_file):
     audio_path = f"audio_{uuid.uuid4()}.wav"
@@ -35,6 +106,7 @@ def extract_audio(video_file):
     subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     return audio_path
 
+
 def classify_accent(input_file_or_url):
     model = get_model()