Spaces:

enzer1992
/

Ytm-downloader

Runtime error

App Files Files Community

enzer1992 commited on Jan 2

Commit

7cbe46d

verified ·

1 Parent(s): 1d2d333

Update app.py

Browse files

Files changed (1) hide show

app.py +101 -233

app.py CHANGED Viewed

@@ -1,233 +1,101 @@
-from flask import Flask, render_template_string, request, jsonify, send_file
-import os
-from pathlib import Path
-from yt_dlp import YoutubeDL
-from PIL import Image
-import threading
-import uuid
-app = Flask(__name__)
-# Store download status
-downloads = {}
-# HTML template as a string in the same file
-HTML_TEMPLATE = """
-<!DOCTYPE html>
-<html>
-<head>
-    <title>YouTube Music Downloader</title>
-    <meta name="viewport" content="width=device-width, initial-scale=1">
-    <link href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css" rel="stylesheet">
-    <style>
-        .download-card {
-            margin: 20px 0;
-            padding: 15px;
-            border-radius: 8px;
-            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
-        }
-        .progress {
-            height: 25px;
-        }
-        .download-btn {
-            display: inline-block;
-            padding: 8px 16px;
-            background-color: #28a745;
-            color: white;
-            text-decoration: none;
-            border-radius: 4px;
-            margin-top: 10px;
-        }
-        .download-btn:hover {
-            background-color: #218838;
-            color: white;
-        }
-    </style>
-</head>
-<body>
-    <div class="container mt-5">
-        <h1 class="text-center mb-4">YouTube Music Downloader</h1>
-        <div class="row justify-content-center">
-            <div class="col-md-8">
-                <div class="card">
-                    <div class="card-body">
-                        <form id="downloadForm">
-                            <div class="input-group mb-3">
-                                <input type="text" class="form-control" id="url" placeholder="Enter YouTube URL" required>
-                                <button class="btn btn-primary" type="submit">Download</button>
-                            </div>
-                        </form>
-                    </div>
-                </div>
-                <div id="downloads"></div>
-            </div>
-        </div>
-    </div>
-    <script>
-        document.getElementById('downloadForm').onsubmit = async (e) => {
-            e.preventDefault();
-            const url = document.getElementById('url').value;
-            const response = await fetch('/download', {
-                method: 'POST',
-                headers: {'Content-Type': 'application/json'},
-                body: JSON.stringify({url: url})
-            });
-            const data = await response.json();
-            if (data.download_id) {
-                createDownloadCard(data.download_id);
-                document.getElementById('url').value = '';
-            }
-        };
-        function createDownloadCard(downloadId) {
-            const card = document.createElement('div');
-            card.className = 'download-card bg-light';
-            card.id = `download-${downloadId}`;
-            card.innerHTML = `
-                <div class="progress mb-3">
-                    <div class="progress-bar progress-bar-striped progress-bar-animated"
-                         role="progressbar" style="width: 0%">0%</div>
-                </div>
-                <div class="status">Downloading...</div>
-            `;
-            document.getElementById('downloads').prepend(card);
-            pollStatus(downloadId);
-        }
-        async function pollStatus(downloadId) {
-            while (true) {
-                const response = await fetch(`/status/${downloadId}`);
-                const data = await response.json();
-                const card = document.getElementById(`download-${downloadId}`);
-                const progressBar = card.querySelector('.progress-bar');
-                const statusDiv = card.querySelector('.status');
-                progressBar.style.width = data.progress;
-                progressBar.textContent = data.progress;
-                if (data.status === 'completed') {
-                    const downloadLink = document.createElement('a');
-                    downloadLink.href = `/download/${downloadId}`;
-                    downloadLink.className = 'download-btn';
-                    downloadLink.download = ''; // Ensures file downloads instead of opening
-                    downloadLink.textContent = 'Download MP3';
-                    statusDiv.innerHTML = 'Download complete! ';
-                    statusDiv.appendChild(downloadLink);
-                    break;
-                } else if (data.status === 'failed') {
-                    statusDiv.innerHTML = `Error: ${data.error}`;
-                    progressBar.className = 'progress-bar bg-danger';
-                    break;
-                }
-                await new Promise(resolve => setTimeout(resolve, 1000));
-            }
-        }
-    </script>
-</body>
-</html>
-"""
-class MusicDownloader:
-    def __init__(self):
-        self.download_dir = Path("downloads")
-        self.download_dir.mkdir(exist_ok=True)
-    def download(self, url: str, download_id: str) -> bool:
-        downloads[download_id] = {'status': 'downloading', 'progress': '0%'}
-        ydl_opts = {
-            'format': 'bestaudio/best',
-            'postprocessors': [{
-                'key': 'FFmpegExtractAudio',
-                'preferredcodec': 'mp3',
-                'preferredquality': '192',
-            }],
-            'outtmpl': str(self.download_dir / '%(title)s.%(ext)s'),
-            'writethumbnail': True,
-            'progress_hooks': [lambda d: self._progress_hook(d, download_id)],
-        }
-        try:
-            with YoutubeDL(ydl_opts) as ydl:
-                info = ydl.extract_info(url, download=True)
-                title = info.get('title', 'unknown_title')
-                # Convert thumbnail
-                webp_path = self.download_dir / f"{title}.webp"
-                if webp_path.exists():
-                    self._convert_thumbnail(webp_path)
-                downloads[download_id]['status'] = 'completed'
-                downloads[download_id]['title'] = title
-                return True
-        except Exception as e:
-            downloads[download_id]['status'] = 'failed'
-            downloads[download_id]['error'] = str(e)
-            return False
-    def _progress_hook(self, d, download_id):
-        if d['status'] == 'downloading':
-            downloads[download_id]['progress'] = d.get('_percent_str', '0%')
-    def _convert_thumbnail(self, thumbnail_path: Path):
-        try:
-            with Image.open(thumbnail_path) as img:
-                jpg_path = thumbnail_path.with_suffix('.jpg')
-                img.convert('RGB').save(jpg_path, 'JPEG')
-                thumbnail_path.unlink()
-        except Exception:
-            pass
-@app.route('/')
-def index():
-    return render_template_string(HTML_TEMPLATE)
-@app.route('/download', methods=['POST'])
-def start_download():
-    url = request.json.get('url')
-    if not url:
-        return jsonify({'error': 'No URL provided'}), 400
-    download_id = str(uuid.uuid4())
-    downloader = MusicDownloader()
-    threading.Thread(target=downloader.download, args=(url, download_id)).start()
-    return jsonify({'download_id': download_id})
-@app.route('/status/<download_id>')
-def get_status(download_id):
-    if download_id not in downloads:
-        return jsonify({'status': 'not_found'}), 404
-    return jsonify(downloads[download_id])
-@app.route('/download/<download_id>')
-def get_file(download_id):
-    if download_id not in downloads or downloads[download_id]['status'] != 'completed':
-        return 'File not found', 404
-    title = downloads[download_id]['title']
-    file_path = os.path.join('downloads', f'{title}.mp3')
-    if not os.path.exists(file_path):
-        return 'File not found', 404
-    return send_file(
-        file_path,
-        as_attachment=True,
-        download_name=f"{title}.mp3",
-        mimetype='audio/mpeg'
-    )
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=5000, debug=True)

