statistics mode

app.py

@@ -2,55 +2,29 @@ import gradio as gr
 import pandas as pd
 from PIL import Image
 import os
+from collections import Counter
+
+# --- App Configuration ---
 
 # Define your sections and labels
 SECTION_LABELS = {
     "Oil Pore Related Issues": [
-        "Very Large Pores (Not Red)",
-        "Whiteheads (Clogged Pores)",
-        "Blackheads (Clogged Pores)",
-        "Shinny Skin",
-        "Sebaceous Filaments (Sebum)"
+        "Very Large Pores (Not Red)", "Whiteheads (Clogged Pores)", "Blackheads (Clogged Pores)",
+        "Shinny Skin", "Sebaceous Filaments (Sebum)"
     ],
     "Acne and Blemishes": [
-        "Pustules",
-        "Papules",
-        "Nodules",
-        "Cysts",
-        "Acne",
-        "Rosacea",
-        "Telangiectasia",
-        "Milia",
-        "Scars",
-        "Ice Berg Scars",
-    ],
-    "Redness and Irritation": [
-        "Redness",
-        "Irritation",
-    ],
-    "Dryness and Texture Issues": [
-        "Dryness",
-        "Fine Lines / Wrinkles",
-        "Skin Flakes"
-    ],
-    "Aging and Elasticity Issues": [
-        "Loose Skin",
-        "Deep Wrinkles"
-    ],
-    "Pigmentation Issues": [
-        "Dark Spots",
-        "Melasma",
-        "Freckles"
+        "Pustules", "Papules", "Nodules", "Cysts", "Acne", "Rosacea",
+        "Telangiectasia", "Milia", "Scars", "Ice Berg Scars",
     ],
+    "Redness and Irritation": ["Redness", "Irritation"],
+    "Dryness and Texture Issues": ["Dryness", "Fine Lines / Wrinkles", "Skin Flakes"],
+    "Aging and Elasticity Issues": ["Loose Skin", "Deep Wrinkles"],
+    "Pigmentation Issues": ["Dark Spots", "Melasma", "Freckles"],
     "Rosacea": [
-        "Erythematous Telangiectasia Rosacea",
-        "Papillopustular Rosacea",
-        "Fimatoz Rosacea",
-        "Ocular Rosacea"
+        "Erythematotelangiectatic Rosacea", "Papulopustular Rosacea",
+        "Phymatous Rosacea", "Ocular Rosacea"
     ],
-    "Eczema": [
-        "Seboreik Dermatit"
-    ]
+    "Eczema": ["Deborrheic Dermatitis"]
 }
 
 # Define the sections for each column to control UI layout
@@ -62,21 +36,20 @@ column4_sections = ["Rosacea", "Eczema"]
 # Combine all section lists to define the exact UI order
 UI_ORDERED_SECTIONS = column1_sections + column2_sections + column3_sections + column4_sections
 
-# Flattened labels list, created in the SAME order as the UI checkboxes will be. This is the fix.
+# Flattened labels list, created in the SAME order as the UI checkboxes will be.
 ALL_LABELS = [
     label
     for section_name in UI_ORDERED_SECTIONS
     for label in SECTION_LABELS.get(section_name, [])
 ]
 
-
-# Global state
+# --- Global State ---
 images = []
 current_index = 0
 results = []
 annotations = {}
 
-# Core functions
+# --- Core Annotation Functions ---
 
 def display_image(idx):
     """Displays the image at the given index and its saved annotations."""
@@ -90,7 +63,6 @@ def display_image(idx):
         return [img, caption] + states
     return [None, "No images uploaded"] + [False] * len(ALL_LABELS)
 
-
 def navigate(delta):
     """Navigates to the next or previous image."""
     global current_index
@@ -99,31 +71,23 @@ def navigate(delta):
     current_index = (current_index + delta + len(images)) % len(images)
     return display_image(current_index)
 
-
 def submit(*selections):
     """Saves the current annotations to the state and writes to a CSV file."""
     if not images:
         return "No image to label", None
-
-    # Save selections to our annotations dictionary
     annotations[current_index] = list(selections)
     fname = os.path.basename(images[current_index])
     chosen_labels = [lbl for lbl, sel in zip(ALL_LABELS, selections) if sel]
     
     global results
-    # Remove any previous entry for this image to avoid duplicates
     results = [r for r in results if r['image'] != fname]
     results.append({'image': fname, 'labels': ', '.join(chosen_labels)})
-
-    # Write the updated results to a CSV file
     df = pd.DataFrame(results)
     df.to_csv('image_labels.csv', index=False)
-
     return "Labels saved!", 'image_labels.csv'
 
-
 def upload_images(files):
