[fix error]

app.py

@@ -28,10 +28,11 @@ def reflection_removal(input_image):
     if not input_image.lower().endswith((".jpg", ".jpeg", ".png")):
         return ["File is not supported (only .jpg, .jpeg, .png)."]
     
-    input_file_name = os.path.basename(input_image)
-    file_path = os.path.join(UPLOAD_DIR, input_file_name)
+    file_path = os.path.join(UPLOAD_DIR, os.path.basename(input_image))
     shutil.copy(input_image, file_path)
     
+    input_filename = os.path.splitext(os.path.basename(file_path))[0]
+    
     cmd = [
         "python", "test.py",
         "--dataroot", UPLOAD_DIR,
@@ -44,18 +45,14 @@ 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.startswith(input_file_name.split('.')[0]) and file.endswith("_fake.png"):
+            if file.endswith("_fake.png") and file.startswith(input_filename):
                 result_path = os.path.join(root, file)
-                output_images.append(Image.open(result_path))
-    
-    if output_images:
-        return output_images
-    else:
-        return ["No results found."]
+                return [Image.open(result_path)]
+
+    return ["No results found."]
+
 
 def use_sample_image(sample_image_name):
     sample_image_path = os.path.join(SAMPLE_DIR, sample_image_name)