Update app.py

app.py

@@ -30,7 +30,7 @@ for key in ["processed_frames", "slider_value", "processed_video", "start_time"]
 with st.sidebar:
     st.header("Upload & Settings")
     source_file = st.file_uploader("Upload image/video", type=["jpg", "jpeg", "png", "bmp", "webp", "mp4"])
-    confidence = float(st.slider("Confidence Threshold", 10, 100, 30)) / 100
+    confidence = float(st.slider("Confidence Threshold", 10, 100, 20)) / 100
     fps_options = {
         "Original FPS": None,
         "3 FPS": 3,
@@ -64,8 +64,8 @@ Early wildfire detection using YOLOv8 AI vision model. See detected results abov
 Click on video frames to load and play examples.
 """)
 
-# Function to create synchronized video pair HTML
-def create_synced_video_pair(orig_url, proc_url, pair_id):
+# Function to create simple video pair HTML
+def create_video_pair(orig_url, proc_url):
     try:
         orig_bytes = requests.get(orig_url).content
         proc_bytes = requests.get(proc_url).content
@@ -76,52 +76,19 @@ 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>
+                <video 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>
+                <video width="100%" controls>
                     <source src="data:video/mp4;base64,{proc_b64}" type="video/mp4">
                     Your browser does not support the video tag.
                 </video>
             </div>
         </div>
-        <script>
-            document.addEventListener('DOMContentLoaded', function() {{
-                const origVideo = document.getElementById('orig_{pair_id}');
-                const procVideo = document.getElementById('proc_{pair_id}');
-                
-                if (origVideo && procVideo) {{
-                    origVideo.addEventListener('play', function() {{
-                        procVideo.currentTime = origVideo.currentTime;
-                        procVideo.play().catch(e => console.log('Play error:', e));
-                    }});
-                    procVideo.addEventListener('play', function() {{
-                        origVideo.currentTime = procVideo.currentTime;
-                        origVideo.play().catch(e => console.log('Play error:', e));
-                    }});
-                    
-                    origVideo.addEventListener('pause', function() {{
-                        procVideo.pause();
-                    }});
-                    procVideo.addEventListener('pause', function() {{
-                        origVideo.pause();
-                    }});
-                    
-                    origVideo.addEventListener('seeked', function() {{
-                        procVideo.currentTime = origVideo.currentTime;
-                    }});
-                    procVideo.addEventListener('seeked', function() {{
-                        origVideo.currentTime = procVideo.currentTime;
-                    }});
-                }} else {{
-                    console.log('Video elements not found for {pair_id}');
-                }}
-            }});
-        </script>
         """
         return html
     except Exception as e:
@@ -134,7 +101,7 @@ st.header("Your Results")
 result_cols = st.columns(2)
 viewer_slot = st.empty()
 
-# Example videos with synchronization (LA before T)
+# Example videos (LA before T)
 st.header("Example Results")
 examples = [
     ("LA Example", "LA1.mp4", "LA2.mp4"),
@@ -144,8 +111,7 @@ for title, orig_file, proc_file in examples:
     st.subheader(title)
     orig_url = f"https://huggingface.co/spaces/tstone87/ccr-colorado/resolve/main/{orig_file}"
     proc_url = f"https://huggingface.co/spaces/tstone87/ccr-colorado/resolve/main/{proc_file}"
-    pair_id = title.replace(" ", "").lower()
-    video_html = create_synced_video_pair(orig_url, proc_url, pair_id)
+    video_html = create_video_pair(orig_url, proc_url)
     st.markdown(video_html, unsafe_allow_html=True)
 
 # Load model
@@ -180,6 +146,9 @@ if process_button and source_file and model:
         output_fps = fps_options[video_option] if fps_options[video_option] else orig_fps
         sample_interval = max(1, int(orig_fps / output_fps)) if output_fps else 1
         
+        # Set fixed output FPS to 2 (500ms per frame = 2 FPS)
+        fixed_output_fps = 2
+        
         st.session_state.start_time = time.time()
         frame_count = 0
         processed_count = 0
@@ -198,9 +167,7 @@ if process_button and source_file and model:
                 progress = frame_count / total_frames
                 
                 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"
@@ -209,7 +176,7 @@ if process_button and source_file and model:
                     eta_str = "Calculating..."
                 
                 progress_bar.progress(min(progress, 1.0))
-                progress_text.text(f"Progress: {progress:.1%} | Elapsed: {elapsed_str} | ETA: {eta_str}")
+                progress_text.text(f"Progress: {progress:.1%}\nElapsed: {elapsed_str}\nETA: {eta_str}")
             
             frame_count += 1
             success, frame = vidcap.read()
@@ -222,7 +189,7 @@ if process_button and source_file and model:
             writer = ffmpeg.write_frames(
                 out_path,
                 (width, height),
-                fps=output_fps or orig_fps,
+                fps=fixed_output_fps,  # Fixed at 2 FPS (500ms per frame)
                 codec='libx264',
                 pix_fmt_in='bgr24',
                 pix_fmt_out='yuv420p'
@@ -240,7 +207,7 @@ if process_button and source_file and model:
             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(f"Processing complete! Total time: {elapsed_final_str}")
+            progress_text.text(f"Progress: 100%\nElapsed: {elapsed_final_str}\nETA: 0m 0s")
             with result_cols[0]:
                 st.video(source_file)
             with result_cols[1]: