Update app.py

app.py

@@ -21,14 +21,20 @@ def extract_frames(gif_path):
         return None, f"Error loading GIF: {str(e)}"
 
 def preprocess_frame(frame):
-    """Preprocess a frame: apply Gaussian blur and adaptive thresholding."""
+    """Preprocess a frame: isolate mid-to-light pixels and enhance circular patterns."""
     # Apply Gaussian blur to reduce noise
     blurred = cv2.GaussianBlur(frame, (9, 9), 0)
-    # Adaptive thresholding to enhance lighter regions (CME features)
-    thresh = cv2.adaptiveThreshold(
-        blurred, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY, 11, 2
-    )
-    return thresh
+
+    # Isolate mid-to-light pixels (intensity range 100–200 in grayscale)
+    lower_bound = 100
+    upper_bound = 200
+    mask = cv2.inRange(blurred, lower_bound, upper_bound)
+
+    # Apply morphological operation to enhance circular patterns
+    kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (5, 5))
+    enhanced = cv2.dilate(mask, kernel, iterations=2)
+
+    return enhanced
 
 def detect_circles(frame_diff, image_center, min_radius=20, max_radius=200):
     """Detect circles in a frame difference image, centered at the Sun."""
@@ -37,8 +43,8 @@ def detect_circles(frame_diff, image_center, min_radius=20, max_radius=200):
         cv2.HOUGH_GRADIENT,
         dp=1.5,  # Resolution for better detection
         minDist=100,  # Prevent overlapping circles
-        param1=100,  # Edge threshold to reduce noise
-        param2=20,  # Accumulator threshold to detect faint circles
+        param1=80,  # Lower edge threshold to capture fainter edges
+        param2=15,  # Lower accumulator threshold to detect faint circles
         minRadius=min_radius,
         maxRadius=max_radius
     )
@@ -56,7 +62,7 @@ def detect_circles(frame_diff, image_center, min_radius=20, max_radius=200):
     return None
 
 def analyze_gif(gif_file):
-    """Analyze a GIF for growing concentric circles centered at the Sun."""
+    """Analyze a GIF for growing concentric circles of mid-to-light pixels."""
     try:
         # Handle Gradio file input
         gif_path = gif_file.name if hasattr(gif_file, 'name') else gif_file
@@ -85,7 +91,7 @@ def analyze_gif(gif_file):
 
             # Compute absolute difference between consecutive frames
             frame_diff = cv2.absdiff(frame2, frame1)
-            # Enhance contrast for lighter pixels
+            # Enhance contrast for the difference image
             frame_diff = cv2.convertScaleAbs(frame_diff, alpha=3.0, beta=0)
 
             # Detect circles centered at the Sun
@@ -114,9 +120,9 @@ def analyze_gif(gif_file):
 
         # Generate output frames and report
         results = []
-        report = "Analysis Report:\n"
+        report = "Analysis Report (as of 07:34 PM PDT, May 24, 2025):\n"
         if growing_circle_data:
-            report += f"Detected {len(growing_circle_data)} frames with growing concentric circles:\n"
+            report += f"Detected {len(growing_circle_data)} frames with growing concentric circles of mid-to-light pixels:\n"
             for c in growing_circle_data:
                 # Visualize the frame with detected circle
                 output_frame = cv2.cvtColor(c["output_frame"], cv2.COLOR_GRAY2RGB)
@@ -126,9 +132,9 @@ def analyze_gif(gif_file):
                 results.append(output_frame)
 
                 report += f"Frame {c['frame']}: Center at {c['center']}, Radius {c['radius']} pixels\n"
-            report += "\nConclusion: Growing concentric circles detected, indicative of a potential Earth-directed CME."
+            report += "\nConclusion: Growing concentric circles of mid-to-light pixels detected, indicative of a potential Earth-directed CME."
         else:
-            report += "No growing concentric circles detected. CME may not be Earth-directed."
+            report += "No growing concentric circles of mid-to-light pixels detected. CME may not be Earth-directed."
 
         return report, results
     except Exception as e:
@@ -143,7 +149,7 @@ iface = gr.Interface(
         gr.Gallery(label="Frames with Growing Circles")
     ],
     title="Solar CME Detection",
-    description="Upload a GIF of solar images to detect growing concentric circles indicative of Earth-directed coronal mass ejections (CMEs)."
+    description="Upload a GIF of solar images to detect growing concentric circles of mid-to-light pixels indicative of Earth-directed coronal mass ejections (CMEs)."
 )
 
 if __name__ == "__main__":