Update app.py

app.py

@@ -74,8 +74,8 @@ title = "Welcome to **ADR Tracker**"
 description1 = "This app takes text (up to a few sentences) and predicts to what extent the text describes severe (or non-severe) adverse reaction to medications. Please do NOT use for medical diagnosis."
 
 css = """
-body { font-family: 'Roboto', sans-serif; background-color: #333; color: #87CEEB; }
-h1, h2, h3, h4, h5, h6, p, label, .markdown { color: #87CEEB; } /* Ensuring that all text elements are consistently light blue */
+body { font-family: 'Roboto', sans-serif; background-color: #333; color: #163E64; }
+h1, h2, h3, h4, h5, h6, p, label, .markdown { color: #163E64; } /* Ensuring that all text elements are consistently light blue */
 .textbox { width: 100%; border-radius: 10px; border: 1px solid #ccc; background-color: white; color: black; }
 .button { background-color: #FF6347; color: white; border: none; border-radius: 10px; padding: 10px 20px; cursor: pointer; }
 """
@@ -101,10 +101,18 @@ with gr.Blocks(css=css) as demo:
         api_name="adr"
     )
     
+    gr.Markdown("### Legend")
+    legend = gr.HTML(value="<div style='margin-top: 20px;'><strong>Legend:</strong><br>" +
+                            "<mark style='background-color:#E63946;'>Severity</mark> " +
+                            "<mark style='background-color:#2A9D8F;'>Sign/Symptom</mark> " +
+                            "<mark style='background-color:#457B9D;'>Medication</mark> " +
+                            "<mark style='background-color:#F4A261;'>Age/Sex</mark> " +
+                            "<mark style='background-color:#9C6644;'>Diagnostic Procedure</mark> " +
+                            "<mark style='background-color:#BDB2FF;'>Biological Structure</mark></div>")
+    gr.Row([legend])
+
     with gr.Row():
         gr.Markdown("### Click on any of the examples below to see how it works:")
         gr.Examples([["A 35 year-old male had severe headache after taking Aspirin. The lab results were normal."],
                      ["A 35 year-old female had minor pain in upper abdomen after taking Acetaminophen."]], 
                     [prob1], [label, local_plot, htext], main, cache_examples=True)
-
-demo.launch()