bug fixes

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, request, jsonify, send_from_directory
+from flask import Flask, render_template, request, jsonify, send_from_directory, make_response
 import os
 import tempfile
 import atexit
@@ -18,33 +18,49 @@ import base64
 from concurrent.futures import ThreadPoolExecutor
 import asyncio
 from flask_compress import Compress
+from flask_caching import Cache
+import hashlib
 
 # 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')
 if not os.path.exists(UPLOAD_DIR):
     os.makedirs(UPLOAD_DIR, exist_ok=True)
 
+# Initialize Flask app once with all configurations
 app = Flask(__name__, static_folder='static')
+
+# Add all app configs
 app.config['UPLOAD_FOLDER'] = UPLOAD_DIR
 app.config['MAX_CONTENT_LENGTH'] = 200 * 1024 * 1024  # 200MB max file size
 app.config['MAX_FILES'] = 10  # Maximum number of files that can be uploaded at once
 
-ALLOWED_EXTENSIONS = {'mp3', 'wav', 'ogg', 'flac'}  # Added FLAC support
-FILE_LIFETIME = timedelta(hours=1)  # Files will be deleted after 1 hour
+# Add cache config
+cache_config = {
+    "DEBUG": True,
+    "CACHE_TYPE": "SimpleCache",
+    "CACHE_DEFAULT_TIMEOUT": 300
+}
+app.config.from_mapping(cache_config)
+
+# Initialize cache
+cache = Cache(app)
 
-# Store file creation times
+# Rest of your constants
+ALLOWED_EXTENSIONS = {'mp3', 'wav', 'ogg', 'flac'}
+FILE_LIFETIME = timedelta(hours=1)
 file_timestamps = {}
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-THREAD_POOL = ThreadPoolExecutor(max_workers=8)  # Increased from 4 to 8
+THREAD_POOL = ThreadPoolExecutor(max_workers=8)
 
 def allowed_file(filename):
     return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
 # Optimize metadata extraction
+@cache.memoize(timeout=3600)  # Cache for 1 hour
 def extract_metadata(filepath):
     try:
         audio = File(filepath, easy=True)  # Added easy=True for faster loading
@@ -87,7 +103,9 @@ def extract_metadata(filepath):
 def favicon():
     return send_from_directory('static', 'favicon.ico', mimetype='image/x-icon')
 
+# Cache the index page
 @app.route('/')
+@cache.cached(timeout=300)  # Cache for 5 minutes
 def index():
     return render_template('index.html', title="Soundscape - 3D Music Visualizer")
 
@@ -124,6 +142,20 @@ def request_entity_too_large(error):
         'error': 'File too large. Maximum total size is 200MB'
     }), 413
 
+# Cache static file responses
+@app.route('/static/<path:filename>')
+@cache.cached(timeout=3600)  # Cache for 1 hour
+def serve_static(filename):
+    response = make_response(send_from_directory('static', filename))
+    
+    # Add cache control headers
+    response.headers['Cache-Control'] = 'public, max-age=3600'
+    response.headers['ETag'] = hashlib.md5(
+        str(time.time()).encode()
+    ).hexdigest()
+    
+    return response
+
 @app.route('/upload', methods=['POST'])
 def upload_file():
     logger.info('Upload request received')
@@ -165,6 +197,9 @@ def upload_file():
                 # Process metadata in parallel
                 metadata = extract_metadata(filepath)
                 
+                # Invalidate cache if needed
+                invalidate_metadata_cache(filepath)
+                
                 return {
                     'filename': file.filename,
                     'success': True,
@@ -213,5 +248,9 @@ def add_header(response):
         response.headers['Cache-Control'] = 'public, max-age=300'
     return response
 
+# Add cache invalidation for uploaded files
+def invalidate_metadata_cache(filepath):
+    cache.delete_memoized(extract_metadata, filepath)
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)