-    """Handles image uploads, resetting the application state."""
+    """Handles image uploads, resetting the annotation state."""
     global images, current_index, results, annotations
     images = [f.name for f in files]
     current_index = 0
@@ -137,133 +101,156 @@ def load_annotations(csv_file):
     """Loads annotations from an uploaded CSV file."""
     global annotations, results
     if csv_file is None or not images:
-        # If no CSV is uploaded or no images are loaded, do nothing.
         return display_image(current_index)
-
     try:
         df = pd.read_csv(csv_file.name)
-        # Create a quick lookup map from filename to its index in the `images` list
         image_map = {os.path.basename(name): i for i, name in enumerate(images)}
-        
-        # Reset existing annotations and results
         annotations = {}
         results = df.to_dict('records')
-
         for _, row in df.iterrows():
             fname = row['image']
-            # Check if the image from the CSV is in the currently uploaded images
             if fname in image_map:
                 img_idx = image_map[fname]
-                # Handle cases where labels might be empty (NaN)
-                if pd.notna(row['labels']):
-                    saved_labels = set(l.strip() for l in row['labels'].split(','))
-                else:
-                    saved_labels = set()
-                
-                # Create the boolean state list for the checkboxes
+                saved_labels = set(l.strip() for l in row['labels'].split(',')) if pd.notna(row['labels']) else set()
                 states = [label in saved_labels for label in ALL_LABELS]
                 annotations[img_idx] = states
     except Exception as e:
         print(f"Error loading annotations: {e}")
-        # In case of error, just refresh the current view without changes
-        return display_image(current_index)
-
-    # After loading, refresh the view to show the annotations for the current image
     return display_image(current_index)
 
+# --- New Statistics and Mode-Switching Functions ---
+
+def calculate_statistics(files):
+    """Reads multiple CSVs and calculates the frequency of each label."""
+    if not files:
+        return [gr.update() for _ in ALL_LABELS] # No change
+    
+    label_counts = Counter()
+    for file_obj in files:
+        try:
+            df = pd.read_csv(file_obj.name)
+            if 'labels' in df.columns:
+                df.dropna(subset=['labels'], inplace=True)
+                for label_str in df['labels']:
+                    labels = [l.strip() for l in label_str.split(',')]
+                    label_counts.update(labels)
+        except Exception as e:
+            print(f"Could not process file {file_obj.name}: {e}")
+            continue
+            
+    # Create the list of gr.update objects for the checkbox labels
+    updated_checkboxes = []
+    for label in ALL_LABELS:
+        count = label_counts.get(label, 0)
+        new_label_text = f"{label} (Count: {count})"
+        updated_checkboxes.append(gr.update(label=new_label_text))
+    return updated_checkboxes
+
+def toggle_mode(current_mode):
+    """Switches the UI between Annotation and Statistics modes."""
+    if current_mode == "Annotation":
+        new_mode = "Statistics"
+        btn_text = "Switch to Annotation Mode"
+        anno_visible = False
+        stats_visible = True
+        # Return default labels when switching away from annotation mode
+        label_updates = [gr.update(label=lbl) for lbl in ALL_LABELS]
+    else: # Current mode is "Statistics"
+        new_mode = "Annotation"
+        btn_text = "Switch to Statistics Mode"
+        anno_visible = True
+        stats_visible = False
+        # Reset labels back to their original names
+        label_updates = [gr.update(label=lbl) for lbl in ALL_LABELS]
+        
+    return [new_mode, gr.update(value=btn_text), gr.update(visible=anno_visible), gr.update(visible=stats_visible)] + label_updates
 
 # --- Gradio UI Definition ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## Dermatology Annotation Tool")
-    with gr.Row():
-        image_upload = gr.File(label="1. Upload Images", file_count="multiple", file_types=["image"])
-        csv_upload = gr.File(label="2. (Optional) Upload Annotations CSV", file_types=[".csv"], visible=False)
+    gr.Markdown("# Dermatology Annotation & Statistics Tool")
+    
+    # State to track the current mode
+    app_mode = gr.State("Annotation")
 
