Update app.py

app.py

@@ -385,81 +385,58 @@ async def jio_download(request: JioDownloadRequest):
 
 
 
-def process_url(title: str) -> str:
-    try:
-        # Step 1: Download the URL
-        api_url = f"https://www.saveporn.net/download/{title}/"
-        session = cloudscraper.create_scraper()
-                
-        # Use the correct API endpoint
-        response = session.get(api_url)
-        logger.info(response)
-        
-        # Check if the request was successful
-        response.raise_for_status()
-        soup = BeautifulSoup(response.text, 'html.parser')
-        table = soup.find('table')
-        data = []
-        if table:
-            rows = table.find_all('tr')
-            for row in rows:
-                cols = row.find_all('td')
-                row_data = {
-                    "format_id": cols[0].get_text() if len(cols) > 0 else "",
-                    "format": cols[1].get_text() if len(cols) > 1 else "",
-                    "url": cols[2].find('a')['href'] if len(cols) > 2 and cols[2].find('a') else ""
-                }
-                data.append(row_data)
-        return data
-    except requests.exceptions.RequestException as e:
-        # Handle network-related errors
-        raise Exception(f"Failed to process URL: {str(e)}")
-    except Exception as e:
-        # Handle any other unexpected errors
-        raise Exception(f"An unexpected error occurred: {str(e)}")
+
+
 
 
 
 
 EXTRACT_API = os.getenv("EXTRACT_API")
+ALT_API = os.getenv("ALT_API")
+
 
 def extract_video_info(video_url: str) -> str:
-    api_url = f'{EXTRACT_API}?url={video_url}'
-    logger.info(api_url)
-    session = cloudscraper.create_scraper()
-    try:
-        response = session.get(api_url, timeout=20)
-        
-        if response.status_code == 200:
-            json_response = response.json()
-            result = []
-            # 检查 formats 列表是否存在且不为空
-            if 'formats' in json_response:
-                for format_item in json_response['formats']:
-                    format_url = format_item.get('url')
-                    format_id = format_item.get('format_id')
-                    p_cookies = format_item.get('cookies')
-                    if format_id and format_url:
-                        result.append({
-                            "url": format_url,
-                            "format_id": format_id,
-                            "cookies": p_cookies
-                        })
-                title = json_response.get('title')
-                logger.info(title)
-                if "pornhub.com" in video_url:
-                    p_result = [item for item in result if 'hls' in item['format_id']]
-                    return p_result
+    api_urls = [f'{EXTRACT_API}?url={video_url}', f'{ALT_API}?url={video_url}']
+    for api_url in api_urls:
+        logger.info(api_url)
+        session = cloudscraper.create_scraper()
+        try:
+            response = session.get(api_url, timeout=20)
+
+            if response.status_code == 200:
+                json_response = response.json()
+                result = []
+                if 'formats' in json_response:
+                    for format_item in json_response['formats']:
+                        format_url = format_item.get('url')
+                        format_id = format_item.get('format_id')
+                        p_cookies = format_item.get('cookies')
+                        if format_id and format_url:
+                            result.append({
+                                "url": format_url,
+                                "format_id": format_id,
+                                "cookies": p_cookies
+                            })
+                    title = json_response.get('title')
+                    logger.info(title)
+                    if "pornhub.com" in video_url:
+                        p_result = [item for item in result if 'hls' in item['format_id']]
+                        return p_result
+                    else:
+                        if len(result) == 1:
+                            new_item = {
+                                "format_id": "This is Fake, Don't Choose This One",
+                                "url": "none"
+                            }
+                            result.append(new_item)
+                        return result
                 else:
-                    new_result = [item for item in result if 'hls' not in item['format_id']]
-                    return new_result
+                    return {"error": "No formats available"}
             else:
-                return {"error": "No formats available"}
-        else:
-            return {"error": f"Request failed with status code {response.status_code}, API: {api_url}"}
-    except Exception as e:
-        logger.error(f"An error occurred: {e}")
-        return {"error": str(e)}
+                logger.warning(f"Request failed with status code {response.status_code}, API: {api_url}")
+        except Exception as e:
+            logger.error(f"An error occurred: {e}")
+    return {"error": "Both APIs failed to provide valid results."}
 
 
 @app.post("/test")