[feat] add Auto Aspect Ratio

app.py

@@ -42,6 +42,16 @@ def reflection_removal(input_image):
     
     input_filename = os.path.splitext(os.path.basename(file_path))[0]
     
+    aspect_ratio = 1.0
+    
+    try:
+        with Image.open(file_path) as img:
+            width, height = img.size
+            aspect_ratio = width / height
+            print(f"Aspect Ratio: {aspect_ratio}")
+    except Exception as e:
+        pass
+    
     cmd = [
         "python", "test.py",
         "--dataroot", UPLOAD_DIR,
@@ -50,7 +60,8 @@ def reflection_removal(input_image):
         "--direction", "AtoB", "--dataset_mode", "single",
         "--norm", "batch", "--epoch", "310",
         "--num_test", str(count_files(UPLOAD_DIR)),
-        "--gpu_ids", "-1"
+        "--gpu_ids", "-1",
+        "--aspect_ratio", str(aspect_ratio)
     ]
     subprocess.run(cmd, check=True)