Upload 2 files

app.py

@@ -2290,7 +2290,20 @@ def create_interface():
                     treatment_outcomes = treatment_df['outcome_score'].values
                     logger.info(f"Using treatment outcomes from {treatment_file}")
                 else:
-                    # Create a fallback set of treatment outcomes
+                    # Check if we should show an error or use mock data
+                    if not app_state.get('synthetic_data', False):
+                        # Show error for missing treatment file
+                        logger.error("No treatment file available and not using synthetic data")
+                        error_fig = plt.figure(figsize=(10, 6))
+                        message = "Error: Treatment outcomes file not found. Please retrain the VAE in Tab 1."
+                        plt.text(0.5, 0.5, message, 
+                                horizontalalignment='center', verticalalignment='center',
+                                fontsize=14, color='red')
+                        plt.axis('off')
+                        
+                        return [error_fig, error_fig, "Error: Treatment outcomes file not found."]
+                    
+                    # Create a fallback set of treatment outcomes for synthetic data
                     logger.info("No treatment outcomes found - creating mock data")
                     n_samples = len(app_state['latents'])
                     
@@ -2377,17 +2390,7 @@ def create_interface():
                     metrics_text = f"{metrics_text}\n\nNOTE: Using synthetic data for demonstration."
                 
                 return [importance_fig, performance_fig, metrics_text]
-            else:
-                # No treatment file available
-                error_fig = plt.figure(figsize=(10, 6))
-                message = "Error: Treatment outcomes file not found. Please retrain the VAE in Tab 1."
-                plt.text(0.5, 0.5, message, 
-                        horizontalalignment='center', verticalalignment='center',
-                        fontsize=14, color='red')
-                plt.axis('off')
                 
-                return [error_fig, error_fig, "Error: Treatment outcomes file not found."]
-            
             except Exception as e:
                 logger.error(f"Error in RF training: {str(e)}", exc_info=True)
                 error_fig = plt.figure(figsize=(10, 6))