Update app.py

app.py

@@ -23,8 +23,7 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-top_image = Image.open('banner_top.png')
-bottom_image = Image.open('banner_bottom.png')
+top_image = Image.open('favicon.ico')
 main_image = Image.open('main_banner.png')
 
 upload_path = ""
@@ -119,23 +118,34 @@ def FINAL(img) :
 
     Plot_Result(img, raw_plate, digits)
 
+
+
 if selected_type == "Upload Image":
     st.info('✨ Supports all popular image formats 📷 - PNG, JPG, BMP 😉')
     uploaded_file = st.file_uploader("Upload Image of car's number plate 🚓", type=["png","jpg","bmp","jpeg"])
 
     if uploaded_file is not None:
+        with open(os.path.join(upload_path,uploaded_file.name),"wb") as f:
+            f.write((uploaded_file).getbuffer())
+        with st.spinner(f"Working... 💫"):
+            uploaded_image = os.path.abspath(os.path.join(upload_path,uploaded_file.name))
+            downloaded_image = os.path.abspath(os.path.join(download_path,str("output_"+uploaded_file.name)))
 
-        image = Image.open(uploaded_file).convert('RGB')
-        st.image(image, use_column_width=True)
-
-        cv2.imwrite('inpout_img.jpg', image)
+            with open(uploaded_image,'rb') as imge:
+                img_bytes = imge.read()
 
-        image = cv2.imread('inpout_img.jpg')
+            img = Image.open(io.BytesIO(img_bytes))
+            results = FINAL(img)
+            for img in results:
+                img_base64 = Image.fromarray(img)
+                img_base64.save(downloaded_image, format="JPEG")
 
-        with st.spinner(f"Working... 💫"):
-            result = FINAL(image)
+            final_image = Image.open(downloaded_image)
+            print("Opening ",final_image)
+            st.markdown("---")
+            st.image(final_image, caption='This is how your final image looks like 😉')
 
-        st.image(result)    
+   
         
         #with st.spinner(f"Working... 💫"):
          #   uploaded_image = os.path.abspath(os.path.join(upload_path,uploaded_file.name))