Update app.py

app.py

@@ -127,6 +127,8 @@ if selected_type == "Upload Image":
          #   f.write((uploaded_file).getbuffer())
         image = Image.open(uploaded_file).convert('RGB')
         st.image(image, use_column_width=True)
+
+        st.write(FINAL(image))  
         
         #with st.spinner(f"Working... 💫"):
          #   uploaded_image = os.path.abspath(os.path.join(upload_path,uploaded_file.name))
@@ -156,52 +158,14 @@ if selected_type == "Upload Image":
             
            # img = cv2.imdecode(img, cv2.IMREAD_COLOR)
 
-            results = FINAL(image)  
+            #results = FINAL(image)  
     
-            for img in results.imgs:
-                img_base64 = Image.fromarray(img)
-                img_base64.save(downloaded_image, format="JPEG")
+            #for img in results.imgs:
+                #img_base64 = Image.fromarray(img)
+                #img_base64.save(downloaded_image, format="JPEG")
               
 
-            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 😉')
-            with open(downloaded_image, "rb") as file:
-                if uploaded_file.name.endswith('.jpg') or uploaded_file.name.endswith('.JPG'):
-                    if st.download_button(
-                                            label="Download Output Image 📷",
-                                            data=file,
-                                            file_name=str("output_"+uploaded_file.name),
-                                            mime='image/jpg'
-                                         ):
-                        download_success()
-                if uploaded_file.name.endswith('.jpeg') or uploaded_file.name.endswith('.JPEG'):
-                    if st.download_button(
-                                            label="Download Output Image 📷",
-                                            data=file,
-                                            file_name=str("output_"+uploaded_file.name),
-                                            mime='image/jpeg'
-                                         ):
-                        download_success()
-
-                if uploaded_file.name.endswith('.png') or uploaded_file.name.endswith('.PNG'):
-                    if st.download_button(
-                                            label="Download Output Image 📷",
-                                            data=file,
-                                            file_name=str("output_"+uploaded_file.name),
-                                            mime='image/png'
-                                         ):
-                        download_success()
-
-                if uploaded_file.name.endswith('.bmp') or uploaded_file.name.endswith('.BMP'):
-                    if st.download_button(
-                                            label="Download Output Image 📷",
-                                            data=file,
-                                            file_name=str("output_"+uploaded_file.name),
-                                            mime='image/bmp'
-                                         ):
-                        download_success()
+            
     else:
         st.warning('⚠ Please upload your Image 😯')