Update app.py

app.py

@@ -27,7 +27,7 @@ download_noto_sans_cjk()
 
 app = Flask(__name__)
 
-def take_screenshot(url, width=1080, height=720):
+def take_screenshot(url, screenw=1080, screenh=720):
     options = webdriver.ChromeOptions()
     options.add_argument('--headless')
     options.add_argument('--no-sandbox')
@@ -35,12 +35,12 @@ def take_screenshot(url, width=1080, height=720):
 
     try:
         wd = webdriver.Chrome(options=options)
-        wd.set_window_size(width, height)
+        wd.set_window_size(screenw, screenh)  # Set window size based on screenw and screenh
         wd.get(url)
         wd.implicitly_wait(10)
         screenshot = wd.get_screenshot_as_png()
     except WebDriverException:
-        # エラーが発生した場合、空の画像を返す
+        # Return a blank image as a placeholder if there's an error
         return Image.new('RGB', (1, 1))
     finally:
         if wd:
@@ -54,34 +54,35 @@ def screenshot():
     if not url:
         return "URL parameter is required.", 400
 
-    # パラメーターを取得
-    screenw = request.args.get('screenw', default=None, type=int)
-    screenh = request.args.get('screenh', default=None, type=int)
+    # ウィンドウサイズのパラメーターを取得
+    screenw = request.args.get('screenw', default=1080, type=int)
+    screenh = request.args.get('screenh', default=720, type=int)
+    
+    # 画像リサイズのパラメーターを取得
+    width = request.args.get('width', type=int)
+    height = request.args.get('height', type=int)
 
-    # スクリーンショットを撮影
-    original_image = take_screenshot(url)
+    # スクリーンショットを取得
+    image = take_screenshot(url, screenw=screenw, screenh=screenh)
 
     # リサイズ処理
-    if screenw and screenh:
-        # 両方のパラメーターが指定されている場合
-        resized_image = original_image.resize((screenw, screenh))
-    elif screenw:
-        # width のみが指定されている場合、高さを比率に基づいて自動計算
-        aspect_ratio = original_image.height / original_image.width
-        resized_height = int(screenw * aspect_ratio)
-        resized_image = original_image.resize((screenw, resized_height))
-    elif screenh:
-        # height のみが指定されている場合、幅を比率に基づいて自動計算
-        aspect_ratio = original_image.width / original_image.height
-        resized_width = int(screenh * aspect_ratio)
-        resized_image = original_image.resize((resized_width, screenh))
-    else:
-        # 両方のパラメーターが指定されていない場合
-        resized_image = original_image
-
-    # 画像をメモリ上に保存してレスポンスとして返す
+    if width and height:
+        # 両方のパラメーターが提供された場合
+        image = image.resize((width, height), Image.ANTIALIAS)
+    elif width:
+        # width のみが提供された場合、アスペクト比に基づいて height を計算
+        aspect_ratio = image.height / image.width
+        height = int(width * aspect_ratio)
+        image = image.resize((width, height), Image.ANTIALIAS)
+    elif height:
+        # height のみが提供された場合、アスペクト比に基づいて width を計算
+        aspect_ratio = image.width / image.height
+        width = int(height * aspect_ratio)
+        image = image.resize((width, height), Image.ANTIALIAS)
+
+    # 結果画像をメモリに保存してレスポンスとして返却
     img_io = BytesIO()
-    resized_image.save(img_io, 'PNG')
+    image.save(img_io, 'PNG')
     img_io.seek(0)
 
     return send_file(img_io, mimetype='image/png')