Update app.py

app.py

@@ -20,7 +20,7 @@ class MediaDownloader:
 
     def create_directories(self):
         self.image_path = os.path.join(self.temp_dir, "images")
-        self.video_path = os.path.join(self.temp_dir, "videos")  # Fixed this line
+        self.video_path = os.path.join(self.temp_dir, "videos")
         os.makedirs(self.image_path, exist_ok=True)
         os.makedirs(self.video_path, exist_ok=True)
 
@@ -30,7 +30,31 @@ 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',
@@ -46,34 +70,11 @@ class MediaDownloader:
                 'Accept-Language': 'en-US,en;q=0.5',
                 'Accept-Encoding': 'gzip, deflate, br',
                 'Connection': 'keep-alive',
-                'Upgrade-Insecure-Requests': '1',
-                'Sec-Fetch-Dest': 'document',
-                'Sec-Fetch-Mode': 'navigate',
-                'Sec-Fetch-Site': 'none',
-                'Sec-Fetch-User': '?1'
+                'Upgrade-Insecure-Requests': '1'
             }
         }
 
-        if 'tiktok.com' in url:
-            ydl_opts.update({
-                'format': 'best',
-                'force_generic_extractor': False,
-                'extract_flat': False,
-                'quiet': False,
-                'no_warnings': False,
-                'socket_timeout': 60,
-                'retries': 10,
-                'cookiefile': None,
-                'extractor_args': {
-                    'tiktok': {
-                        'embed_api': True,
-                        'api_hostname': 'api16-normal-c-useast1a.tiktokv.com',
-                        'download_api': True,
-                        'allow_redirects': True
-                    }
-                }
-            })
-        elif 'instagram.com' in url:
+        if 'instagram.com' in url:
             ydl_opts.update({
                 'format': 'best',
                 'extract_flat': True,
@@ -116,10 +117,7 @@ class MediaDownloader:
                 'Accept': 'image/webp,*/*',
                 'Accept-Language': 'en-US,en;q=0.5',
                 'Accept-Encoding': 'gzip, deflate, br',
-                'Connection': 'keep-alive',
-                'Sec-Fetch-Dest': 'image',
-                'Sec-Fetch-Mode': 'no-cors',
-                'Sec-Fetch-Site': 'cross-site'
+                'Connection': 'keep-alive'
             }
 
             for img_url in j.urls: