Update app.py

app.py

@@ -7,6 +7,7 @@ import re
 from gtts import gTTS
 import tempfile
 import numpy as np
+import cv2
 
 # Configure API
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
@@ -174,6 +175,64 @@ class CICE_Assessment:
 # Initialize the assessment tool
 assessor = CICE_Assessment()
 
+def compress_video(input_path, output_path, target_width=640, target_height=480, target_fps=15, target_bitrate='500k'):
+    """Compress and resize video to reduce file size and processing time"""
+    
+    try:
+        # Open the video
+        cap = cv2.VideoCapture(input_path)
+        
+        # Get original properties
+        original_fps = cap.get(cv2.CAP_PROP_FPS)
+        original_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+        original_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+        
+        # Calculate aspect ratio and new dimensions
+        aspect_ratio = original_width / original_height
+        if aspect_ratio > target_width / target_height:
+            new_width = target_width
+            new_height = int(target_width / aspect_ratio)
+        else:
+            new_height = target_height
+            new_width = int(target_height * aspect_ratio)
+        
+        # Set up video writer with compression
+        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+        out = cv2.VideoWriter(output_path, fourcc, min(target_fps, original_fps), (new_width, new_height))
+        
+        frame_skip = max(1, int(original_fps / target_fps))
+        frame_count = 0
+        
+        while True:
+            ret, frame = cap.read()
+            if not ret:
+                break
+            
+            # Skip frames to reduce FPS
+            if frame_count % frame_skip == 0:
+                # Resize frame
+                resized_frame = cv2.resize(frame, (new_width, new_height), interpolation=cv2.INTER_AREA)
+                out.write(resized_frame)
+            
+            frame_count += 1
+        
+        cap.release()
+        out.release()
+        
+        # Get file sizes for comparison
+        original_size = os.path.getsize(input_path) / (1024 * 1024)  # MB
+        compressed_size = os.path.getsize(output_path) / (1024 * 1024)  # MB
+        
+        print(f"✅ Video compressed: {original_size:.2f}MB → {compressed_size:.2f}MB")
+        print(f"   Resolution: {original_width}x{original_height} → {new_width}x{new_height}")
+        print(f"   FPS: {original_fps:.1f} → {min(target_fps, original_fps):.1f}")
+        
+        return output_path
+    
+    except Exception as e:
+        print(f"⚠️ Compression failed, using original: {str(e)}")
+        return input_path
+
 def process_video(video):
     """Process uploaded or recorded video"""
     
@@ -186,6 +245,15 @@ def process_video(video):
     progress_messages = []
     
     try:
+        # Compress video if needed
+        file_size_mb = os.path.getsize(video) / (1024 * 1024)
+        
+        if file_size_mb > 10:  # Compress if larger than 10MB
+            progress_messages.append(f"📦 Compressing video ({file_size_mb:.1f}MB)...")
+            with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tmp_file:
+                compressed_path = tmp_file.name
+            video = compress_video(video, compressed_path)
+        
         # Start assessment
         progress_messages.append("🏥 Starting CICE 2.0 Healthcare Team Assessment...")
         
@@ -209,7 +277,7 @@ def process_video(video):
         
         # Create visual summary HTML
         summary_html = f"""
-        <div style="max-width:800px; margin:20px auto; padding:30px; border-radius:15px; box-shadow:0 4px 6px rgba(0,0,0,0.1);">
+        <div style="max-width:800px; margin:20px auto; padding:30px; background:white; border-radius:15px; box-shadow:0 4px 6px rgba(0,0,0,0.1);">
             <h2 style="text-align:center; color:#1f2937;">CICE 2.0 Assessment Summary</h2>
             
             <div style="display:flex; justify-content:space-around; margin:30px 0;">
@@ -295,7 +363,15 @@ def create_interface():
                     sources=["upload", "webcam"],
                     format="mp4",
                     include_audio=True,
-                    interactive=True
+                    interactive=True,
+                    webcam_constraints={
+                        "video": {
+                            "width": {"ideal": 640, "max": 640},
+                            "height": {"ideal": 480, "max": 480},
+                            "frameRate": {"ideal": 15, "max": 24}
+                        },
+                        "audio": True
+                    }
                 )
                 
                 analyze_btn = gr.Button("🔍 Analyze Video", variant="primary", size="lg")
@@ -306,6 +382,10 @@ def create_interface():
                 2. Click **Analyze Video** to start the assessment
                 3. Wait for the AI to process (1-2 minutes)
                 4. Review the detailed assessment and listen to audio feedback
+                
+                **Note:** Videos are automatically optimized for faster processing:
+                - Webcam recording: 640x480 @ 15fps
+                - Uploaded videos > 10MB will be compressed
                 """)
             
             with gr.Column(scale=2):