Update run.py

run.py

@@ -1,16 +1,58 @@
-import pandas as pd 
+
+import pandas as pd
 import gradio as gr
 
-df = pd.DataFrame({"A" : [14, 4, 5, 4, 1], 
-				"B" : [5, 2, 54, 3, 2], 
-				"C" : [20, 20, 7, 3, 8], 
-				"D" : [14, 3, 6, 2, 6], 
-				"E" : [23, 45, 64, 32, 23]}) 
 
-t = df.style.highlight_max(color = 'lightgreen', axis = 0)
+# Sample DataFrame
+data = {
+    "Name": [
+        "Alice",
+        "Bob",
+        "Charlie",
+        "David",
+        "Eva",
+        "Charlize",
+        "Barbara",
+        "Hans",
+        "Sandra",
+        "Erik",
+    ],
+    "Age": [25, 30, 35, 40, 45, 29, 80, 95, 10, 25],
+    "Score": [88, 92, 87, 94, 90, 88, 92, 87, 94, 90],
+    "likes_cheese": [
+        "no",
+        "yes",
+        "no_opinion",
+        "yes",
+        "no",
+        "no",
+        "yes",
+        "no_opinion",
+        "yes",
+        "no",
+    ],
+}
+df = pd.DataFrame(data)
+
+# Define colors for the different values in the "likes_cheese" column
+colors = {
+    "yes": "lightgreen",
+    "no": "lightcoral",
+    "no_opinion": "#eee",
+}
+
+
+def highlight_row(s):
+    color = colors[s["likes_cheese"]]
+    # we set the text color to dark gray here so that it is readable
+    return ["color: #333; background-color: %s" % color for _ in s]
+
+
+styler = df.style.apply(highlight_row, axis="columns")
+
 
 with gr.Blocks() as demo:
-    gr.Dataframe(t)
-    
+    gr.Dataframe(styler)
+
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()