Update app.py

app.py

@@ -5,6 +5,7 @@ import cv2
 import tempfile
 from ultralytics import YOLO
 import numpy as np
+from datetime import datetime
 
 # Azure Storage config
 SAS_TOKEN = "sv=2024-11-04&ss=bfqt&srt=sco&sp=rwdlacupiytfx&se=2025-04-30T04:25:22Z&st=2025-04-16T20:25:22Z&spr=https&sig=HYrJBoOYc4PRe%2BoqBMl%2FmoL5Kz4ZYugbTLuEh63sbeo%3D"
@@ -26,8 +27,29 @@ def list_videos():
     except Exception as e:
         return [f"Error: {str(e)}"]
 
+def get_latest_video():
+    try:
+        blob_service_client = BlobServiceClient(account_url=f"https://{ACCOUNT_NAME}.blob.core.windows.net", credential=SAS_TOKEN)
+        container_client = blob_service_client.get_container_client(CONTAINER_NAME)
+        blobs = container_client.list_blobs(name_starts_with=VIDEOS_FOLDER)
+        latest_blob = None
+        latest_time = None
+        for blob in blobs:
+            if blob.name.endswith(".mp4"):
+                blob_client = container_client.get_blob_client(blob.name)
+                props = blob_client.get_blob_properties()
+                if not latest_time or props.last_modified > latest_time:
+                    latest_time = props.last_modified
+                    latest_blob = blob.name
+        return latest_blob if latest_blob else "No videos found"
+    except Exception as e:
+        return f"Error: {str(e)}"
+
 def annotate_video_with_bboxes(video_path):
     try:
+        if not video_path or video_path == "No videos found":
+            return "Error: No video selected"
+        
         # Download video from Azure
         blob_service_client = BlobServiceClient(account_url=f"https://{ACCOUNT_NAME}.blob.core.windows.net", credential=SAS_TOKEN)
         blob_client = blob_service_client.get_blob_client(CONTAINER_NAME, video_path)
@@ -40,6 +62,10 @@ def annotate_video_with_bboxes(video_path):
         
         # Process video with YOLO
         cap = cv2.VideoCapture(temp_input.name)
+        if not cap.isOpened():
+            os.remove(temp_input.name)
+            return "Error: Could not open video"
+        
         fps = cap.get(cv2.CAP_PROP_FPS)
         w = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
         h = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
@@ -86,10 +112,12 @@ def annotate_video_with_bboxes(video_path):
 # Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("## PRISM Site Diary V3 - Video Annotator")
-    video_dropdown = gr.Dropdown(label="Select Video from Azure Blob", choices=list_videos(), interactive=True)
-    annotate_btn = gr.Button("Annotate Video")
+    video_dropdown = gr.Dropdown(label="Select Video from Azure Blob", choices=list_videos(), interactive=True, allow_custom_value=True)
+    auto_btn = gr.Button("Annotate Latest Video")
+    manual_btn = gr.Button("Annotate Selected Video")
     output_video = gr.Video(label="Annotated Video")
-    annotate_btn.click(fn=annotate_video_with_bboxes, inputs=video_dropdown, outputs=output_video)
+    auto_btn.click(fn=annotate_video_with_bboxes, inputs=gr.State(get_latest_video()), outputs=output_video)
+    manual_btn.click(fn=annotate_video_with_bboxes, inputs=video_dropdown, outputs=output_video)
     video_dropdown.change(fn=list_videos, outputs=video_dropdown)
 
 demo.launch()