Update app.py

app.py

@@ -1,7 +1,6 @@
 #!/usr/bin/env python3
 """
 CICE 2.0 Healthcare Assessment Tool - Gradio Version (Fixed)
-Simplified version with working recording and upload functionality
 """
 
 import gradio as gr
@@ -254,8 +253,42 @@ def process_video_assessment(video_file):
     
     return assessment_result, summary_html, audio_path
 
+def process_recorded_video(video_data_base64):
+    """Process recorded video data directly without download/upload"""
+    
+    if not video_data_base64:
+        return "❌ No video data received", "", None
+    
+    try:
+        # Decode base64 video data
+        video_bytes = base64.b64decode(video_data_base64)
+        
+        # Save to temporary file
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".webm") as tmp_file:
+            tmp_file.write(video_bytes)
+            temp_video_path = tmp_file.name
+        
+        # Process the video using our assessment tool
+        assessor = CICEAssessment()
+        assessment_result = assessor.analyze_video(temp_video_path)
+        
+        if assessment_result.startswith("❌"):
+            return assessment_result, "", None
+        
+        # Create summary and audio
+        summary_html = assessor.create_summary_report(assessment_result)
+        audio_path = assessor.generate_audio_feedback(assessment_result)
+        
+        # Clean up temp file
+        os.unlink(temp_video_path)
+        
+        return assessment_result, summary_html, audio_path
+        
+    except Exception as e:
+        return f"❌ Error processing recorded video: {str(e)}", "", None
+
 def create_video_recorder_html():