+    with gr.Row():
+        mode_toggle_btn = gr.Button("Switch to Statistics Mode")
+
+    # --- Annotation Mode UI ---
+    with gr.Group(visible=True) as annotation_ui:
+        with gr.Row():
+            image_upload = gr.File(label="1. Upload Images", file_count="multiple", file_types=["image"])
+            csv_upload = gr.File(label="2. (Optional) Upload Annotations CSV", file_types=[".csv"], visible=False)
+        # Image display and controls
+        with gr.Row():
+            with gr.Column(scale=2):
+                img = gr.Image(label="Image")
+                caption = gr.Label(value="No images uploaded")
+                with gr.Row():
+                    prev_btn = gr.Button("⬅️ Previous")
+                    next_btn = gr.Button("Next ➡️")
+            with gr.Column(scale=1):
+                submit_btn = gr.Button("Submit Labels")
+                status = gr.Label()
+                csv_downloader = gr.File(label="Download labels CSV")
+
+    # --- Statistics Mode UI ---
+    with gr.Group(visible=False) as statistics_ui:
+        stats_csv_upload = gr.File(
+            label="Upload one or more annotation CSV files",
+            file_count="multiple",
+            file_types=[".csv"]
+        )
+    
+    gr.Markdown("---")
+    
+    # --- Shared UI (Checkboxes) ---
     checkbox_components = []
-
     with gr.Row(): # Main row for the four columns of labels
-        # Column 1
-        with gr.Column(scale=1, min_width=0):
-            for section_name in column1_sections:
-                if section_name in SECTION_LABELS:
-                    with gr.Group():
-                        gr.Markdown(f"### {section_name}")
-                        for lbl in SECTION_LABELS[section_name]:
-                            cb = gr.Checkbox(label=lbl)
-                            checkbox_components.append(cb)
-        
-        # Column 2
-        with gr.Column(scale=1, min_width=0):
-            for section_name in column2_sections:
-                if section_name in SECTION_LABELS:
-                    with gr.Group():
-                        gr.Markdown(f"### {section_name}")
-                        for lbl in SECTION_LABELS[section_name]:
-                            cb = gr.Checkbox(label=lbl)
-                            checkbox_components.append(cb)
-
-        # Column 3
-        with gr.Column(scale=1, min_width=0):
-            for section_name in column3_sections:
-                if section_name in SECTION_LABELS:
-                    with gr.Group():
-                        gr.Markdown(f"### {section_name}")
-                        for lbl in SECTION_LABELS[section_name]:
-                            cb = gr.Checkbox(label=lbl)
-                            checkbox_components.append(cb)
-        
-        # Column 4
-        with gr.Column(scale=1, min_width=0):
-            for section_name in column4_sections:
-                if section_name in SECTION_LABELS:
-                    with gr.Group():
-                        gr.Markdown(f"### {section_name}")
-                        for lbl in SECTION_LABELS[section_name]:
-                            cb = gr.Checkbox(label=lbl)
-                            checkbox_components.append(cb)
-
-    # Image display and controls
-    with gr.Row():
-        with gr.Column(scale=2): # Image display column
-            img = gr.Image(label="Image")
-            caption = gr.Label(value="No images uploaded")
-            with gr.Row():
-                prev_btn = gr.Button("⬅️ Previous")
-                next_btn = gr.Button("Next ➡️")
-        with gr.Column(scale=1): # Controls and download column
-            submit_btn = gr.Button("Submit Labels")
-            status = gr.Label()
-            csv_downloader = gr.File(label="Download labels CSV")
+        for col_sections in [column1_sections, column2_sections, column3_sections, column4_sections]:
+            with gr.Column(scale=1, min_width=220):
+                for section_name in col_sections:
+                    if section_name in SECTION_LABELS:
+                        with gr.Group():
+                            gr.Markdown(f"### {section_name}")
+                            for lbl in SECTION_LABELS[section_name]:
+                                cb = gr.Checkbox(label=lbl)
+                                checkbox_components.append(cb)
 
     # --- Event Handling ---
     
-    # When images are uploaded
+    # Mode switching
+    mode_toggle_btn.click(
+        fn=toggle_mode,
+        inputs=app_mode,
+        outputs=[app_mode, mode_toggle_btn, annotation_ui, statistics_ui] + checkbox_components
+    )
+    
+    # Statistics calculation
+    stats_csv_upload.upload(
+        fn=calculate_statistics,
+        inputs=stats_csv_upload,
+        outputs=checkbox_components
+    )
+
+    # Annotation functionality
     image_upload.upload(
         fn=upload_images,
         inputs=image_upload,
         outputs=[img, caption] + checkbox_components + [image_upload, csv_upload]
     )
-
-    # When a CSV annotation file is uploaded
     csv_upload.upload(
         fn=load_annotations,
         inputs=csv_upload,
         outputs=[img, caption] + checkbox_components
     )
-
-    # When previous button is clicked
     prev_btn.click(
         fn=lambda: navigate(-1),
         outputs=[img, caption] + checkbox_components
     )
-
-    # When next button is clicked
     next_btn.click(
         fn=lambda: navigate(1),
         outputs=[img, caption] + checkbox_components
     )
-
-    # When submit button is clicked
     submit_btn.click(
         fn=submit,
         inputs=checkbox_components,
@@ -271,4 +258,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)