Update app.py

app.py

@@ -2248,35 +2248,39 @@ def main():
             if not (os.path.exists(weights_path) and os.path.exists(config_path)):
                 st.sidebar.warning("⚠️ YOLO models not found. Using basic people detector.")
                 
+                # Create a download button
                 if st.sidebar.button("Download YOLO Models"):
-                    with st.sidebar.spinner("Downloading YOLO models..."):
-                        # Ensure models directory exists
-                        os.makedirs(models_dir, exist_ok=True)
+                    # Use a placeholder in the sidebar to show status
+                    download_status = st.sidebar.empty()
+                    download_status.info("Downloading YOLO models... Please wait.")
+                    
+                    # Ensure models directory exists
+                    os.makedirs(models_dir, exist_ok=True)
+                    
+                    # Download YOLOv3 config
+                    try:
+                        import urllib.request
                         
-                        # Download YOLOv3 config
-                        try:
-                            import urllib.request
-                            
-                            # Download config file
-                            if not os.path.exists(config_path):
-                                urllib.request.urlretrieve(
-                                    "https://raw.githubusercontent.com/pjreddie/darknet/master/cfg/yolov3.cfg", 
-                                    config_path
-                                )
-                                st.sidebar.success("Downloaded YOLOv3 config file")
-                            
-                            # Download weights file (this is large - about 240MB)
-                            if not os.path.exists(weights_path):
-                                urllib.request.urlretrieve(
-                                    "https://pjreddie.com/media/files/yolov3.weights", 
-                                    weights_path
-                                )
-                                st.sidebar.success("Downloaded YOLOv3 weights file")
-                            
-                            st.sidebar.success("YOLO models downloaded successfully! Please refresh the page.")
-                        except Exception as e:
-                            st.sidebar.error(f"Error downloading YOLO models: {str(e)}")
-                            st.sidebar.info("You can manually download the models from: https://pjreddie.com/darknet/yolo/")
+                        # Download config file
+                        if not os.path.exists(config_path):
+                            download_status.info("Downloading configuration file...")
+                            urllib.request.urlretrieve(
+                                "https://raw.githubusercontent.com/pjreddie/darknet/master/cfg/yolov3.cfg", 
+                                config_path
+                            )
+                        
+                        # Download weights file (this is large - about 240MB)
+                        if not os.path.exists(weights_path):
+                            download_status.info("Downloading weights file (large, ~240MB)...")
+                            urllib.request.urlretrieve(
+                                "https://pjreddie.com/media/files/yolov3.weights", 
+                                weights_path
+                            )
+                        
+                        download_status.success("✅ YOLO models downloaded successfully! Please refresh the page.")
+                    except Exception as e:
+                        download_status.error(f"Error downloading YOLO models: {str(e)}")
+                        download_status.info("You can manually download the models from: https://pjreddie.com/darknet/yolo/")
             else:
                 st.sidebar.success("✅ YOLO models found. Using advanced object detection.")