Update app.py

app.py

@@ -27,25 +27,23 @@ 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)
 
-        # 4) ページのロードを待機
+        # 4) ページのロード完了を待機
         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)"
@@ -55,14 +53,15 @@ def render_fullpage_screenshot(html_code):
             "document.body.scrollHeight, document.documentElement.scrollHeight)"
         )
 
-        # 7) 保険として1.2倍にしてウィンドウサイズを設定
-        #    (floatをintにキャストし、少し余裕を持たせる)
-        adj_width = int(scroll_width * 1.2)
-        adj_height = int(scroll_height * 1.2)
-        driver.set_window_size(adj_width, adj_height)
+        # 7) ウィンドウサイズをページ全体 + 5%上下左右マージン (＝幅・高さ共に10%拡大) に設定
+        margin_factor = 0.10  # 上下左右それぞれ5%を足す => 合計で10%増やすイメージ
+        adjusted_width = int(scroll_width * (1 + margin_factor))
+        adjusted_height = int(scroll_height * (1 + margin_factor))
+
+        driver.set_window_size(adjusted_width, adjusted_height)
         time.sleep(2)  # レイアウトが変わるので少し待つ
 
-        # 念のためページ最上部にスクロール
+        # 念のためページ最上部にスクロールしてからキャプチャ
         driver.execute_script("window.scrollTo(0, 0)")
         time.sleep(1)
 
@@ -86,8 +85,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枚の画像として取得します。"
+    title="Full Page Screenshot with Margin",
+    description="HTMLをヘッドレスブラウザでレンダリングし、ページ全体＋5%のマージンをつけて1枚の画像として取得します。"
 )
 
 if __name__ == "__main__":