[feat] output validation

app.py

@@ -28,7 +28,8 @@ def reflection_removal(input_image):
     if not input_image.lower().endswith((".jpg", ".jpeg", ".png")):
         return ["File is not supported (only .jpg, .jpeg, .png)."]
     
-    file_path = os.path.join(UPLOAD_DIR, os.path.basename(input_image))
+    input_file_name = os.path.basename(input_image)
+    file_path = os.path.join(UPLOAD_DIR, input_file_name)
     shutil.copy(input_image, file_path)
     
     cmd = [
@@ -43,14 +44,18 @@ def reflection_removal(input_image):
     ]
     subprocess.run(cmd, check=True)
     
+    output_images = []
+    
     for root, _, files in os.walk(RESULTS_DIR):
         for file in files:
-            if file.endswith("_fake.png"):
+            if file.startswith(input_file_name.split('.')[0]) and file.endswith("_fake.png"):
                 result_path = os.path.join(root, file)
-                return [Image.open(result_path)]
-
-    return ["No results found."]
-
+                output_images.append(Image.open(result_path))
+    
+    if output_images:
+        return output_images
+    else:
+        return ["No results found."]
 
 def use_sample_image(sample_image_name):
     sample_image_path = os.path.join(SAMPLE_DIR, sample_image_name)