Update app.py

app.py

@@ -7,6 +7,7 @@ from fastapi.concurrency import run_in_threadpool
 import yt_dlp
 import ffmpeg
 import urllib.parse
+from urllib.parse import urlparse
 import os
 from datetime import datetime, timedelta
 import schedule
@@ -549,6 +550,34 @@ def get_youtube_download_url_from_apis(video_url: str) -> str:
 
     logger.info("Both APIs failed to retrieve a download URL.")
     return None
+    
+reads_api = os.getenv("reads_api")
+def call_extract_endpoint(post_url: str) -> Dict[str, Any]:
+   
+    try:
+        response = requests.get(f"{reads_api}/extract", params={"url": post_url})
+        response.raise_for_status()
+        return response.json()
+    except requests.RequestException as e:
+        raise requests.RequestException(f"Failed to call extract endpoint: {str(e)}")
+    except ValueError as e:
+        raise ValueError(f"Invalid response format: {str(e)}")
+
+        
+def is_threads_url(url: str) -> bool:
+    """Validate if URL is a valid Threads URL"""
+    try:
+        parsed = urlparse(url)
+        logger.info(parsed)
+        # Check if netloc matches known Threads domains
+        if parsed.netloc not in ['threads.net', 'www.threads.net', 'threads.com', 'www.threads.com']:
+            return False
+        # Check if path contains /post/ or /t/ after a username
+        if '/post/' in parsed.path or '/t/' in parsed.path:
+            return True
+        return False
+    except Exception:
+        return False
 
 @app.post("/multi")
 async def multi_download(request: Request):
@@ -593,9 +622,19 @@ async def multi_download(request: Request):
             error_message = "Quality above 720p is for Premium Members Only. Please check the URL for more information."
             help_url = "https://chrunos.com/premium-shortcuts/"  # Replace with your actual URL
             return {"error": error_message, "url": help_url}
-    
-    
-    is_youtube_url = re.search(r'(youtube\.com|youtu\.be)', video_url) is not None
+    if is_threads_url(video_url):
+        return call_extract_endpoint(video_url)
+    else:
+        dl_url = await get_track_download_url(video_url, quality)
+        if dl_url:
+            remaining_requests_value = rate_limiter.max_requests - rate_limiter.get_current_count(user_ip)
+            dl_url["requests_remaining"] = remaining_requests_value
+            return dl_url
+        else:
+            return {
+                "error": "Failed to Fetch the video."
+                }
+    #is_youtube_url = re.search(r'(youtube\.com|youtu\.be)', video_url) is not None
     
     '''if is_youtube_url:
         dl_url = get_youtube_download_url_from_apis(video_url)
@@ -616,15 +655,7 @@ async def multi_download(request: Request):
 
     else:
 '''
-    dl_url = await get_track_download_url(video_url, quality)
-    if dl_url:
-        remaining_requests_value = rate_limiter.max_requests - rate_limiter.get_current_count(user_ip)
-        dl_url["requests_remaining"] = remaining_requests_value
-        return dl_url
-    else:
-        return {
-            "error": "Failed to Fetch the video."
-            }
+