-    """Create the HTML/JavaScript for video recording interface - SIMPLIFIED VERSION"""
+    """Create the HTML/JavaScript for video recording interface with direct analysis"""
     
     recorder_html = """
     <div id="video-recorder-container" style="text-align: center; padding: 20px; background-color: #f0f0f0; border-radius: 10px;">
@@ -264,11 +297,16 @@ def create_video_recorder_html():
         <div style="margin: 15px 0;">
             <button id="start" class="btn" style="background-color: #4CAF50; color: white; padding: 10px 20px; margin: 5px; border: none; border-radius: 4px; cursor: pointer; font-size: 16px;">🔴 Start Recording</button>
             <button id="stop" class="btn" disabled style="background-color: #f44336; color: white; padding: 10px 20px; margin: 5px; border: none; border-radius: 4px; cursor: pointer; opacity: 0.5; font-size: 16px;">⏹️ Stop Recording</button>
+            <button id="analyze" class="btn" disabled style="background-color: #FF9800; color: white; padding: 10px 20px; margin: 5px; border: none; border-radius: 4px; cursor: pointer; opacity: 0.5; font-size: 16px;">🔍 Analyze Recording</button>
             <button id="reset" class="btn" style="background-color: #2196F3; color: white; padding: 10px 20px; margin: 5px; border: none; border-radius: 4px; cursor: pointer; font-size: 16px;">🔄 Reset</button>
         </div>
         <div id="timer" style="margin-top: 10px; font-size: 18px; font-weight: bold; color: #333;">Ready to record</div>
         <div id="status" style="margin-top: 10px; color: #666; font-size: 14px;"></div>
         <div id="fileStatus" style="margin-top: 10px; color: #0066cc; font-weight: bold;"></div>
+        
+        <!-- Hidden elements for Gradio communication -->
+        <input type="hidden" id="videoDataInput" />
+        <button id="hiddenAnalyzeBtn" style="display: none;"></button>
     </div>
 
     <script>
@@ -276,10 +314,13 @@ def create_video_recorder_html():
         var video = document.querySelector('#video');
         var startBtn = document.querySelector('#start');
         var stopBtn = document.querySelector('#stop');
+        var analyzeBtn = document.querySelector('#analyze');
         var resetBtn = document.querySelector('#reset');
         var timerDiv = document.querySelector('#timer');
         var statusDiv = document.querySelector('#status');
         var fileStatusDiv = document.querySelector('#fileStatus');
+        var videoDataInput = document.querySelector('#videoDataInput');
+        var hiddenAnalyzeBtn = document.querySelector('#hiddenAnalyzeBtn');
         
         var mediaRecorder;
         var recordedBlobs = [];
@@ -287,6 +328,7 @@ def create_video_recorder_html():
         var startTime;
         var timerInterval;
         var recordingCount = 0;
+        var recordedVideoBlob = null;
 
         // Clear any previous status
         fileStatusDiv.textContent = '';
@@ -346,6 +388,7 @@ def create_video_recorder_html():
 
             // Clear previous recording data
             recordedBlobs = [];
+            recordedVideoBlob = null;
             recordingCount++;
 
             fileStatusDiv.textContent = '📹 Recording in progress...';
@@ -378,6 +421,8 @@ def create_video_recorder_html():
             startBtn.style.opacity = '0.5';
             stopBtn.disabled = false;
             stopBtn.style.opacity = '1';
+            analyzeBtn.disabled = true;
+            analyzeBtn.style.opacity = '0.5';
 
             startTime = Date.now();
             timerInterval = setInterval(updateTimer, 1000);
@@ -393,48 +438,30 @@ def create_video_recorder_html():
             mediaRecorder.onstop = function() {
                 clearInterval(timerInterval);
                 timerDiv.textContent = '⏹️ Recording stopped - Processing...';
-                statusDiv.textContent = 'Processing and preparing video for download...';
-                fileStatusDiv.textContent = '💾 Creating video file...';
+                statusDiv.textContent = 'Processing video for analysis...';
+                fileStatusDiv.textContent = '💾 Preparing video for analysis...';
                 fileStatusDiv.style.color = '#0066cc';
 
                 // Process the recorded video
                 setTimeout(function() {
                     try {
                         // Combine all recorded chunks into a single blob
-                        var blob = new Blob(recordedBlobs, {type: 'video/webm'});
-                        console.log('Total video size:', blob.size, 'bytes');
-
-                        // Create download link
-                        var downloadUrl = URL.createObjectURL(blob);
-                        var downloadLink = document.createElement('a');
-                        downloadLink.href = downloadUrl;
-                        downloadLink.download = 'recorded_video.webm';
-                        downloadLink.style.display = 'none';
-                        document.body.appendChild(downloadLink);
-                        
+                        recordedVideoBlob = new Blob(recordedBlobs, {type: 'video/webm'});
+                        console.log('Total video size:', recordedVideoBlob.size, 'bytes');
+
                         timerDiv.textContent = '✅ Recording complete!';
-                        statusDiv.innerHTML = '<strong>✅ Recording successful!</strong><br>Your video will download automatically.';
+                        statusDiv.textContent = '✅ Recording successful! Click "Analyze Recording" to start assessment.';
                         statusDiv.style.color = 'green';
                         fileStatusDiv.innerHTML = '✅ Video recorded successfully (' + 
-                            (blob.size / 1024 / 1024).toFixed(2) + ' MB)<br>' +
-                            '<span style="color: #0066cc;">⬇️ Auto-downloading...</span><br>' +
-                            '<span style="color: #666; font-size: 12px;">After download, use the "Upload Video & Assess" tab to analyze.</span>';
+                            (recordedVideoBlob.size / 1024 / 1024).toFixed(2) + ' MB)<br>' +
+                            '<span style="color: #0066cc;">Ready for CICE 2.0 analysis!</span>';
                         fileStatusDiv.style.color = 'green';
 
-                        // Auto-download the video file
-                        setTimeout(function() {
-                            downloadLink.click();
-                            document.body.removeChild(downloadLink);
-                            URL.revokeObjectURL(downloadUrl);
-                            
-                            // Update status after download
-                            setTimeout(function() {
-                                fileStatusDiv.innerHTML = '✅ Video downloaded as "recorded_video.webm"<br>' +
-                                    '<span style="color: #0066cc;">📁 Go to "Upload Video & Assess" tab and upload your downloaded video.</span>';
-                            }, 1000);
-                        }, 500);
-
-                        // Clear recorded data to free memory
+                        // Enable analyze button
+                        analyzeBtn.disabled = false;
+                        analyzeBtn.style.opacity = '1';
+
+                        // Clear recorded blobs to free memory
                         recordedBlobs = [];
                     } catch (e) {
                         console.error('Error processing video:', e);
@@ -463,17 +490,49 @@ def create_video_recorder_html():
             }
         });
 
+        analyzeBtn.addEventListener('click', function() {
+            if (!recordedVideoBlob) {
+                alert('No recording available. Please record a video first.');
+                return;
+            }
+
+            analyzeBtn.disabled = true;
+            analyzeBtn.style.opacity = '0.5';
+            fileStatusDiv.textContent = '🤖 Starting CICE 2.0 analysis...';
+            fileStatusDiv.style.color = '#ff6600';
+
+            // Convert blob to base64
+            var reader = new FileReader();
+            reader.onload = function() {
+                var base64Data = reader.result.split(',')[1]; // Remove data:video/webm;base64, prefix
+                console.log('Video converted to base64, length:', base64Data.length);
+                
+                // Set the video data in hidden input
+                videoDataInput.value = base64Data;
+                
+                // Trigger the hidden Gradio button
+                hiddenAnalyzeBtn.click();
+                
+                fileStatusDiv.textContent = '🤖 Analysis in progress... This may take 1-2 minutes.';
+            };
+            reader.readAsDataURL(recordedVideoBlob);
+        });
+
         resetBtn.addEventListener('click', function() {
             // Reset the interface
             if (mediaRecorder && mediaRecorder.state === 'recording') {
                 mediaRecorder.stop();
             }
             recordedBlobs = [];
+            recordedVideoBlob = null;
+            videoDataInput.value = '';
             
             startBtn.disabled = false;
             startBtn.style.opacity = '1';
             stopBtn.disabled = true;
             stopBtn.style.opacity = '0.5';
+            analyzeBtn.disabled = true;
+            analyzeBtn.style.opacity = '0.5';
             timerDiv.textContent = 'Ready to record';
             statusDiv.textContent = '✅ Camera and microphone ready! Click Start to begin recording.';
             statusDiv.style.color = '#666';
@@ -566,20 +625,20 @@ def create_gradio_app():
         - 💬 Interactive Q&A about results
         
         **📝 Workflow:**
-        1. **Record Tab:** Record your video → Video auto-downloads
-        2. **Upload Tab:** Upload the downloaded video → Analyze with CICE 2.0
+        1. **Record & Assess Tab:** Record video → Click "Analyze Recording" → Get immediate CICE 2.0 assessment
+        2. **Upload Tab:** Upload existing videos → Analyze with CICE 2.0
         3. **Q&A Tab:** Ask specific questions about the assessment
         """)
         
