Update app.py

app.py

@@ -54,15 +54,15 @@ PROJECT_MD = r"""
 # 🧠 AD-Stage-Net: Four-Stage Alzheimer’s MRI Classification
 
 This demo runs strong single CNN backbones (and an optional 2-model ensemble) to classify brain MRI slices into four stages:
-- No Impairment**
-- Very Mild Impairment
-- Mild Impairment
-- Moderate Impairment
-
-**Code:** <https://github.com/katelynhur/AD-Stage-Net>  
-**Data sources:**  
-- Kaggle (Luke Chugh): <https://www.kaggle.com/datasets/lchugh1/alzheimers-mri-dataset>  
-- Kaggle (Marco Pinamonti): <https://www.kaggle.com/datasets/marcopeix/alzheimer-mri-dataset>  
+- **No Impairment**
+- **Very Mild Impairment**
+- **Mild Impairment**
+- **Moderate Impairment**
+
+**Code:** <https://github.com/katelynhur/AD-Stage-Net>  
+**Data sources:**  
+- Kaggle (Luke Chugh): <https://www.kaggle.com/datasets/lchugh1/alzheimers-mri-dataset>  
+- Kaggle (Marco Pinamonti): <https://www.kaggle.com/datasets/marcopeix/alzheimer-mri-dataset>  
 - HuggingFace (Falah): <https://huggingface.co/datasets/fahadalhuzali/brain-MRI-ad-classification>
 
 > ⚠️ For research only; not a medical device.
@@ -473,12 +473,12 @@ def run_infer(files, model1, model2, ckpt_map_json):
 #     singles_tbl = load_best_singles_table(LEADER_CSV)
 #     pairs_tbl   = load_best_pairs_table()
 
-#     css = """
-#     .gradio-container { background-color: #12192E !important; }
-#     .gradio-container h1, .gradio-container p, .gradio-container .label, .gradio-container .gr-markdown, .gradio-container .gr-label-text {
-#         color: white !important;
-#     }
-#     """
+    # css = """
+    # .gradio-container { background-color: #12192E !important; }
+    # .gradio-container h1, .gradio-container p, .gradio-container .label, .gradio-container .gr-markdown, .gradio-container .gr-label-text {
+    #     color: white !important;
+    # }
+    # """
 
 #     with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
 #         gr.Markdown(PROJECT_MD)
@@ -533,20 +533,44 @@ def build_app():
     singles_tbl = load_best_singles_table(LEADER_CSV)
     pairs_tbl   = load_best_pairs_table()
 
-    # The updated CSS to force the text to be white
+    # # The updated CSS to force the text to be white
+    # css = """
+    # .gradio-container {
+    #     background-color: #12192E !important;
+    # }
+    
+    # .gradio-container h1,
+    # .gradio-container p,
+    # .gradio-container .label,
+    # .gradio-container .gr-markdown *,
+    # .gradio-container .gr-label-text,
+    # .gradio-container .gr-radio-label {
+    #     color: white !important;
+    # }
+    # """
+
     css = """
     .gradio-container {
         background-color: #12192E !important;
     }
+
+    /* Force all text elements inside the markdown to be white */
+    .gradio-container .gr-markdown * {
+        color: white !important;
+    }
     
-    .gradio-container h1,
-    .gradio-container p,
-    .gradio-container .label,
-    .gradio-container .gr-markdown *,
+    /* Ensure the labels for the dropdowns are also white */
     .gradio-container .gr-label-text,
     .gradio-container .gr-radio-label {
         color: white !important;
     }
+    
+    /* Revert the text color for the input/output boxes to black */
+    .gradio-container .gr-input,
+    .gradio-container .gr-output,
+    .gradio-container .gr-button {
+        color: black !important;
+    }
     """
 
     with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo: