Update app.py

app.py

@@ -5,6 +5,7 @@ import streamlit as st
 from PIL import Image
 from collections import Counter
 import os
+import requests
 
 # Streamlit app title
 st.set_page_config(page_title="Object Detection App", page_icon="🖼️", layout="centered")
@@ -21,30 +22,39 @@ The application will display the image with bounding boxes around the objects.
 st.markdown("### Step 1: Upload an Image for Object Detection")
 uploaded_file = st.file_uploader("Choose an image...", type=["png", "jpg", "jpeg"])
 
-# Function to disable progress bar in cvlib download function
-def download_file_without_progress(url, file_name, dest_dir):
-    """
-    Downloads the file without displaying progress bar.
-    """
+# Function to download a file without using the progress bar
+def download_file_no_progress(url, file_name, dest_dir):
     file_path = os.path.join(dest_dir, file_name)
     if not os.path.exists(file_path):
-        # Perform the download without progress bar
         print(f"Downloading {file_name}...")
-        os.system(f"wget {url} -O {file_path}")
+        # Disable the progress bar by using requests
+        response = requests.get(url, stream=True)
+        with open(file_path, 'wb') as f:
+            for chunk in response.iter_content(chunk_size=1024):
+                if chunk:
+                    f.write(chunk)
     return file_path
 
-# Modify the detect_common_objects function to accept a skip_progress argument
+# Function to perform object detection with manual model file paths
 def detect_common_objects_skip_progress(image, model='yolov3', skip_progress=False):
-    # If skip_progress is True, we modify the behavior of the model download process
     if skip_progress:
-        # Manually set the file paths for the config and weights files if needed
-        model_config = 'path_to_config.cfg'  # Provide the correct path to config file
-        model_weights = 'path_to_weights.weights'  # Provide the correct path to weights file
+        # Manually set paths for config and weights if progress bar should be disabled
+        model_config = 'path_to_config.cfg'  # Provide the path to the .cfg file
+        model_weights = 'path_to_weights.weights'  # Provide the path to the .weights file
     else:
-        model_config = cv.download_file(url="https://github.com/opencv/opencv_contrib/blob/master/modules/dnn/data/yolov3.cfg", file_name="yolov3.cfg", dest_dir=".")
-        model_weights = cv.download_file(url="https://github.com/opencv/opencv_contrib/blob/master/modules/dnn/data/yolov3.weights", file_name="yolov3.weights", dest_dir=".")
+        # Download the config and weights files if not already present
+        model_config = download_file_no_progress(
+            url="https://github.com/opencv/opencv_contrib/blob/master/modules/dnn/data/yolov3.cfg", 
+            file_name="yolov3.cfg", 
+            dest_dir="."
+        )
+        model_weights = download_file_no_progress(
+            url="https://github.com/opencv/opencv_contrib/blob/master/modules/dnn/data/yolov3.weights", 
+            file_name="yolov3.weights", 
+            dest_dir="."
+        )
     
-    # Run detection
+    # Run the object detection
     box, label, count = cv.detect_common_objects(image, model=model)
     return box, label, count