Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import numpy as np
 from datetime import datetime
 import matplotlib.pyplot as plt
+import time  # Added import for time module
 from services.detection_service import detect_faults_solar, detect_faults_windmill
 from services.anomaly_service import track_anomalies, predict_anomaly
 from models.solar_model import load_solar_model
@@ -82,6 +83,30 @@ if 'paused' not in st.session_state:
 SNAPSHOT_FOLDER = "./snapshots"
 os.makedirs(SNAPSHOT_FOLDER, exist_ok=True)
 
+# Function to resize and pad image to 640x640 while preserving aspect ratio
+def preprocess_image(image, target_size=(640, 640)):
+    h, w = image.shape[:2]
+    target_h, target_w = target_size
+
+    # Calculate scaling factor to maintain aspect ratio
+    scale = min(target_w / w, target_h / h)
+    new_w, new_h = int(w * scale), int(h * scale)
+
+    # Resize image
+    resized_image = cv2.resize(image, (new_w, new_h), interpolation=cv2.INTER_AREA)
+
+    # Create a blank 640x640 image
+    padded_image = np.zeros((target_h, target_w, 3), dtype=np.uint8)
+
+    # Calculate padding offsets
+    top = (target_h - new_h) // 2
+    left = (target_w - new_w) // 2
+
+    # Place resized image in the center
+    padded_image[top:top + new_h, left:left + new_w] = resized_image
+
+    return padded_image
+
 # Core monitor function
 def monitor_feed(video_path, detection_type, model):
     if st.session_state.paused and st.session_state.last_frame is not None:
@@ -98,6 +123,8 @@ def monitor_feed(video_path, detection_type, model):
             return None, None, None, None, None
 
         frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+        # Preprocess frame to 640x640
+        frame_rgb = preprocess_image(frame_rgb, target_size=(640, 640))
         faults = detect_faults_solar(model, frame_rgb) if detection_type == "Solar Panel" else detect_faults_windmill(model, frame_rgb)
         num_anomalies = len(faults)