Update app.py

app.py

@@ -849,6 +849,12 @@ def batch_file_processing(file_paths):
         output_text_list,
     )
 
+def highlight_stability(s):
+    return [
+        "background-color: yellow; font-weight: bold;" if "stability" in str(v).lower() else ""
+        for v in s
+    ]
+
 
 def variable_outputs(file_inputs):
 
@@ -964,9 +970,11 @@ def variable_outputs(file_inputs):
 
         if isinstance(df_builder_pivot, pd.DataFrame):
             logger.debug(f"df_builder_pivot: {df_builder_pivot}")
-            df_builder_pivot["Trust Bucket®"] = df_builder_pivot["Trust Bucket®"].apply(
-                lambda x: f"<span style='background-color:yellow;'>{x}</span>" if "stability" in x.lower() else x
-            )
+            styled_df = df_builder_pivot.style.apply(highlight_stability, axis=1)
+            styled_html = styled_df.render()
+
+
+            
 
             markdown_5 = gr.Markdown(
                 "<span style='font-size:20px; font-weight:bold;'> 4) TrustBuilders® </span>",
@@ -980,16 +988,16 @@ def variable_outputs(file_inputs):
                 + "<br> Note: Even if Trust Buckets® for Customers and Prospects overlap, the most effective statements are very different. This provides clear guidance for acquisition versus loyalty activities.",
                 visible=True,
             )
-
-            table_builder_2 = gr.Dataframe(
-                value=df_builder_pivot,
-                headers=list(df_builder_pivot.columns),
-                interactive=False,
-                label=f"{dataset_name}",
-                visible=True,
-                height=800,
-                wrap=True,
-            )
+            table_builder_2 = gr.HTML(styled_html)
+            # table_builder_2 = gr.Dataframe(
+            #     value=df_builder_pivot,
+            #     headers=list(df_builder_pivot.columns),
+            #     interactive=False,
+            #     label=f"{dataset_name}",
+            #     visible=True,
+            #     height=800,
+            #     wrap=True,
+            # )
 
             plots_visible.append(markdown_5)
             plots_visible.append(markdown_6)