adding confidence chart

app.py

@@ -243,6 +243,65 @@ def login_auth(username, password):
 
 
 
+# Function to create individual bar charts for each defect type
+def generate_confidence_bar_chart(annotations):
+    # Categorize confidence scores
+    confidence_bins = {'<25%': 0, '25%-75%': 0, '>75%': 0}
+    defect_bins = {
+        "Nicks": confidence_bins.copy(),
+        "Dents": confidence_bins.copy(),
+        "Scratches": confidence_bins.copy(),
+        "Pittings": confidence_bins.copy(),
+    }
+
+    # Populate bins based on annotations
+    for annotation in annotations:
+        defect = annotation["category_name"]
+        score = annotation["score"] * 100  # Convert to percentage
+        if score < 25:
+            defect_bins[defect]['<25%'] += 1
+        elif 25 <= score <= 75:
+            defect_bins[defect]['25%-75%'] += 1
+        else:
+            defect_bins[defect]['>75%'] += 1
+
+    # Define colors for each defect
+    category_styles = {
+        'Nicks': 'rgba(255, 60, 60, 0.7)',       # Red
+        'Dents': 'rgba(255, 148, 156, 0.7)',     # Light Red
+        'Scratches': 'rgba(255, 116, 28, 0.7)',  # Orange
+        'Pittings': 'rgba(255, 180, 28, 0.7)'    # Yellow
+    }
+
+    # Generate individual charts
+    charts = []
+    for defect, bins in defect_bins.items():
+        fig = go.Figure()
+        fig.add_trace(go.Bar(
+            name=defect,
+            x=list(bins.keys()),  # Confidence ranges
+            y=list(bins.values()),  # Counts
+            text=[f"{v} defects" for v in bins.values()],  # Hover text
+            hoverinfo="text",
+            marker_color=category_styles.get(defect, 'rgba(255, 0, 0, 0.7)')  # Default to red
+        ))
+
+        # Customize layout
+        fig.update_layout(
+            title=f"{defect} Confidence Score Distribution",
+            xaxis_title="Confidence Range",
+            yaxis_title="Defect Count",
+            template="plotly_white"
+        )
+        charts.append(fig)
+
+    return charts  # Return list of charts
+
+
+
+
+
+
 # Gradio interface components
 with gr.Blocks() as demo:
 
@@ -448,8 +507,25 @@ with gr.Blocks() as demo:
                               inputs=output_annotations, 
                               outputs=frequency_graph_component)
         
+        
+
+    # Gradio row for confidence bar chart
+    with gr.Row(visible=False) as confidence_bar_chart_row:
+        nicks_confidence_bar_chart = gr.Plot(label="Nicks Confidence Score Distribution")
+        dents_confidence_bar_chart = gr.Plot(label="Dents Confidence Score Distribution")
+        scratches_confidence_bar_chart = gr.Plot(label="Scratches Confidence Score Distribution")
+        pittings_confidence_bar_chart = gr.Plot(label="Pittings Confidence Score Distribution")
+
 
     
+    with gr.Row(visible=False) as confidence_btn_row:
+        confidence_chart_btn = gr.Button("Generate Confidence Chart", variant="primary")
+        confidence_chart_btn.click(
+            generate_confidence_bar_chart,
+            inputs=output_annotations,  # Pass the annotations
+            outputs=[nicks_confidence_bar_chart,dents_confidence_bar_chart,scratches_confidence_bar_chart,pittings_confidence_bar_chart]
+        )
+    
      # Login row, initially visible
     with gr.Row(visible=True) as login_row:
         with gr.Column():
@@ -474,15 +550,26 @@ with gr.Blocks() as demo:
                             gr.update(visible=login_state),  # Show area_graph_row
                             gr.update(visible=login_state),  # Show area_btn_row
                             gr.update(visible=login_state),  # Show frequency_graph_row
-                            gr.update(visible=login_state),   # Show frequency_btn_row
+                            gr.update(visible=login_state),  # Show frequency_btn_row
+                            gr.update(visible=login_state),  #Show Confidence chart
+                            gr.update(visible=login_state),  #Show Confidence btn
                             gr.update(visible=not login_state) # for login
                         ),
                         inputs=login_successful,
-                        outputs=[header_row, input_row, area_graph_row, area_btn_row, frequency_graph_row, frequency_btn_row, login_row]
+                        outputs=[header_row, 
+                                 input_row, 
+                                 area_graph_row, 
+                                 area_btn_row, 
+                                 frequency_graph_row, 
+                                 frequency_btn_row,
+                                 confidence_bar_chart_row, 
+                                 confidence_btn_row, 
+                                 login_row]
                     )
 
                 with gr.Column(scale=2):
                     gr.Markdown("")
+