Update app.py

app.py

@@ -30,38 +30,14 @@ class MediaDownloader:
         except Exception as e:
             print(f"Cleanup error: {str(e)}")
 
-    def download_tiktok(self, url):
-        try:
-            api_url = "https://ssstik.io/abc?url=" + url
-            headers = {
-                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36"
-            }
-            response = requests.get(api_url, headers=headers)
-            if response.status_code == 200:
-                video_url = response.json().get("video_url")
-                if video_url:
-                    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S_%f")
-                    output_path = os.path.join(self.video_path, f"tiktok_{timestamp}.mp4")
-                    video_response = requests.get(video_url, headers=headers)
-                    with open(output_path, "wb") as f:
-                        f.write(video_response.content)
-                    return output_path
-            return None
-        except Exception as e:
-            print(f"TikTok download error: {str(e)}")
-            return None
-
     def download_video(self, url):
-        if 'tiktok.com' in url:
-            return self.download_tiktok(url)
-
         output_template = os.path.join(self.video_path, '%(title)s.%(ext)s')
         ydl_opts = {
-            'format': 'bestvideo+bestaudio/best',
+            'format': 'best',
             'outtmpl': output_template,
             'ignoreerrors': True,
             'no_warnings': True,
-            'quiet': False,
+            'quiet': True,
             'extract_flat': False,
             'socket_timeout': 30,
             'http_headers': {
@@ -74,7 +50,16 @@ class MediaDownloader:
             }
         }
 
-        if 'instagram.com' in url:
+        if 'tiktok.com' in url:
+            ydl_opts.update({
+                'format': 'best',
+                'quiet': False,
+                'no_warnings': False,
+                'extractor_args': {'TikTok': {'download_api': True}},
+                'force_generic_extractor': False,
+                'cookiesfrombrowser': None
+            })
+        elif 'instagram.com' in url:
             ydl_opts.update({
                 'format': 'best',
                 'extract_flat': True,
@@ -155,28 +140,21 @@ class MediaDownloader:
         if not url.startswith(('http://', 'https://')):
             url = 'https://' + url
 
-        if 'instagram.com' in url:
-            video_path = self.download_video(url)
-            if video_path:
-                return [video_path]
-            image_paths = self.download_images(url)
-            if image_paths:
-                return image_paths
-        else:
-            video_path = self.download_video(url)
-            if video_path:
-                return [video_path]
-            image_paths = self.download_images(url)
-            if image_paths:
-                return image_paths
+        video_path = self.download_video(url)
+        if video_path:
+            return [video_path]
+
+        image_paths = self.download_images(url)
+        if image_paths:
+            return image_paths
 
         return None
 
 @app.route('/')
 def home():
     return """
-    <h1>download</h1>
-    <p>Use: /download?url=url</p>
+    <h1>Social Media Downloader API</h1>
+    <p>Use: /download?url=YOUR_URL_HERE</p>
     """
 
 @app.route('/download')
@@ -184,7 +162,7 @@ def download():
     try:
         url = request.args.get('url')
         if not url:
-            return "No URL", 400
+            return "No URL provided", 400
 
         downloader = MediaDownloader()
 
@@ -220,7 +198,7 @@ def download():
                     return response
             else:
                 downloader.cleanup()
-                return "Failed to download", 400
+                return "Failed to download media", 400
 
         except Exception as e:
             downloader.cleanup()