Upload app.py

app.py

@@ -31,6 +31,13 @@ try:
 except:
     pass  # In case we don't have permission to change permissions
 
+# Try to create results directory at startup
+try:
+    os.makedirs('results/inference_atlantic', exist_ok=True)
+    os.chmod('results/inference_atlantic', 0o755)
+except:
+    print("⚠️ WARNING: Could not create results directory - will use temp directory for results")
+
 # Global variables for progress tracking
 current_progress = {'step': 'idle', 'progress': 0, 'details': ''}
 
@@ -606,14 +613,27 @@ def run_inference_pipeline(csv_file_path, model_path, normalization_path):
         # Step 11: Save results
         update_progress('Saving results', 95, 'Saving inference results...')
         
-        # Create results directory
-        results_dir = Path('results/inference_atlantic')
-        results_dir.mkdir(parents=True, exist_ok=True)
-        
-        # Save to results directory
-        timestamp = pd.Timestamp.now().strftime('%Y%m%d_%H%M%S')
-        results_file = results_dir / f'inference_results_{timestamp}.csv'
-        results_df.to_csv(results_file, index=False)
+        # Create results directory with permission handling
+        try:
+            results_dir = Path('results/inference_atlantic')
+            results_dir.mkdir(parents=True, exist_ok=True)
+            
+            # Save to results directory
+            timestamp = pd.Timestamp.now().strftime('%Y%m%d_%H%M%S')
+            results_file = results_dir / f'inference_results_{timestamp}.csv'
+            results_df.to_csv(results_file, index=False)
+            print(f"✅ Results saved to: {results_file}")
+        except PermissionError:
+            # Fallback to temp directory if results directory has permission issues
+            import tempfile
+            temp_dir = Path(tempfile.gettempdir())
+            timestamp = pd.Timestamp.now().strftime('%Y%m%d_%H%M%S')
+            results_file = temp_dir / f'inference_results_{timestamp}.csv'
+            results_df.to_csv(results_file, index=False)
+            print(f"⚠️ WARNING: Results saved to temp directory due to permissions: {results_file}")
+        except Exception as e:
+            print(f"❌ ERROR: Could not save results file: {str(e)}")
+            # Continue anyway - we still have the temp file below
         
         # Also save to temporary file for compatibility
         output_file = tempfile.NamedTemporaryFile(
@@ -842,23 +862,37 @@ def get_progress():
 
 @app.route('/download_results')
 def download_results():
-    # Find the most recent results file
-    upload_dir = app.config['UPLOAD_FOLDER']
-    temp_dir = tempfile.gettempdir()
+    # Find the most recent results file in multiple locations
+    search_directories = [
+        'results/inference_atlantic',  # Default results directory
+        app.config['UPLOAD_FOLDER'],   # Uploads directory
+        tempfile.gettempdir()          # System temp directory
+    ]
+    
+    latest_file = None
+    latest_time = 0
     
-    # Look for results files in both directories
-    for directory in [upload_dir, temp_dir]:
+    for directory in search_directories:
         if os.path.exists(directory):
-            files = [f for f in os.listdir(directory) if f.endswith('_inference_results.csv')]
-            if files:
-                latest_file = max(files, key=lambda x: os.path.getctime(os.path.join(directory, x)))
-                return send_file(
-                    os.path.join(directory, latest_file),
-                    as_attachment=True,
-                    download_name='vessel_inference_results.csv'
-                )
+            try:
+                files = [f for f in os.listdir(directory) if f.endswith('_inference_results.csv')]
+                for file in files:
+                    file_path = os.path.join(directory, file)
+                    file_time = os.path.getctime(file_path)
+                    if file_time > latest_time:
+                        latest_time = file_time
+                        latest_file = file_path
+            except PermissionError:
+                continue  # Skip directories we can't read
+    
+    if latest_file and os.path.exists(latest_file):
+        return send_file(
+            latest_file,
+            as_attachment=True,
+            download_name='vessel_inference_results.csv'
+        )
     
-    return jsonify({'error': 'No results file found'}), 404
+    return jsonify({'error': 'No results file found. Please run inference first.'}), 404
 
 ########################################
 #           SOCKETIO EVENTS            #