fix to energy corrections

app.py

@@ -72,8 +72,8 @@ for subset, df in elements_df.items():
 
 map_functional = {
     "PBE": "compatible_pbe",
-    "PBESol": "compatible_pbesol",
-    "SCAN": "compatible_scan",
+    "PBESol (No correction scheme)": "compatible_pbesol",
+    "SCAN (No correction scheme)": "compatible_scan",
 }
 
 
@@ -116,14 +116,16 @@ def create_phase_diagram(
 
     # Fetch all entries from the Materials Project database
     def get_energy_correction(energy_correction, row):
-        if energy_correction == "Database specific, or MP2020":
+        if energy_correction == "Database specific, or MP2020" and map_functional[functional] == 'pbe':
             return (
                 row["energy_corrected"] - row["energy"]
                 if not np.isnan(row["energy_corrected"])
                 else 0
             )
-        elif energy_correction == "The 110 PBE Method":
-            return row["energy"] * 1.1
+        elif energy_correction == "The 110 PBE Method" and map_functional[functional] == 'pbe':
+            return row["energy"] * 1.1-row['energy']
+        elif map_functional[functional] != 'pbe':
+            return 0
 
     entries = [
         ComputedStructureEntry(
@@ -194,7 +196,7 @@ energy_correction_dropdown = gr.Dropdown(
     label="Energy correction",
 )
 plot_style_dropdown = gr.Dropdown(choices=["2D", "3D"], label="Plot Style")
-functional_dropdown = gr.Dropdown(choices=["PBE", "PBESol", "SCAN"], label="Functional")
+functional_dropdown = gr.Dropdown(choices=["PBE", "PBESol (No correction scheme)", "SCAN (No correction scheme)"], label="Functional")
 finite_temp_toggle = gr.Checkbox(label="Enable Finite Temperature Estimation")
 
 warning_message = "⚠️ This application uses energy correction schemes directly"