Update app.py

app.py

@@ -116,7 +116,9 @@ def predict_brain_age(nifti_file, actual_age): #sex
     bad_output_html = f"<span style='color:{color}; font-weight:bold;'>Brain Age Difference: {brain_age_difference:.2f} years</span>"
 
     # Return formatted outputs
-    return f"Predicted Brain Age: {predicted_brain_age_corrected:.2f} years", bad_output_html
+    #return f"Predicted Brain Age: {predicted_brain_age_corrected:.2f} years", bad_output_html
+    return f"Predicted Brain Age: {predicted_brain_age_corrected:.2f} years", f"Brain Age Difference: {brain_age_difference:.2f} years"
+
 
 # 🔹 Gradio Interface Setup
 with gr.Blocks() as demo:
@@ -138,13 +140,15 @@ with gr.Blocks() as demo:
         with gr.Column(scale=1):
             mri_input = gr.File(label="Upload a T1w MRI (NIfTI .nii.gz)")
             age_input = gr.Number(label="Enter Age", value=50)
-            #sex_input = gr.Radio(["Male", "Female"], label="Select Sex")
+            sex_input = gr.Radio(["Male", "Female"], label="Select Sex")
             submit_button = gr.Button("Predict")
 
         with gr.Column(scale=2):
             brain_age_output = gr.Textbox(label="Predicted Brain Age", interactive=False)
             #bad_output = gr.HTML(label="Brain Age Difference")  # Use gr.HTML for colored text
-            bad_output = gr.HTML(value="", show_label=True, label="Brain Age Difference")
+            #bad_output = gr.HTML(value="", show_label=True, label="Brain Age Difference")
+            bad_output = gr.Textbox(label="Brain Age Difference", interactive=False)
+
 
     submit_button.click(
         fn=predict_brain_age,