+import os
+from flask import Flask, render_template, request, jsonify, send_file
+import threading
+import uuid
+from pathlib import Path
+from yt_dlp import YoutubeDL
+from PIL import Image
+app = Flask(__name__, template_folder=os.getcwd())  # Set template_folder to the current directory
+downloads = {}
+class MusicDownloader:
+    def __init__(self):
+        self.download_dir = Path("downloads")
+        self.download_dir.mkdir(exist_ok=True)
+    def download(self, url: str, download_id: str) -> bool:
+        downloads[download_id] = {'status': 'downloading', 'progress': '0%'}
+        ydl_opts = {
+            'format': 'bestaudio/best',
+            'postprocessors': [{
+                'key': 'FFmpegExtractAudio',
+                'preferredcodec': 'mp3',
+                'preferredquality': '192',
+            }],
+            'outtmpl': str(self.download_dir / '%(title)s.%(ext)s'),
+            'writethumbnail': True,
+            'progress_hooks': [lambda d: self._progress_hook(d, download_id)],
+        }
+        try:
+            with YoutubeDL(ydl_opts) as ydl:
+                info = ydl.extract_info(url, download=True)
+                title = info.get('title', 'unknown_title')
+                webp_path = self.download_dir / f"{title}.webp"
+                if webp_path.exists():
+                    self._convert_thumbnail(webp_path)
+                downloads[download_id]['status'] = 'completed'
+                downloads[download_id]['title'] = title
+                return True
+        except Exception as e:
+            downloads[download_id]['status'] = 'failed'
+            downloads[download_id]['error'] = str(e)
+            return False
+    def _progress_hook(self, d, download_id):
+        if d['status'] == 'downloading':
+            downloads[download_id]['progress'] = d.get('_percent_str', '0%')
+    def _convert_thumbnail(self, thumbnail_path: Path):
+        try:
+            with Image.open(thumbnail_path) as img:
+                jpg_path = thumbnail_path.with_suffix('.jpg')
+                img.convert('RGB').save(jpg_path, 'JPEG')
+                thumbnail_path.unlink()
+        except Exception:
+            pass
+@app.route('/')
+def index():
+    return render_template('index.html')  # No need for templates folder, index.html is in the same folder
+@app.route('/download', methods=['POST'])
+def start_download():
+    url = request.json.get('url')
+    if not url:
+        return jsonify({'error': 'No URL provided'}), 400
+    download_id = str(uuid.uuid4())
+    downloader = MusicDownloader()
+    threading.Thread(target=downloader.download, args=(url, download_id)).start()
+    return jsonify({'download_id': download_id})
+@app.route('/status/<download_id>')
+def get_status(download_id):
+    if download_id not in downloads:
+        return jsonify({'status': 'not_found'}), 404
+    return jsonify(downloads[download_id])
+@app.route('/download/<download_id>')
+def get_file(download_id):
+    if download_id not in downloads or downloads[download_id]['status'] != 'completed':
+        return 'File not found', 404
+    title = downloads[download_id]['title']
+    file_path = os.path.join('downloads', f'{title}.mp3')
+    if not os.path.exists(file_path):
+        return 'File not found', 404
+    return send_file(file_path, as_attachment=True)
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)  # Port set for Hugging Face Spaces