fixed problem for two different image types (PIL, uploaded_file)

app.py

@@ -75,6 +75,18 @@ def start_evaluation(uploaded_file):
     evaluate(reversed_image)
 
 
+def start_evaluation_pil_img(pil_image):
+    # Convert the PIL image to a NumPy array
+    opencv_image = np.array(pil_image)
+    # Convert the image from RGB (PIL format) to BGR (OpenCV format)
+    opencv_image = cv2.cvtColor(opencv_image, cv2.COLOR_RGB2BGR)
+    # Reverse the color channels back to RGB for display in Streamlit
+    reversed_image = opencv_image[:, :, ::-1]
+    
+    st.image(reversed_image, caption="Input Image")
+    evaluate(reversed_image)
+
+
 st.markdown(DESCRIPTION)
 
 
@@ -101,7 +113,7 @@ for idx, image_path in enumerate(image_paths):
         st.image(gt_paths[idx], use_column_width=True, caption="Ground Truth")
 
 if img is not None:
-    start_evaluation(img)
+    start_evaluation_pil_img(img)
 
 def reset_image_select():
     img = None
@@ -110,6 +122,5 @@ def reset_image_select():
 uploaded_file= None
 uploaded_file = st.file_uploader("Please upload an image", type=["jpeg", "jpg", "png"], on_change=reset_image_select)
 if (uploaded_file is not None) and (img is None):
-    pil_img = Image.open(uploaded_file)
-    start_evaluation(pil_img)
+    start_evaluation(uploaded_file)