latest changes

app.py

@@ -12,13 +12,13 @@ from collections import deque
 HF_MODEL_REPO_ID = "owinymarvin/timesformer-crime-detection"
 
 # These must match the values used during your training
-NUM_FRAMES = 8 # Changed back to 8 as that was your original training setup for this model
+NUM_FRAMES = 8 # Still using 8 frames, as that was your original training setup
 TARGET_IMAGE_HEIGHT = 224
 TARGET_IMAGE_WIDTH = 224
 
 # --- Prediction Timing ---
 # How long to record (in seconds) before making a prediction
-RECORDING_DURATION_SECONDS = 3.0
+RECORDING_DURATION_SECONDS = 10.0 # CHANGED: Now records for 10 seconds
 # How often the model should predict (after the recording duration)
 # Setting this to a very high number (like 9999) means it essentially predicts only once
 # after the recording is done until reset. Or you can leave it at 1.0 if you want it to trigger often.
@@ -105,6 +105,7 @@ def reset_app_state():
     recording_start_time = None
     last_prediction_time = time.time()
     print("App state reset.")
+    # Return initial messages for the UI
     return "Ready to record...", "Ready for new prediction."
 
 # --- Gradio Interface ---
@@ -115,6 +116,7 @@ with gr.Blocks() as demo:
         This demo uses a finetuned TimesFormer model ({HF_MODEL_REPO_ID}) to predict crime actions from a live webcam feed.
         It records **{RECORDING_DURATION_SECONDS} seconds** of video, then automatically triggers a prediction.
         The model processes **{NUM_FRAMES} frames** per prediction.
+        Click 'Reset' to start a new video recording.
         Please allow webcam access.
         """
     )
@@ -129,7 +131,7 @@ with gr.Blocks() as demo:
             status_output = gr.Textbox(label="Status", value="Ready to record...")
             
             # Reset Button
-            reset_button = gr.Button("Reset / Start New Recording Cycle")
+            reset_button = gr.Button("Reset / Start New Video")
             
         with gr.Column():
             prediction_output = gr.Textbox(label="Prediction Result", value="Recording will start automatically.")