Update app.py

app.py

@@ -14,19 +14,14 @@ import os
 mp_face_detection = mp.solutions.face_detection
 
 def convert_video_if_needed(video_path):
-    """Convert video to mp4 if it's in unsupported formats."""
-    if video_path.lower().endswith('.avi') or video_path.lower().endswith('.mov'):
+    """Convert video to mp4 if it's in avi format"""
+    if video_path.lower().endswith('.avi'):
         output_path = video_path.rsplit('.', 1)[0] + '.mp4'
         try:
-            cap = cv2.VideoCapture(video_path, cv2.CAP_FFMPEG)  # Explicit backend
-            if not cap.isOpened():
-                raise ValueError("Cannot open the video file for conversion.")
-                
+            cap = cv2.VideoCapture(video_path)
             fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-            fps = cap.get(cv2.CAP_PROP_FPS) or 30.0  # Default to 30 fps if not found
-            width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
-            height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
-            out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
+            out = cv2.VideoWriter(output_path, fourcc, 30.0, 
+                                (int(cap.get(3)), int(cap.get(4))))
             
             while cap.isOpened():
                 ret, frame = cap.read()
@@ -39,18 +34,23 @@ def convert_video_if_needed(video_path):
             return output_path
         except Exception as e:
             print(f"Error converting video: {e}")
-            return video_path  # Fallback to the original path
+            return video_path
     return video_path
 
 def process_video(video_path):
-    """Process video to calculate respiration rate."""
-    cap = cv2.VideoCapture(video_path, cv2.CAP_FFMPEG)  # Explicit backend
+    """Process video to calculate respiration rate"""
+    cap = cv2.VideoCapture(video_path, cv2.CAP_FFMPEG)
     if not cap.isOpened():
         raise ValueError("Error opening video file. Please check the format or path.")
 
     points = (0, 0, 0, 0)
     fps = cap.get(cv2.CAP_PROP_FPS)
+    if not fps or fps == 0:
+        fps = 30.0  # Default FPS
+
     num_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+    if not num_frames or num_frames <= 0:
+        num_frames = 1  # Avoid division by zero
 
     # Detect face for the first frame
     with mp_face_detection.FaceDetection(model_selection=1, min_detection_confidence=0.5) as face_detection:
@@ -89,7 +89,7 @@ def process_video(video_path):
 
     cap.release()
 
-    # Apply Band-Pass Filter
+    # Apply Band-Pass-Filter
     data = val_list
     lowcut = 0.16
     highcut = 0.5
@@ -103,7 +103,10 @@ def process_video(video_path):
 
     # Find peaks and calculate metrics
     peaks, _ = find_peaks(filtered)
-    time_length = num_frames / fps
+    time_length = num_frames / fps if fps > 0 else 0.0
+    if time_length == 0.0:
+        raise ValueError("Unable to determine video duration.")
+
     respiration_times = peaks / fps
     rr = math.ceil((len(peaks) / time_length) * 60)
 
@@ -134,7 +137,7 @@ def process_video(video_path):
     }
 
 def process_input(video_path):
-    """Handle input video processing with error handling."""
+    """Handle input video processing with error handling"""
     try:
         if video_path is None:
             return ["Please upload a video file."] * 5 + [None]
@@ -176,7 +179,30 @@ interface = gr.Interface(
     title="Respiration Rate Analyzer",
     description="🫁 Analyze your respiration rate with ease!",
     theme="default",
-    css="""body { background-color: #001f3f; color: #f0f8ff; }"""
+    css="""
+        body {
+            background-color: #001f3f;
+            color: #f0f8ff;
+        }
+        .output {
+            font-size: 16px;
+            color: #f0f8ff;
+        }
+        .label {
+            font-weight: bold;
+            color: #0074D9;
+        }
+        .plot {
+            border: 2px solid #0074D9;
+            border-radius: 10px;
+        }
+        .gradio-container {
+            background-color: #002b57;
+        }
+        .output-markdown p {
+            color: #f0f8ff !important;
+        }
+    """
 )
 
 interface.launch()