Update app.py

app.py

@@ -27,7 +27,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)
 
@@ -45,6 +45,7 @@ def render_fullpage_screenshot(html_code):
         )
 
         # 6) ページ全体の幅・高さを正確に取得
+        #    bodyとdocumentElementの値を比較し、より大きい方を使用
         scroll_width = driver.execute_script(
             "return Math.max("
             "document.body.scrollWidth, document.documentElement.scrollWidth)"
@@ -54,16 +55,13 @@ def render_fullpage_screenshot(html_code):
             "document.body.scrollHeight, document.documentElement.scrollHeight)"
         )
 
-        # 7) ウィンドウサイズに2%のマージンを加える
-        #    端が切れやすい場合に、余白をつけて確実に表示領域を広げる
-        margin_width = int(scroll_width * 0.02)   # 幅の2%
-        margin_height = int(scroll_height * 0.02) # 高さの2%
+        # 上下方向に4%の余裕を加えた高さを計算
+        # （上下に2%ずつ、合計4%を足すイメージ）
+        adjusted_height = int(scroll_height * 1.04)
 
-        final_width = scroll_width + margin_width
-        final_height = scroll_height + margin_height
-
-        driver.set_window_size(final_width, final_height)
-        time.sleep(2)  # レイアウト再計算待ち
+        # 7) ウィンドウサイズを (幅はそのまま, 高さのみ4%増し) に設定
+        driver.set_window_size(scroll_width, adjusted_height)
+        time.sleep(2)  # レイアウトが変わるので少し待つ
 
         # 念のためページ最上部にスクロール
         driver.execute_script("window.scrollTo(0, 0)")
@@ -89,8 +87,8 @@ iface = gr.Interface(
     fn=render_fullpage_screenshot,
     inputs=gr.Textbox(lines=15, label="HTMLコード入力"),
     outputs=gr.Image(type="pil", label="ページ全体のスクリーンショット"),
-    title="Full Page Screenshot App",
-    description="HTMLをヘッドレスブラウザでレンダリングし、ページ全体を1枚の画像として取得します（端に2%の余裕）。"
+    title="Full Page Screenshot App (上下4%余裕)",
+    description="HTMLをヘッドレスブラウザでレンダリングし、ページ全体を1枚の画像として取得します。上下にだけ4%の余裕を加えます。"
 )
 
 if __name__ == "__main__":