Update app.py

app.py

@@ -1,5 +1,10 @@
 import os
 import requests
+from flask import Flask, request, send_file
+from selenium import webdriver
+from selenium.common.exceptions import WebDriverException
+from PIL import Image
+from io import BytesIO
 
 # フォントをダウンロードする関数
 def download_noto_sans_cjk():
@@ -20,16 +25,9 @@ def download_noto_sans_cjk():
 # フォントのダウンロードとインストールを実行
 download_noto_sans_cjk()
 
-
-from flask import Flask, request, send_file
-from selenium import webdriver
-from selenium.common.exceptions import WebDriverException
-from PIL import Image
-from io import BytesIO
-
 app = Flask(__name__)
 
-def take_screenshot(url):
+def take_screenshot(url, width=1080, height=720):
     options = webdriver.ChromeOptions()
     options.add_argument('--headless')
     options.add_argument('--no-sandbox')
@@ -37,7 +35,7 @@ def take_screenshot(url):
 
     try:
         wd = webdriver.Chrome(options=options)
-        wd.set_window_size(1080, 720)  # Adjust the window size here
+        wd.set_window_size(width, height)  # Adjust the window size here
         wd.get(url)
         wd.implicitly_wait(10)
         screenshot = wd.get_screenshot_as_png()
@@ -56,8 +54,12 @@ def screenshot():
     if not url:
         return "URL parameter is required.", 400
 
-    # Take the screenshot of the provided URL
-    image = take_screenshot(url)
+    # パラメーターを取得してスクリーンサイズを設定
+    screenw = request.args.get('screenw', default=1080, type=int)
+    screenh = request.args.get('screenh', default=720, type=int)
+
+    # Take the screenshot of the provided URL with specified width and height
+    image = take_screenshot(url, width=screenw, height=screenh)
 
     # Save the screenshot to an in-memory file and return as response
     img_io = BytesIO()