Update app.py

app.py

@@ -1,9 +1,10 @@
 import spaces
 
-# Add this dummy function (not recommended)
-@spaces.GPU
-def dummy_gpu_function():
-    pass
+# Configure GPU L405
+@spaces.GPU(duration=120)
+def process_video_with_gpu(video, resize_option, assessor):
+    """GPU-accelerated video processing"""
+    return process_video_core(video, resize_option, assessor)
 
 import gradio as gr
 from google import genai
@@ -452,8 +453,31 @@ def get_video_info(video_path):
     except:
         return None, None, None, None
 
-# Define the processing function
-def process_video(video, resize_option):
+# Function to save video immediately after recording
+def save_recorded_video(video):
+    """Save the recorded video and make it downloadable"""
+    if video is None:
+        return None
+    
+    try:
+        # Create a copy of the video file with a timestamp
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        output_filename = f"recorded_video_{timestamp}.mp4"
+        temp_output = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4', prefix=f"recorded_{timestamp}_")
+        
+        # Copy the video file
+        import shutil
+        shutil.copy2(video, temp_output.name)
+        temp_output.close()
+        
+        print(f"📹 Video saved: {output_filename}")
+        return temp_output.name
+    except Exception as e:
+        print(f"⚠️ Failed to save video: {str(e)}")
+        return None
+
+# Define the core processing function (separate from GPU wrapper)
+def process_video_core(video, resize_option, assessor):
     """Process uploaded or recorded video"""
 
     if video is None:
@@ -589,6 +613,11 @@ def process_video(video, resize_option):
         print(error_msg)
         return error_msg, None, None, None
 
+# Wrapper function that calls the GPU-accelerated version
+def process_video(video, resize_option):
+    """Wrapper function to call GPU-accelerated processing"""
+    return process_video_with_gpu(video, resize_option, assessor)
+
 # Create and launch the Gradio interface
 print("🚀 Launching CICE 2.0 Healthcare Assessment Tool...")
 
@@ -631,19 +660,27 @@ with gr.Blocks(title="CICE 2.0 Healthcare Assessment Tool", theme=gr.themes.Soft
                 format="mp4",
                 include_audio=True,
                 interactive=True,
-                webcam_constraints={"video": {"width": 800, "height": 600}}
+                webcam_constraints={"video": {"width": 800, "height": 600}},
+                autoplay=False,  # Disable autoplay for faster loading
+                show_download_button=True  # Show download button immediately
             )
 
-            analyze_btn = gr.Button("🔍 Analyze Video", variant="primary", size="lg")
+            # Add download component for recorded videos
+            recorded_video_download = gr.File(
+                label="📥 Download Recorded Video",
+                interactive=False,
+                visible=False
+            )
 
             gr.Markdown("""
             ### 📝 Instructions:
             1. **Select video resolution** (lower = faster processing)
             2. **Upload** a pre-recorded video or **Record** using your webcam
-            3. Click **Analyze Video** to start the assessment
-            4. Wait for the AI to process (1-2 minutes)
-            5. Listen to the **1-minute audio summary**
-            6. Download the **PDF report** for documentation
+            3. Video will be saved and downloadable immediately after recording stops
+            4. Click **Analyze Video** (on the right) to start the assessment
+            5. Wait for the AI to process (1-2 minutes)
+            6. Listen to the **1-minute audio summary**
+            7. Download the **PDF report** for documentation
 
             **Video Resolution Guide:**
             - **640x480**: Fastest processing, uses least quota
@@ -663,6 +700,9 @@ with gr.Blocks(title="CICE 2.0 Healthcare Assessment Tool", theme=gr.themes.Soft
         with gr.Column(scale=2):
             gr.Markdown("### 📊 Assessment Results")
 
+            # Move analyze button here (to the right column)
+            analyze_btn = gr.Button("🔍 Analyze Video", variant="primary", size="lg")
+
             # Visual summary
             summary_output = gr.HTML(
                 label="Visual Summary",
@@ -707,9 +747,21 @@ with gr.Blocks(title="CICE 2.0 Healthcare Assessment Tool", theme=gr.themes.Soft
     - 📊 1-minute audio summary (MP3 format)
     - 📄 Complete PDF assessment report
 
-    **Powered by Google Gemini 2.0 Flash**
+    **Powered by Google Gemini 2.0 Flash | GPU-Accelerated with HuggingFace L405**
     """)
 
+    # Auto-save video when recording stops
+    video_input.stop_recording(
+        fn=save_recorded_video,
+        inputs=[video_input],
+        outputs=[recorded_video_download],
+        api_name="save_video"
+    ).then(
+        fn=lambda x: gr.update(visible=True if x else False),
+        inputs=[recorded_video_download],
+        outputs=[recorded_video_download]
+    )
+
     # Connect the analyze button
     analyze_btn.click(
         fn=process_video,