Spaces:
Paused
Paused
File size: 3,673 Bytes
0051124 4f4db76 b0d5ffc b975c56 3d3aff7 b0d5ffc fce85cb 3d3aff7 b0d5ffc a7ae2e2 b975c56 b0d5ffc 3d3aff7 a7ae2e2 4f4db76 b975c56 3d3aff7 a7ae2e2 b0d5ffc 3d3aff7 b0d5ffc 7f3d6f1 b0d5ffc 7f3d6f1 a7ae2e2 b0d5ffc 7f3d6f1 b0d5ffc a7ae2e2 b0d5ffc a7ae2e2 7f3d6f1 b0d5ffc d9a1abd 7f3d6f1 b0d5ffc a7ae2e2 3d3aff7 a7ae2e2 7f3d6f1 a7ae2e2 7f3d6f1 a7ae2e2 7f3d6f1 a7ae2e2 b0d5ffc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 |
import gradio as gr
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.common.exceptions import WebDriverException, TimeoutException
import tempfile
import os
from PIL import Image
import time
import base64
from bs4 import BeautifulSoup # HTML解析用
def take_screenshot_from_html(html_code, height_adjustment_percent=3):
options = Options()
options.add_argument("--headless=new")
options.add_argument("--no-sandbox")
options.add_argument("--disable-dev-shm-usage")
options.add_argument("--disable-gpu")
options.add_argument("--window-position=-2400,-2400")
try:
driver = webdriver.Chrome(options=options)
# BeautifulSoupでHTMLを解析
soup = BeautifulSoup(html_code, 'html.parser')
# <head>タグがあるか確認し、なければ追加
if not soup.head:
head = soup.new_tag("head")
soup.insert(0, head)
# <meta charset>タグがあるか確認し、なければ追加（UTF-8を指定）
if not soup.head.find("meta", charset=True):
meta_charset = soup.new_tag("meta")
meta_charset["charset"] = "UTF-8"
soup.head.insert(0, meta_charset)
# 修正したHTMLを文字列に戻す
modified_html = str(soup)
html_base64 = base64.b64encode(modified_html.encode('utf-8')).decode('utf-8')
driver.get(f"data:text/html;base64,{html_base64}")
time.sleep(3)
width = driver.execute_script(
"return Math.max(document.body.scrollWidth, document.body.offsetWidth, "
"document.documentElement.clientWidth, document.documentElement.scrollWidth, "
"document.documentElement.offsetWidth);"
)
original_height = driver.execute_script(
"return Math.max(document.body.scrollHeight, document.body.offsetHeight, "
"document.documentElement.clientHeight, document.documentElement.scrollHeight, "
"document.documentElement.offsetHeight);"
)
adjusted_height = int(original_height * (1 + height_adjustment_percent / 100))
driver.set_window_size(min(width + 100, 1920), min(adjusted_height + 100, 5000))
time.sleep(1)
with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp:
temp_filename = temp.name
driver.save_screenshot(temp_filename)
driver.quit()
image = Image.open(temp_filename)
os.unlink(temp_filename)
return image, "スクリーンショットを取得しました。"
except (WebDriverException, TimeoutException) as e:
driver.quit()
return None, f"エラー: {str(e)}"
except Exception as e:
driver.quit()
return None, f"予期しないエラー: {str(e)}"
with gr.Blocks(title="HTML Screenshot Tool") as demo:
gr.Markdown("# HTML スクリーンショットツール")
gr.Markdown("HTMLコードを入力して、レンダリングされたページのフルスクリーンショットを取得します。")
html_input = gr.Textbox(label="HTMLコード", placeholder="<p>Hello, world!</p>", lines=5)
height_adjustment_slider = gr.Slider(minimum=0, maximum=10, value=3, step=0.1, label="高さ調整率 (%)")
html_button = gr.Button("スクリーンショットを取得")
html_image_out = gr.Image(label="スクリーンショット")
html_text_out = gr.Textbox(label="結果")
html_button.click(
take_screenshot_from_html,
inputs=[html_input, height_adjustment_slider],
outputs=[html_image_out, html_text_out]
)
demo.launch() |