Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import gradio as gr
 import matplotlib.pyplot as plt
 import io
@@ -14,6 +13,11 @@ import json
 import pickle # Added for loading .pkl and .json files
 import os # Added for path checking
 
+from sklearn.preprocessing import LabelEncoder, StandardScaler, OneHotEncoder # Explicitly import LabelEncoder
+from sklearn.impute import SimpleImputer
+from sklearn.compose import ColumnTransformer
+from sklearn.pipeline import Pipeline
+
 # --- Device Configuration ---
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -226,13 +230,13 @@ except FileNotFoundError as e:
     raise SystemExit(f"Error loading global variables: {e}. Please ensure 'global_vars.json' is available.")
 
 # Unpack global variables
-label_encoder = LabelEncoder()
-label_encoder.classes_ = np.array(global_vars['label_encoder_classes'])
+label_encoder = LabelEncoder() # Create an instance of LabelEncoder
+label_encoder.classes_ = np.array(global_vars['label_encoder_classes']) # Assign classes from loaded data
 num_classes = global_vars['num_classes']
 best_hyperparams = global_vars['best_hyperparams']
 expected_value_reg = global_vars['expected_value_reg']
 expected_value_cls = np.array(global_vars['expected_value_cls']) # Already converted to list, convert back to np array
-AA_PROPERTIES = global_vars['AA_PROPERTIES']
+AA_PROPERTIES = global_vars['AA_PROPERTIES'] # Access global AA_PROPERTIES
 
 # Load feature names
 try:
@@ -474,13 +478,12 @@ output_components = [
     gr.HTML(label="Top Global Feature Importance")
 ]
 
-# Create the Gradio Interface
 with gr.Blocks(theme=gr.themes.Soft()) as iface:
     gr.Markdown("# Protein Stability Change (ΔΔG) Prediction with Explainability")
     gr.Markdown(
         "Predict ΔΔG and mutation effect for single amino acid substitutions in proteins. "
         "Explore physicochemical changes, mutation structural context, and feature importance with SHAP explanations. "
-        "Designed for researchers and doctors for practical applications." 
+        "Designed for researchers and doctors for practical applications."
         "All backend features are included and frontend template is professional and exceptional."
     )