Update app.py

app.py

@@ -7,6 +7,12 @@ import re
 from gtts import gTTS
 import tempfile
 import base64
+from streamlit_webrtc import webrtc_streamer, WebRtcMode, RTCConfiguration
+import av
+import numpy as np
+import cv2
+import threading
+import queue
 
 # Page configuration
 st.set_page_config(
@@ -174,6 +180,45 @@ def parse_assessment_score(assessment_text):
         pass
     return 0, 0, "Unknown", "#6b7280"
 
+# Video recording state
+if "recording" not in st.session_state:
+    st.session_state.recording = False
+if "recorded_frames" not in st.session_state:
+    st.session_state.recorded_frames = []
+if "frame_queue" not in st.session_state:
+    st.session_state.frame_queue = queue.Queue()
+
+def video_frame_callback(frame):
+    """Callback function to process video frames during recording"""
+    if st.session_state.recording:
+        img = frame.to_ndarray(format="bgr24")
+        st.session_state.frame_queue.put(img)
+    return frame
+
+def save_recorded_video():
+    """Save recorded frames to a video file"""
+    if st.session_state.recorded_frames:
+        # Create temporary file
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4')
+        temp_path = temp_file.name
+        temp_file.close()
+        
+        # Video properties
+        height, width, _ = st.session_state.recorded_frames[0].shape
+        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+        fps = 20  # frames per second
+        
+        # Create video writer
+        out = cv2.VideoWriter(temp_path, fourcc, fps, (width, height))
+        
+        # Write frames
+        for frame in st.session_state.recorded_frames:
+            out.write(frame)
+        
+        out.release()
+        return temp_path
+    return None
+
 def main():
     # Header
     st.markdown("""
@@ -224,72 +269,126 @@ def main():
     col1, col2 = st.columns([2, 1])
     
     with col1:
-        st.header("📹 Upload Healthcare Team Video")
+        st.header("📹 Record or Upload Healthcare Team Video")
         
-        uploaded_file = st.file_uploader(
-            "Choose a video file",
-            type=['mp4', 'webm', 'avi', 'mov'],
-            help="Upload a video of healthcare team interaction for CICE 2.0 assessment"
-        )
+        # Tab selection for recording vs uploading
+        tab1, tab2 = st.tabs(["🎥 Record Video", "📁 Upload Video"])
         
-        if uploaded_file is not None:
-            st.success(f"✅ Video uploaded: {uploaded_file.name}")
+        with tab1:
+            st.subheader("Live Video Recording")
+            
+            # WebRTC configuration
+            rtc_configuration = RTCConfiguration({
+                "iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]
+            })
+            
+            # Recording controls
+            col_start, col_stop, col_status = st.columns([1, 1, 2])
+            
+            with col_start:
+                if st.button("🔴 Start Recording", disabled=st.session_state.recording):
+                    st.session_state.recording = True
+                    st.session_state.recorded_frames = []
+                    # Clear the queue
+                    while not st.session_state.frame_queue.empty():
+                        st.session_state.frame_queue.get()
+                    st.rerun()
             
-            # Display video
-            st.video(uploaded_file)
+            with col_stop:
+                if st.button("⏹️ Stop Recording", disabled=not st.session_state.recording):
+                    st.session_state.recording = False
+                    
+                    # Collect frames from queue
+                    frames_collected = []
+                    while not st.session_state.frame_queue.empty():
+                        frame = st.session_state.frame_queue.get()
+                        frames_collected.append(frame)
+                    
+                    st.session_state.recorded_frames = frames_collected
+                    st.rerun()
             
-            # Analyze button
-            if st.button("🔍 Analyze with CICE 2.0", type="primary"):
-                if not api_key:
-                    st.error("❌ Please enter your Google Gemini API key in the sidebar")
+            with col_status:
+                if st.session_state.recording:
+                    st.markdown("🔴 **Recording...**")
+                elif st.session_state.recorded_frames:
+                    st.markdown(f"✅ **Recorded {len(st.session_state.recorded_frames)} frames**")
                 else:
-                    try:
-                        assessor = CICE_Assessment(api_key)
-                        
-                        with st.spinner("🤖 Analyzing video with CICE 2.0 framework... This may take 1-2 minutes"):
-                            # Reset file pointer
-                            uploaded_file.seek(0)
-                            assessment_result = assessor.analyze_video(uploaded_file)
-                        
-                        # Parse score
-                        observed, percentage, level, color = parse_assessment_score(assessment_result)
-                        
-                        # Display summary
-                        st.markdown(f"""
-                        <div class="score-display">
-                            <h2>CICE 2.0 Assessment Results</h2>
-                            <div style="display: flex; justify-content: space-around; margin: 30px 0;">
-                                <div style="text-align: center;">
-                                    <div style="font-size: 48px; font-weight: bold; color: {color};">{observed}/18</div>
-                                    <div style="color: #6b7280;">Competencies Observed</div>
-                                </div>
-                                <div style="text-align: center;">
-                                    <div style="font-size: 48px; font-weight: bold; color: {color};">{percentage:.0f}%</div>
-                                    <div style="color: #6b7280;">Overall Score</div>
-                                </div>
-                            </div>
-                            <div style="text-align: center; padding: 20px; background: #f9fafb; border-radius: 10px;">
-                                <div style="font-size: 24px; font-weight: bold; color: {color};">Performance Level: {level}</div>
-                            </div>
-                        </div>
-                        """, unsafe_allow_html=True)
-                        
-                        # Display detailed assessment
-                        st.markdown('<div class="assessment-box">', unsafe_allow_html=True)
-                        st.markdown("### 📋 Detailed Assessment Report")
-                        st.write(assessment_result)
-                        st.markdown('</div>', unsafe_allow_html=True)
-                        
-                        # Download option
-                        st.download_button(
-                            label="📥 Download Assessment Report",
-                            data=assessment_result,
-                            file_name=f"cice_assessment_{datetime.now().strftime('%Y%m%d_%H%M%S')}.txt",
-                            mime="text/plain"
-                        )
-                        
-                    except Exception as e:
-                        st.error(f"❌ Error during assessment: {str(e)}")
+                    st.markdown("⚪ **Ready to record**")
+            
+            # WebRTC streamer
+            webrtc_ctx = webrtc_streamer(
+                key="video-recorder",
+                mode=WebRtcMode.SENDONLY,
+                rtc_configuration=rtc_configuration,
+                video_frame_callback=video_frame_callback,
+                media_stream_constraints={
+                    "video": {
+                        "width": {"ideal": 640, "max": 640},
+                        "height": {"ideal": 480, "max": 480},
+                        "frameRate": {"ideal": 20, "max": 30}
+                    },
+                    "audio": False  # Disable audio for now
+                },
+                async_processing=True,
+            )
+            
+            # Process recorded video
+            if st.session_state.recorded_frames and not st.session_state.recording:
+                st.success(f"✅ Recording complete! Captured {len(st.session_state.recorded_frames)} frames")
+                
+                if st.button("💾 Save & Analyze Video"):
+                    with st.spinner("Saving video..."):
+                        video_path = save_recorded_video()
+                        if video_path:
+                            st.success("Video saved successfully!")
+                            
+                            # Read the saved video for analysis
+                            with open(video_path, 'rb') as f:
+                                video_bytes = f.read()
+                            
+                            # Create a file-like object for analysis
+                            class VideoFile:
+                                def __init__(self, data, name):
+                                    self.data = data
+                                    self.name = name
+                                
+                                def read(self):
+                                    return self.data
+                                
+                                def seek(self, position):
+                                    pass
+                            
+                            uploaded_file = VideoFile(video_bytes, "recorded_video.mp4")
+                            
+                            # Proceed with analysis
+                            if not api_key:
+                                st.error("❌ Please enter your Google Gemini API key in the sidebar")
+                            else:
+                                analyze_video(uploaded_file, api_key)
+                            
+                            # Clean up temporary file
+                            os.unlink(video_path)
+        
+        with tab2:
+            st.subheader("Upload Video File")
+            uploaded_file = st.file_uploader(
+                "Choose a video file",
+                type=['mp4', 'webm', 'avi', 'mov'],
+                help="Upload a video of healthcare team interaction for CICE 2.0 assessment"
+            )
+            
+            if uploaded_file is not None:
+                st.success(f"✅ Video uploaded: {uploaded_file.name}")
+                
+                # Display video
+                st.video(uploaded_file)
+                
+                # Analyze button
+                if st.button("🔍 Analyze with CICE 2.0", type="primary"):
+                    if not api_key:
+                        st.error("❌ Please enter your Google Gemini API key in the sidebar")
+                    else:
+                        analyze_video(uploaded_file, api_key)
     
     with col2:
         st.header("ℹ️ About CICE 2.0")
@@ -316,5 +415,55 @@ def main():
         5. Download assessment report
         """)
 
+def analyze_video(uploaded_file, api_key):
+    """Analyze uploaded video with CICE 2.0 assessment"""
+    try:
+        assessor = CICE_Assessment(api_key)
+        
+        with st.spinner("🤖 Analyzing video with CICE 2.0 framework... This may take 1-2 minutes"):
+            # Reset file pointer
+            uploaded_file.seek(0)
+            assessment_result = assessor.analyze_video(uploaded_file)
+        
+        # Parse score
+        observed, percentage, level, color = parse_assessment_score(assessment_result)
+        
+        # Display summary
+        st.markdown(f"""
+        <div class="score-display">
+            <h2>CICE 2.0 Assessment Results</h2>
+            <div style="display: flex; justify-content: space-around; margin: 30px 0;">
+                <div style="text-align: center;">
+                    <div style="font-size: 48px; font-weight: bold; color: {color};">{observed}/18</div>
+                    <div style="color: #6b7280;">Competencies Observed</div>
+                </div>
+                <div style="text-align: center;">
+                    <div style="font-size: 48px; font-weight: bold; color: {color};">{percentage:.0f}%</div>
+                    <div style="color: #6b7280;">Overall Score</div>
+                </div>
+            </div>
+            <div style="text-align: center; padding: 20px; background: #f9fafb; border-radius: 10px;">
+                <div style="font-size: 24px; font-weight: bold; color: {color};">Performance Level: {level}</div>
+            </div>
+        </div>
+        """, unsafe_allow_html=True)
+        
+        # Display detailed assessment
+        st.markdown('<div class="assessment-box">', unsafe_allow_html=True)
+        st.markdown("### 📋 Detailed Assessment Report")
+        st.write(assessment_result)
+        st.markdown('</div>', unsafe_allow_html=True)
+        
+        # Download option
+        st.download_button(
+            label="📥 Download Assessment Report",
+            data=assessment_result,
+            file_name=f"cice_assessment_{datetime.now().strftime('%Y%m%d_%H%M%S')}.txt",
+            mime="text/plain"
+        )
+        
+    except Exception as e:
+        st.error(f"❌ Error during assessment: {str(e)}")
+
 if __name__ == "__main__":
     main()