Update app.py

app.py

@@ -66,6 +66,33 @@ if device == "cpu":  # Optional check
     model.float()  # Ensure model is running on the CPU
 print(f"Model classes: {model.names}")
 
+# Check Image Quality
+def check_image_quality(frame: np.ndarray, input_resolution: int) -> bool:
+    """
+    Checks if the image quality (resolution) meets the required standard.
+    
+    Args:
+    - frame (np.ndarray): The current frame being processed.
+    - input_resolution (int): The required resolution.
+
+    Returns:
+    - bool: True if the resolution is valid, False otherwise.
+    """
+    height, width, _ = frame.shape
+    frame_resolution = width * height
+    
+    if frame_resolution < 12_000_000:  # Minimum resolution of 12MP
+        log_entries.append(f"Frame {frame_count}: Resolution {width}x{height} ({frame_resolution / 1e6:.2f}MP) below 12MP, non-compliant")
+        logging.error(f"Frame {frame_count}: Resolution {width}x{height} ({frame_resolution / 1e6:.2f}MP) below 12MP, non-compliant")
+        return False
+    
+    if frame_resolution < input_resolution:
+        log_entries.append(f"Frame {frame_count}: Output resolution {width}x{height} below input resolution")
+        logging.warning(f"Frame {frame_count}: Output resolution {width}x{height} below input resolution")
+        return False
+    
+    return True
+
 # Generate PDF report with Top 5 Images
 def generate_pdf_report(log_entries, detected_issues, chart_path, map_path, metrics, top_images):
     pdf = FPDF()