Update app.py

app.py

@@ -295,9 +295,15 @@ def calculate_weighted_rank(df, app_mode, w_s, w_t, w_d, w_sa): # Added 4 weight
     ).round(4)
 
     df = df.sort_values(by="Rank Score", ascending=False)
+    visible_cols = [
+        "p-SMILES", "Strength (GPa)", "Tg (°C)",
+        "Degradability", "SA Score",
+        "Confidence", "Classification", "Rank Score"
+    ]
+    display_df = df[visible_cols]
     output_path = tempfile.NamedTemporaryFile(delete=False, suffix=".csv").name
     df.to_csv(output_path, index=False)
-    return df, df, output_path
+    return df, display_df, output_path
 
 # --- 5b. RADAR PLOT LOGIC ---
 def update_radar_plot(smiles, df_results):
@@ -801,9 +807,9 @@ with gr.Blocks(title="PolymerDesignAI") as demo:
             if df is None or df.empty:
                 return None
         
-            if mode == "excellent":
+            if mode == "Excellent":
                 filtered = df[df["Classification"].str.contains("Excellent")]
-            elif mode == "review":
+            elif mode == "Review":
                 filtered = df[df["Classification"].str.contains("Review")]
             else:
                 filtered = df