Update app.py

app.py

@@ -10,7 +10,11 @@ import tempfile
 import time
 import os
 
-def render_fullpage_screenshot(html_code):
+def render_fullpage_screenshot(html_code, extension_percentage):
+    """
+    html_code: HTMLのソースコード文字列
+    extension_percentage: 上下に追加する余裕の% (例: 4なら合計4%の余裕)
+    """
     # 1) HTMLコードを一時ファイルに保存
     tmp_file = tempfile.NamedTemporaryFile(suffix=".html", delete=False)
     tmp_path = tmp_file.name
@@ -27,7 +31,7 @@ def render_fullpage_screenshot(html_code):
     try:
         driver = webdriver.Chrome(options=options)
 
-        # 3) まずはある程度のウィンドウサイズでページを開く
+        # 3) 初期ウィンドウサイズでページを読み込む
         driver.set_window_size(1200, 800)
         driver.get("file://" + tmp_path)
 
@@ -35,60 +39,56 @@ def render_fullpage_screenshot(html_code):
         WebDriverWait(driver, 10).until(
             EC.presence_of_element_located((By.TAG_NAME, "body"))
         )
-        # 外部リソース読み込み等の安定化のため余分に待機
-        time.sleep(2)
+        time.sleep(2)  # 外部リソースの読み込み待機
 
-        # 5) スクロールバーを写したくない場合はCSSで非表示にする
+        # 5) スクロールバーがキャプチャに写らないようCSSで非表示
         driver.execute_script(
             "document.documentElement.style.overflow = 'hidden';"
             "document.body.style.overflow = 'hidden';"
         )
 
-        # 6) ページ全体の幅・高さを正確に取得
-        #    bodyとdocumentElementの値を比較し、より大きい方を使用
+        # 6) ページ全体の幅・高さを正確に取得 (bodyとdocumentElementの両方から)
         scroll_width = driver.execute_script(
-            "return Math.max("
-            "document.body.scrollWidth, document.documentElement.scrollWidth)"
+            "return Math.max(document.body.scrollWidth, document.documentElement.scrollWidth)"
         )
         scroll_height = driver.execute_script(
-            "return Math.max("
-            "document.body.scrollHeight, document.documentElement.scrollHeight)"
+            "return Math.max(document.body.scrollHeight, document.documentElement.scrollHeight)"
         )
 
-        # 上下方向に4%の余裕を加えた高さを計算
-        # （上下に2%ずつ、合計4%を足すイメージ）
-        adjusted_height = int(scroll_height * 1.04)
-
-        # 7) ウィンドウサイズを (幅はそのまま, 高さのみ4%増し) に設定
+        # 7) 上下方向にだけユーザー指定の余裕(%)を加えた高さを計算
+        adjusted_height = int(scroll_height * (1 + extension_percentage/100))
+        
+        # 8) ウィンドウサイズを、幅はそのまま、縦はadjusted_heightに変更
         driver.set_window_size(scroll_width, adjusted_height)
-        time.sleep(2)  # レイアウトが変わるので少し待つ
+        time.sleep(2)  # レイアウトの安定化待機
 
-        # 念のためページ最上部にスクロール
+        # 念のため最上部にスクロール
         driver.execute_script("window.scrollTo(0, 0)")
         time.sleep(1)
 
-        # 8) スクリーンショットを取得
+        # 9) スクリーンショット取得
         png = driver.get_screenshot_as_png()
 
     except Exception as e:
-        # 失敗時は1x1の黒画像を返す
+        # エラー時は1x1の黒画像を返す
         return Image.new('RGB', (1, 1), color=(0, 0, 0))
-
     finally:
         driver.quit()
         if os.path.exists(tmp_path):
             os.remove(tmp_path)
 
-    # 9) PNGバイナリをPIL.Imageに変換して返す
     return Image.open(BytesIO(png))
 
-# Gradioインターフェース
+# Gradioインターフェースの定義
 iface = gr.Interface(
     fn=render_fullpage_screenshot,
-    inputs=gr.Textbox(lines=15, label="HTMLコード入力"),
+    inputs=[
+        gr.Textbox(lines=15, label="HTMLコード入力"),
+        gr.Slider(minimum=0, maximum=20, step=0.5, value=4, label="上下高さ拡張率（%）")
+    ],
     outputs=gr.Image(type="pil", label="ページ全体のスクリーンショット"),
-    title="Full Page Screenshot App (上下4%余裕)",
-    description="HTMLをヘッドレスブラウザでレンダリングし、ページ全体を1枚の画像として取得します。上下にだけ4%の余裕を加えます。"
+    title="Full Page Screenshot App (高さ拡張調整可能)",
+    description="HTMLをヘッドレスブラウザでレンダリングし、ページ全体を1枚の画像として取得します。上下のみユーザー指定の余裕(%)を追加します。"
 )
 
 if __name__ == "__main__":