Upload 2 files

Dockerfile

@@ -16,8 +16,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy application files
 COPY . .
 
-# Create necessary directories
-RUN mkdir -p uploads results
+# Create necessary directories with proper permissions
+RUN mkdir -p uploads results && chmod 755 uploads results
 
 # Expose port
 EXPOSE 7860

app.py

@@ -23,8 +23,13 @@ app.config['UPLOAD_FOLDER'] = 'uploads'
 app.config['MAX_CONTENT_LENGTH'] = 500 * 1024 * 1024  # 500MB max file size
 socketio = SocketIO(app, cors_allowed_origins="*")
 
-# Ensure upload directory exists
+# Ensure upload directory exists with proper permissions
 os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
+# Fix permissions for Hugging Face Spaces
+try:
+    os.chmod(app.config['UPLOAD_FOLDER'], 0o755)
+except:
+    pass  # In case we don't have permission to change permissions
 
 # Global variables for progress tracking
 current_progress = {'step': 'idle', 'progress': 0, 'details': ''}
@@ -730,12 +735,28 @@ def upload_file():
             normalization_path = os.path.join(app.config['UPLOAD_FOLDER'], norm_filename)
             norm_file.save(normalization_path)
         
-        # Check if required files exist
+        # Check if required files exist - try both uploaded and default locations
         if not os.path.exists(model_path):
-            return jsonify({'success': False, 'error': f'Model file not found: {model_path}'})
+            # If uploaded model not found, try default location
+            if model_path.startswith(app.config['UPLOAD_FOLDER']):
+                default_model_path = 'best_model.pth'
+                if os.path.exists(default_model_path):
+                    model_path = default_model_path
+                else:
+                    return jsonify({'success': False, 'error': f'Model file not found in uploads ({model_path}) or root (best_model.pth). Please upload a model file.'})
+            else:
+                return jsonify({'success': False, 'error': f'Model file not found: {model_path}'})
         
         if not os.path.exists(normalization_path):
-            return jsonify({'success': False, 'error': f'Normalization file not found: {normalization_path}'})
+            # If uploaded normalization not found, try default location  
+            if normalization_path.startswith(app.config['UPLOAD_FOLDER']):
+                default_norm_path = 'normalization_params_1_atlanttic_regular_intervals_with_lat_lon_velocity_and_time_difference_filter_outlier_segment_min_20_points.json'
+                if os.path.exists(default_norm_path):
+                    normalization_path = default_norm_path
+                else:
+                    return jsonify({'success': False, 'error': f'Normalization file not found in uploads ({normalization_path}) or root. Please upload a normalization file.'})
+            else:
+                return jsonify({'success': False, 'error': f'Normalization file not found: {normalization_path}'})
         
         # Save CSV file
         csv_filename = secure_filename(csv_file.filename)