-        with gr.Tab("🎥 Record Video"):
+        with gr.Tab("🎥 Record Video & Assess"):
             gr.Markdown("""
             ### Record Live Healthcare Team Interaction
             
             **Instructions:**
             1. Click "Start Recording" to begin recording
             2. Click "Stop Recording" when finished  
-            3. The video will automatically download as 'recorded_video.webm'
-            4. Go to the "📁 Upload Video & Assess" tab to upload and analyze your recording
+            3. Click "Analyze Recording" to automatically analyze with CICE 2.0
+            4. View your assessment results below
             """)
             
             # Video recorder interface
@@ -587,6 +646,33 @@ def create_gradio_app():
                 value=create_video_recorder_html(),
                 label="Video Recorder"
             )
+            
+            # Hidden components for JavaScript communication
+            with gr.Row(visible=False):
+                video_data_hidden = gr.Textbox(elem_id="videoDataInput")
+                analyze_hidden_btn = gr.Button("Hidden Analyze", elem_id="hiddenAnalyzeBtn")
+            
+            # Assessment results from recording
+            with gr.Row():
+                with gr.Column(scale=2):
+                    recording_summary_output = gr.HTML(
+                        label="Recording Assessment Summary",
+                        value="<p>Record a video and click 'Analyze Recording' to see the summary.</p>"
+                    )
+            
+            with gr.Row():
+                recording_assessment_output = gr.Textbox(
+                    label="Detailed Recording Assessment Report",
+                    lines=20,
+                    max_lines=30,
+                    placeholder="Detailed CICE 2.0 assessment will appear here..."
+                )
+            
+            with gr.Row():
+                recording_audio_output = gr.Audio(
+                    label="🔊 Audio Feedback from Recording",
+                    visible=True
+                )
         
         with gr.Tab("📁 Upload Video & Assess"):
             gr.Markdown("### Upload Healthcare Team Video for CICE 2.0 Analysis")
@@ -705,6 +791,13 @@ def create_gradio_app():
             outputs=[assessment_output, summary_output, audio_output]
         )
         
+        # Event handler for recorded video analysis
+        analyze_hidden_btn.click(
+            fn=lambda video_data: store_assessment_state(None, *process_recorded_video(video_data)),
+            inputs=[video_data_hidden],
+            outputs=[recording_assessment_output, recording_summary_output, recording_audio_output]
+        )
+        
         # Q&A handlers
         ask_btn.click(
             fn=qa_wrapper,