good morning: small ui change

.gitignore

@@ -5,3 +5,4 @@
 ./models/*
 forensics/__pycache__/*
 *.cpython-311.pyc
+test.ipynb

app_mcp.py

@@ -237,7 +237,15 @@ def predict_image(img, confidence_threshold):
     results = [infer(img, model_id, confidence_threshold) for model_id in model_ids]
     return img, results
 
-# Update predict_image_with_json to return results as a list of dicts
+def get_consensus_label(results):
+    labels = [r[4] for r in results if len(r) > 4]
+    if not labels:
+        return "No results"
+    consensus = max(set(labels), key=labels.count)
+    color = {"AI": "red", "REAL": "green", "UNCERTAIN": "orange"}.get(consensus, "gray")
+    return f"<b><span style='color:{color}'>{consensus}</span></b>"
+
+# Update predict_image_with_json to return consensus label
 
 def predict_image_with_json(img, confidence_threshold, augment_methods, rotate_degrees, noise_level, sharpen_strength):
     if augment_methods:
@@ -268,7 +276,8 @@ def predict_image_with_json(img, confidence_threshold, augment_methods, rotate_d
         r.get("Real Score", ""),
         r.get("Label", "")
     ] for r in results]
-    return img_pil, forensics_images, table_rows, results
+    consensus = get_consensus_label(table_rows)
+    return img_pil, forensics_images, table_rows, results, consensus
 
 with gr.Blocks(css="#post-gallery { overflow: hidden !important;} .grid-wrap{ overflow-y: hidden !important;} .ms-gr-ant-welcome-icon{ height:unset !important;} .tabs{margin-top:10px;}") as demo:
     with ms.Application() as app:
@@ -306,8 +315,9 @@ with gr.Blocks(css="#post-gallery { overflow: hidden !important;} .grid-wrap{ ov
                         forensics_gallery = gr.Gallery(label="Post Processed Images", visible=True, columns=[4], rows=[2], container=False, height="auto", object_fit="contain", elem_id="post-gallery")
                         with gr.Accordion("Debug Output (Raw JSON)", open=False):
                             debug_json = gr.JSON(label="Raw Model Results")
+                        consensus_md = gr.Markdown(label="Consensus", value="")
 
-                        outputs = [image_output, forensics_gallery, results_table, debug_json]
+                        outputs = [image_output, forensics_gallery, results_table, debug_json, consensus_md]
                 
                 # Show/hide rotate slider based on selected augmentation method
                 augment_checkboxgroup.change(lambda methods: gr.update(visible="rotate" in methods), inputs=[augment_checkboxgroup], outputs=[rotate_slider])

requirements.txt

@@ -1,7 +1,7 @@
-gradio
+gradio[mcp]
 gradio_leaderboard
 transformers
-huggingface_hub
+huggingface_hub[xet]
 torchvision
 torch
 spaces