Update app.py

app.py

@@ -61,6 +61,7 @@ with col2:
 
 st.markdown("""
 Early wildfire detection using YOLOv8 AI vision model. See detected results above and video examples below, or upload your own content!
+Click on video frames to load and play examples.
 """)
 
 # Function to create synchronized video pair HTML
@@ -75,14 +76,14 @@ def create_synced_video_pair(orig_url, proc_url, pair_id):
         <div style="display: flex; gap: 10px; margin-bottom: 20px;">
             <div style="flex: 1;">
                 <h4>Original</h4>
-                <video id="orig_{pair_id}" width="100%" controls preload="metadata">
+                <video id="orig_{pair_id}" width="100%" controls>
                     <source src="data:video/mp4;base64,{orig_b64}" type="video/mp4">
                     Your browser does not support the video tag.
                 </video>
             </div>
             <div style="flex: 1;">
                 <h4>Processed</h4>
-                <video id="proc_{pair_id}" width="100%" controls preload="metadata">
+                <video id="proc_{pair_id}" width="100%" controls>
                     <source src="data:video/mp4;base64,{proc_b64}" type="video/mp4">
                     Your browser does not support the video tag.
                 </video>
@@ -196,16 +197,19 @@ if process_button and source_file and model:
                 elapsed = time.time() - st.session_state.start_time
                 progress = frame_count / total_frames
                 
-                if elapsed > 0 and processed_count > 0:
-                    time_per_frame = elapsed / processed_count
-                    frames_left = (total_frames - frame_count) / sample_interval
-                    eta = frames_left * time_per_frame
-                    eta_str = f"{int(eta // 60)}m {int(eta % 60)}s"
+                if elapsed > 0 and progress > 0:
+                    # Total estimated time = elapsed time / progress
+                    total_estimated_time = elapsed / progress
+                    # ETA = total estimated time - elapsed time
+                    eta = total_estimated_time - elapsed
+                    elapsed_str = f"{int(elapsed // 60)}m {int(elapsed % 60)}s"
+                    eta_str = f"{int(eta // 60)}m {int(eta % 60)}s" if eta > 0 else "Almost done"
                 else:
+                    elapsed_str = "0s"
                     eta_str = "Calculating..."
                 
                 progress_bar.progress(min(progress, 1.0))
-                progress_text.text(f"Progress: {progress:.1%} | ETA: {eta_str}")
+                progress_text.text(f"Progress: {progress:.1%} | Elapsed: {elapsed_str} | ETA: {eta_str}")
             
             frame_count += 1
             success, frame = vidcap.read()
@@ -233,8 +237,10 @@ if process_button and source_file and model:
                 st.session_state.processed_video = f.read()
             os.unlink(out_path)
             
+            elapsed_final = time.time() - st.session_state.start_time
+            elapsed_final_str = f"{int(elapsed_final // 60)}m {int(elapsed_final % 60)}s"
             progress_bar.progress(1.0)
-            progress_text.text("Processing complete!")
+            progress_text.text(f"Processing complete! Total time: {elapsed_final_str}")
             with result_cols[0]:
                 st.video(source_file)
             with result_cols[1]: