Update app.py

app.py

@@ -1,13 +1,12 @@
 import gradio as gr
 import cv2
 import numpy as np
-import pandas as pd
 import pydicom
 
+
 class DicomAnalyzer:
     def __init__(self):
         self.circle_diameter = 9
-        self.results = []
         self.current_image = None
         self.image_display = None
         self.dicom_data = None
@@ -17,10 +16,11 @@ class DicomAnalyzer:
             if file is None:
                 return None, "No file selected"
 
+            # Load DICOM file
             dicom_data = pydicom.dcmread(file.name)
             image = dicom_data.pixel_array.astype(np.float32)
 
-            # Apply rescale slope and intercept
+            # Apply rescale slope and intercept if available
             rescale_slope = getattr(dicom_data, 'RescaleSlope', 1)
             rescale_intercept = getattr(dicom_data, 'RescaleIntercept', 0)
             image = (image * rescale_slope) + rescale_intercept
@@ -29,42 +29,54 @@ class DicomAnalyzer:
             image_display = cv2.normalize(image, None, 0, 255, cv2.NORM_MINMAX).astype(np.uint8)
             image_display = cv2.cvtColor(image_display, cv2.COLOR_GRAY2BGR)
 
-            # Save the current state
+            # Store state
             self.current_image = image
             self.image_display = image_display
             self.dicom_data = dicom_data
 
             return image_display, "File loaded successfully"
         except Exception as e:
-            return None, f"Error loading file: {str(e)}"
+            return None, f"Error loading DICOM file: {str(e)}"
 
     def update_circle_diameter(self, value):
         self.circle_diameter = value
         return f"Circle diameter updated to {value}"
 
+
 def create_interface():
     analyzer = DicomAnalyzer()
 
     with gr.Blocks() as interface:
         gr.Markdown("# CT DICOM Image Analyzer")
 
+        # File Upload and Image Display
         with gr.Row():
             file_input = gr.File(label="Upload DICOM File")
             image_output = gr.Image(label="DICOM Image", interactive=False, type="numpy")
             status_output = gr.Textbox(label="Status", interactive=False)
 
+        # Circle Diameter Slider
         with gr.Row():
             circle_diameter_slider = gr.Slider(
                 minimum=1, maximum=20, value=9, step=1, label="Circle Diameter"
             )
             circle_status = gr.Textbox(label="Circle Diameter Status", interactive=False)
 
-        # Link events
-        file_input.change(fn=analyzer.load_dicom, inputs=file_input, outputs=[image_output, status_output])
-        circle_diameter_slider.change(fn=analyzer.update_circle_diameter, inputs=circle_diameter_slider, outputs=circle_status)
+        # Link Events
+        file_input.change(
+            fn=analyzer.load_dicom,
+            inputs=file_input,
+            outputs=[image_output, status_output],
+        )
+        circle_diameter_slider.change(
+            fn=analyzer.update_circle_diameter,
+            inputs=circle_diameter_slider,
+            outputs=circle_status,
+        )
 
     return interface
 
+
 if __name__ == "__main__":
     interface = create_interface()
     interface.launch()