Update app.py

app.py

@@ -11,6 +11,7 @@ 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')
 
@@ -25,8 +26,8 @@ def get_random_image():
         return None
     random_image_filename = random.choice(image_filenames)
     random_image_path = os.path.join(image_directory, random_image_filename)
-    return random_image_path
-    
+    return random_image_path      
+
 def get_video_base64(video_path):
     with open(video_path, "rb") as file:
         video_bytes = file.read()
@@ -194,54 +195,77 @@ def main():
                     except Exception as e:
                         st.error(f"An error occurred: {e}")
     elif sidebar_option == "Upload file":  
-                  
+          
         
-        fileimage=st.file_uploader('Upload the file for detection 📁',type=['jpg'])
+        fileimage=st.file_uploader('Upload the file for detection 📁',type=['jpg','jpeg','png'])
         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
-                
-                pic = Image.open(default_image_path)
-                image_np = np.array(pic)
-                names={0:'POTH_HOLE'
+
+                    
+            if True:
+                    
+                if fileimage is None:
+                    default_image=get_random_image()
+                    st.warning('⚠️ We are using random image from our backend!.')
+                    st.info('Wait for the results...!')
+                    counter=1
+                    pic=Image.open(default_image)
+                    image_np = np.array(pic)
+                    names={0:'POTH_HOLE'
                         }
-                mod1=YOLO('best.pt')
-                mod1.predict(image_np,save=True,save_txt=True)
-                txt_files_exist = any(filename.endswith('.txt') for filename in os.listdir('runs/detect/predict/labels'))
-                if txt_files_exist:
-                    lis=open('runs/detect/predict/labels/image0.txt','r').readlines()
-                    with st.spinner('Wait for the results...!'):
-                        time.sleep(5)
+                    mod1=YOLO('best.pt')
+                    mod1.predict(image_np,save=True,save_txt=True)
+                    txt_files_exist = any(filename.endswith('.txt') for filename in os.listdir('runs/detect/predict/labels'))
+                    if txt_files_exist:
+                        lis=open('runs/detect/predict/labels/image0.txt','r').readlines()
+                        with st.spinner('Wait for the results...!'):
+                            time.sleep(5)
     
                                     
-                    for line in lis:
-                        process_line(line, image_np,counter)
-                        counter+=1
-                    col1,col2=st.columns(2)
-                    with col1:
+                        for line in lis:
+                            process_line(line, image_np,counter)
+                            counter+=1
+                        col1,col2=st.columns(2)
+                        with col1:
                                 
-                        st.info('Original Image!')
-                        st.image(default_image_path,use_column_width=True)
-                    with col2:
-                        st.info('Detected Image!')
-                        st.image(image_np,use_column_width=True)
-                    st.balloons()
+                            st.info('Original Image!')
+                            st.image(default_image,use_column_width=True)
+                        with col2:
+                            st.info('Detected Image!')
+                            st.image(image_np,use_column_width=True)
+                        st.balloons()
                         
-                    try:
-                        if os.path.exists('runs'):
-                            shutil.rmtree('runs')
-                            st.session_state.original_image = None  # Clear the original_image variable
+                        try:
+                            if os.path.exists('runs'):
+                                shutil.rmtree('runs')
+                                st.session_state.original_image = None  # Clear the original_image variable
                                 
-                    except Exception as e:
-                        st.error(f"An error occurred: {e}")
-                    
-            else:
-                    
-                if fileimage is not None and not fileimage.name.endswith('.jpg'):
-                    st.warning('⚠️ Please upload only JPG images.')
+                        except Exception as e:
+                            st.error(f"An error occurred: {e}")
+                    else:
+                        st.warning('⚠️Please check your image')
+                        st.info("📷✨ **Encountering the 'Please check your image' error?**")
+                        st.write(
+                            """
+                            Our algorithm may not have been able to predict the content of your image. To improve results, consider the following:
+
+                            👉 **Verify image quality and resolution.**
+                            👉 **Ensure the image is clear and well-lit.**
+                            👉 **Check if the image meets our specified format requirements.**
+                            👉 **Consider alternative images for better results.**
+
+                            Our aim is to provide accurate predictions, and addressing these aspects can make a significant difference. If the issue persists, please reach out to our support team. We're here to help! 🤝🔧
+                            """
+                        )
+                        try:
+                            if os.path.exists('runs'):
+                                shutil.rmtree('runs')
+                                st.session_state.original_image = None  # Clear the original_image variable
+                            
+                        except Exception as e:
+                            st.error(f"An error occurred: {e}")
                 else:
                     st.info('Wait for the results...!')
                     counter=1
@@ -311,4 +335,4 @@ if __name__ == '__main__':
     
    
     main()
-    
+