bug fixes1

app.py

@@ -20,6 +20,7 @@ import asyncio
 from flask_compress import Compress
 from flask_caching import Cache
 import hashlib
+from uuid import uuid4
 
 # 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')
@@ -56,6 +57,16 @@ logger = logging.getLogger(__name__)
 
 THREAD_POOL = ThreadPoolExecutor(max_workers=8)
 
+# Track files by session
+session_files = {}
+INACTIVE_SESSION_TIMEOUT = 3600  # 1 hour in seconds
+
+def get_session_id():
+    """Generate or retrieve session ID from request"""
+    if 'X-Session-ID' in request.headers:
+        return request.headers['X-Session-ID']
+    return str(uuid4())
+
 def allowed_file(filename):
     return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
@@ -110,17 +121,19 @@ def index():
     return render_template('index.html', title="Soundscape - 3D Music Visualizer")
 
 def cleanup_old_files():
+    """Enhanced cleanup that handles both time-based and session-based cleanup"""
     while True:
         current_time = datetime.now()
         files_to_delete = []
         
-        # Check all files
+        # Cleanup files by age
         for filename, timestamp in file_timestamps.items():
             if current_time - timestamp > FILE_LIFETIME:
                 filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
                 try:
                     if os.path.exists(filepath):
                         os.remove(filepath)
+                        logger.info(f"Deleted old file: {filename}")
                     files_to_delete.append(filename)
                 except Exception as e:
                     logger.error(f"Error deleting file {filename}: {str(e)}")
@@ -128,6 +141,28 @@ def cleanup_old_files():
         # Remove deleted files from timestamps
         for filename in files_to_delete:
             file_timestamps.pop(filename, None)
+        
+        # Cleanup files by session
+        inactive_sessions = []
+        current_timestamp = time.time()
+        
+        for session_id, session_data in session_files.items():
+            last_access = session_data.get('last_access', 0)
+            if current_timestamp - last_access > INACTIVE_SESSION_TIMEOUT:
+                # Delete all files for this session
+                for filename in session_data.get('files', []):
+                    filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+                    try:
+                        if os.path.exists(filepath):
+                            os.remove(filepath)
+                            logger.info(f"Deleted session file: {filename}")
+                    except Exception as e:
+                        logger.error(f"Error deleting session file {filename}: {str(e)}")
+                inactive_sessions.append(session_id)
+        
+        # Remove inactive sessions
+        for session_id in inactive_sessions:
+            session_files.pop(session_id, None)
             
         time.sleep(300)  # Check every 5 minutes
 
@@ -160,6 +195,17 @@ def serve_static(filename):
 def upload_file():
     logger.info('Upload request received')
     
+    session_id = get_session_id()
+    
+    if session_id not in session_files:
+        session_files[session_id] = {
+            'files': [],
+            'last_access': time.time()
+        }
+    
+    # Update last access time
+    session_files[session_id]['last_access'] = time.time()
+    
     if 'files[]' not in request.files:
         logger.warning('No files in request')
         return jsonify({'success': False, 'error': 'No files uploaded'}), 400
@@ -184,27 +230,26 @@ def upload_file():
                 
                 filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
                 
-                # Use buffered write for faster file saving
                 with open(filepath, 'wb') as f:
                     while True:
-                        chunk = file.read(8192)  # Read in 8KB chunks
+                        chunk = file.read(8192)
                         if not chunk:
                             break
                         f.write(chunk)
                 
                 file_timestamps[filename] = datetime.now()
+                # Track file with session
+                session_files[session_id]['files'].append(filename)
                 
-                # Process metadata in parallel
                 metadata = extract_metadata(filepath)
-                
-                # Invalidate cache if needed
                 invalidate_metadata_cache(filepath)
                 
                 return {
                     'filename': file.filename,
                     'success': True,
                     'filepath': f'/static/uploads/{filename}',
-                    'metadata': metadata
+                    'metadata': metadata,
+                    'session_id': session_id  # Return session ID to client
                 }
             return {
                 'filename': file.filename,
@@ -225,7 +270,8 @@ def upload_file():
 
     return jsonify({
         'success': True,
-        'files': results
+        'files': results,
+        'session_id': session_id
     })
 
 @app.route('/static/uploads/<filename>')
@@ -252,5 +298,26 @@ def add_header(response):
 def invalidate_metadata_cache(filepath):
     cache.delete_memoized(extract_metadata, filepath)
 
+# Add endpoint to explicitly end session
+@app.route('/end-session', methods=['POST'])
+def end_session():
+    session_id = request.headers.get('X-Session-ID')
+    if session_id and session_id in session_files:
+        # Delete all files for this session
+        for filename in session_files[session_id].get('files', []):
+            filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+            try:
+                if os.path.exists(filepath):
+                    os.remove(filepath)
+                    logger.info(f"Deleted session file: {filename}")
+            except Exception as e:
+                logger.error(f"Error deleting session file {filename}: {str(e)}")
+        
+        # Remove session data
+        session_files.pop(session_id, None)
+        return jsonify({'success': True, 'message': 'Session ended and files cleaned up'})
+    
+    return jsonify({'success': False, 'error': 'Session not found'}), 404
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860) 

static/js/main.js

@@ -1310,12 +1310,9 @@ async function handleFiles(files) {
         const response = await fetch('/upload', {
             method: 'POST',
             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) + '%';
-            }
+            headers: sessionId ? {
+                'X-Session-ID': sessionId
+            } : {}
         });
 
         if (!response.ok) {
@@ -1372,6 +1369,12 @@ async function handleFiles(files) {
                     item.classList.toggle('active', i === currentTrackIndex);
                 });
             }
+
+            // Store session ID if we got one
+            if (data.session_id) {
+                sessionId = data.session_id;
+                localStorage.setItem('audioSessionId', sessionId);
+            }
         } else {
             console.error('Upload failed:', data.error);
             showError(data.error || 'Upload failed');
@@ -1455,4 +1458,24 @@ function setupVolumeControl() {
 document.addEventListener('DOMContentLoaded', () => {
     // ... other initialization code ...
     setupVolumeControl();
+});
+
+// Add session management
+let sessionId = localStorage.getItem('audioSessionId');
+
+// Add cleanup on page unload
+window.addEventListener('beforeunload', async () => {
+    if (sessionId) {
+        try {
+            await fetch('/end-session', {
+                method: 'POST',
+                headers: {
+                    'X-Session-ID': sessionId
+                }
+            });
+            localStorage.removeItem('audioSessionId');
+        } catch (error) {
+            console.error('Error ending session:', error);
+        }
+    }
 });