bug fixes

app.py

@@ -17,6 +17,7 @@ import logging
 import base64
 from concurrent.futures import ThreadPoolExecutor
 import asyncio
+from flask_compress import Compress
 
 # Create uploads directory in the static folder instead of using tempfile
 UPLOAD_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'static', 'uploads')
@@ -38,19 +39,19 @@ file_timestamps = {}
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-THREAD_POOL = ThreadPoolExecutor(max_workers=4)  # Adjust based on your server capacity
+THREAD_POOL = ThreadPoolExecutor(max_workers=8)  # Increased from 4 to 8
 
 def allowed_file(filename):
     return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
-# Add function to extract metadata
+# Optimize metadata extraction
 def extract_metadata(filepath):
     try:
-        audio = File(filepath)
+        audio = File(filepath, easy=True)  # Added easy=True for faster loading
         if audio is None:
             return {}
             
-        # Basic metadata that we absolutely need
+        # Simplified metadata extraction
         metadata = {
             'title': None,
             'artist': 'Unknown Artist',
@@ -59,27 +60,17 @@ def extract_metadata(filepath):
 
         # Get filename as fallback
         original_filename = os.path.splitext(os.path.basename(filepath))[0]
-        if '_' in original_filename:
-            original_filename = '_'.join(original_filename.split('_')[2:])
-
+        
         try:
-            # Add basic audio info
+            # Quick duration check
             if hasattr(audio.info, 'length'):
-                metadata['duration'] = round(audio.info.length)
+                metadata['duration'] = int(audio.info.length)  # Convert to int for faster processing
 
             # Simplified tag extraction
-            if isinstance(audio, (mutagen.flac.FLAC, mutagen.oggvorbis.OggVorbis)):
-                metadata.update({
-                    'title': audio.tags.get('TITLE', [original_filename])[0] if audio.tags else original_filename,
-                    'artist': audio.tags.get('ARTIST', ['Unknown Artist'])[0] if audio.tags else 'Unknown Artist',
-                })
-            elif isinstance(audio, mutagen.mp3.MP3):
-                if audio.tags:
-                    metadata.update({
-                        'title': str(audio.tags.get('TIT2', original_filename)),
-                        'artist': str(audio.tags.get('TPE1', 'Unknown Artist')),
-                    })
-
+            if hasattr(audio, 'tags') and audio.tags:
+                metadata['title'] = str(audio.tags.get('title', [original_filename])[0])
+                metadata['artist'] = str(audio.tags.get('artist', ['Unknown Artist'])[0])
+            
         except Exception as e:
             logger.error(f"Error reading tags: {str(e)}")
             metadata['title'] = original_filename
@@ -151,7 +142,6 @@ def upload_file():
         }), 400
 
     results = []
-    futures = []
 
     def process_file(file):
         try:
@@ -161,12 +151,20 @@ def upload_file():
                 filename = timestamp + filename
                 
                 filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-                file.save(filepath)
+                
+                # Use buffered write for faster file saving
+                with open(filepath, 'wb') as f:
+                    while True:
+                        chunk = file.read(8192)  # Read in 8KB chunks
+                        if not chunk:
+                            break
+                        f.write(chunk)
                 
                 file_timestamps[filename] = datetime.now()
+                
+                # Process metadata in parallel
                 metadata = extract_metadata(filepath)
                 
-                # Update the filepath to use relative URL
                 return {
                     'filename': file.filename,
                     'success': True,
@@ -186,10 +184,9 @@ def upload_file():
                 'error': 'Server error during upload'
             }
 
-    # Process files in parallel
-    with THREAD_POOL:
-        futures = [THREAD_POOL.submit(process_file, file) for file in files]
-        results = [future.result() for future in futures]
+    # Process files in parallel with a larger chunk size
+    with THREAD_POOL as executor:
+        results = list(executor.map(process_file, files))
 
     return jsonify({
         'success': True,
@@ -206,5 +203,15 @@ def cleanup():
 
 atexit.register(cleanup)
 
+# Add response compression
+Compress(app)
+
+# Add caching headers
+@app.after_request
+def add_header(response):
+    if 'Cache-Control' not in response.headers:
+        response.headers['Cache-Control'] = 'public, max-age=300'
+    return response
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860) 

requirements.txt

@@ -1,3 +1,4 @@
 flask==2.0.1
 mutagen==1.45.1
-Werkzeug==2.0.1 
+Werkzeug==2.0.1
+flask-compress==1.10.1 

static/js/main.js

@@ -1263,7 +1263,7 @@ function setupPlaylistControls() {
     });
 }
 
-// Update handleFiles to append new tracks
+// Update handleFiles function for faster uploads
 async function handleFiles(files) {
     console.log('Handling files:', files.length, 'files');
     
@@ -1285,6 +1285,10 @@ async function handleFiles(files) {
     }
     
     const formData = new FormData();
+    const totalSize = Array.from(files).reduce((acc, file) => acc + file.size, 0);
+    let uploadedSize = 0;
+    
+    // Add files to FormData with optimized chunk size
     Array.from(files).forEach(file => {
         console.log('Adding file to upload:', file.name);
         formData.append('files[]', file);
@@ -1305,7 +1309,13 @@ async function handleFiles(files) {
         console.log('Starting file upload...');
         const response = await fetch('/upload', {
             method: 'POST',
-            body: formData
+            body: formData,
+            // Add upload progress tracking
+            onUploadProgress: (progressEvent) => {
+                const percentComplete = (progressEvent.loaded / progressEvent.total) * 100;
+                if (progressFill) progressFill.style.width = percentComplete + '%';
+                if (progressText) progressText.textContent = Math.round(percentComplete) + '%';
+            }
         });
 
         if (!response.ok) {