Update app.py

app.py

@@ -9,12 +9,24 @@ import shutil
 import os
 from PIL import Image
 import base64
-
+import random
 
 st.set_page_config(layout="wide",initial_sidebar_state="expanded",
                    page_icon='🔎',page_title='Poth-Hole Detector')
 
+image_directory = "val"  # Assuming "val" is the directory name
+
+# Get a list of image filenames in the directory
+image_filenames = [filename for filename in os.listdir(image_directory) if filename.endswith(".jpg")]
 
+# Function to generate a random image from the list of filenames
+def get_random_image():
+    if not image_filenames:
+        return None
+    random_image_filename = random.choice(image_filenames)
+    random_image_path = os.path.join(image_directory, random_image_filename)
+    return random_image_path
+    
 def get_video_base64(video_path):
     with open(video_path, "rb") as file:
         video_bytes = file.read()
@@ -187,10 +199,11 @@ def main():
         fileimage=st.file_uploader('Upload the file for detection 📁',type=['jpg'])
         st.info("If you haven't filed, our system will employ a default image for prediction 📁. Simply press the 'Predict' button and directly upload your file for analysis 🧐.")
         if st.button('Predict'):
+            default_image_path = get_random_image()
             if fileimage is None:
                 st.warning('🛎️We are using default image 📷')
                 counter=1
-                default_image_path='df.jpg'
+                
                 pic = Image.open(default_image_path)
                 image_np = np.array(pic)
                 names={0:'POTH_HOLE'