Update app.py

app.py

@@ -8,10 +8,8 @@ from datetime import datetime
 import tempfile
 import shutil
 import mimetypes
-from asgiref.wsgi import WsgiToAsgi
 
 app = Flask(__name__)
-asgi_app = WsgiToAsgi(app)
 
 class MediaDownloader:
     def __init__(self):
@@ -25,10 +23,7 @@ class MediaDownloader:
         os.makedirs(self.video_path, exist_ok=True)
 
     def cleanup(self):
-        try:
-            shutil.rmtree(self.temp_dir)
-        except Exception as e:
-            print(f"Cleanup error: {str(e)}")
+        shutil.rmtree(self.temp_dir)
 
     def download_video(self, url):
         output_template = os.path.join(self.video_path, '%(title)s.%(ext)s')
@@ -39,36 +34,11 @@ class MediaDownloader:
             'no_warnings': True,
             'quiet': True,
             'extract_flat': False,
-            'socket_timeout': 30,
             'http_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',
-                'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
-                'Accept-Language': 'en-US,en;q=0.5',
-                'Accept-Encoding': 'gzip, deflate, br',
-                'Connection': 'keep-alive',
-                'Upgrade-Insecure-Requests': '1'
+                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
             }
         }
 
-        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,
-                'quiet': False,
-                'no_warnings': False,
-                'socket_timeout': 30,
-                'retries': 10
-            })
-
         try:
             with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                 info = ydl.extract_info(url, download=True)
@@ -76,8 +46,7 @@ class MediaDownloader:
                     filename = ydl.prepare_filename(info)
                     return filename if os.path.exists(filename) else None
             return None
-        except Exception as e:
-            print(f"Video download error: {str(e)}")
+        except:
             return None
 
     def download_images(self, url):
@@ -97,22 +66,11 @@ class MediaDownloader:
                 return None
 
             downloaded_files = []
-            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',
-                'Accept': 'image/webp,*/*',
-                'Accept-Language': 'en-US,en;q=0.5',
-                'Accept-Encoding': 'gzip, deflate, br',
-                'Connection': 'keep-alive'
-            }
-
             for img_url in j.urls:
                 try:
-                    response = requests.get(
-                        img_url, 
-                        headers=headers,
-                        timeout=30,
-                        allow_redirects=True
-                    )
+                    response = requests.get(img_url, headers={
+                        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
+                    })
                     response.raise_for_status()
 
                     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S_%f")
@@ -137,9 +95,6 @@ class MediaDownloader:
             return None
 
     def download_media(self, url):
-        if not url.startswith(('http://', 'https://')):
-            url = 'https://' + url
-
         video_path = self.download_video(url)
         if video_path:
             return [video_path]
@@ -153,8 +108,8 @@ class MediaDownloader:
 @app.route('/')
 def home():
     return """
-    <h1>Social Media Downloader API</h1>
-    <p>Use: /download?url=YOUR_URL_HERE</p>
+    <h1>downloader</h1>
+    <p>Use: /download?url=url</p>
     """
 
 @app.route('/download')
@@ -162,7 +117,7 @@ def download():
     try:
         url = request.args.get('url')
         if not url:
-            return "No URL provided", 400
+            return "No URL", 400
 
         downloader = MediaDownloader()
 
@@ -171,12 +126,14 @@ def download():
 
             if files and len(files) > 0:
                 if len(files) == 1:
+                    # Single file download
                     response = make_response(send_file(files[0], as_attachment=True))
                     @response.call_on_close
                     def cleanup():
                         downloader.cleanup()
                     return response
                 else:
+                    # Multiple files download
                     def generate():
                         for file_path in files:
                             with open(file_path, 'rb') as f:
@@ -198,7 +155,7 @@ def download():
                     return response
             else:
                 downloader.cleanup()
-                return "Failed to download media", 400
+                return "Failed", 400
 
         except Exception as e:
             downloader.cleanup()
@@ -208,4 +165,4 @@ def download():
         return f"Error: {str(e)}", 500
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+    app.run(host='0.0.0.0', port=8080, debug=True)