feat(agents): добавлены skills для PDF, docs, slides, jupyter и spreadsheet
- Зачем: - расширены возможности AI-ассистентов для работы с документами и презентациями - Что: - добавлен skill pdf для работы с PDF файлами - добавлен skill doc для работы с Word документами - добавлен skill slides для создания презентаций PowerPoint - добавлен skill jupyter-notebook для работы с Jupyter ноутбуками - добавлен skill spreadsheet для работы с таблицами (Excel/CSV) - изменена модель Claude с sonnet на opus - Проверка: - проверка доступности skills в AI-ассистентах
This commit is contained in:
ddadmin committed
1 parent
bd06d5734c
commit
0009270c6e
53 files changed
+6773
-1
No files matched your search
@@ -0,0 +1,273 @@
|
||||
#!/usr/bin/env python3
|
||||
# Copyright (c) OpenAI. All rights reserved.
|
||||
import argparse
|
||||
import os
|
||||
import re
|
||||
import subprocess
|
||||
import tempfile
|
||||
import xml.etree.ElementTree as ET
|
||||
from os import makedirs, replace
|
||||
from os.path import abspath, basename, exists, expanduser, join, splitext
|
||||
from typing import Sequence, cast
|
||||
from zipfile import ZipFile
|
||||
|
||||
from pdf2image import convert_from_path, pdfinfo_from_path
|
||||
|
||||
EMU_PER_INCH: int = 914_400
|
||||
|
||||
|
||||
def calc_dpi_via_ooxml(input_path: str, max_w_px: int, max_h_px: int) -> int:
|
||||
"""Calculate DPI from OOXML `ppt/presentation.xml` slide size (cx/cy in EMUs)."""
|
||||
with ZipFile(input_path, "r") as zf:
|
||||
xml = zf.read("ppt/presentation.xml")
|
||||
root = ET.fromstring(xml)
|
||||
ns = {"p": "http://schemas.openxmlformats.org/presentationml/2006/main"}
|
||||
sld_sz = root.find("p:sldSz", ns)
|
||||
if sld_sz is None:
|
||||
raise RuntimeError("Slide size not found in presentation.xml")
|
||||
cx = int(sld_sz.get("cx") or 0)
|
||||
cy = int(sld_sz.get("cy") or 0)
|
||||
if cx <= 0 or cy <= 0:
|
||||
raise RuntimeError("Invalid slide size values in presentation.xml")
|
||||
width_in = cx / EMU_PER_INCH
|
||||
height_in = cy / EMU_PER_INCH
|
||||
return round(min(max_w_px / width_in, max_h_px / height_in))
|
||||
|
||||
|
||||
def calc_dpi_via_pdf(input_path: str, max_w_px: int, max_h_px: int) -> int:
|
||||
"""Compute DPI from PDF page size.
|
||||
|
||||
For non-PDF inputs, first convert to PDF via LibreOffice to read page size.
|
||||
For PDFs, use the PDF directly (avoids unnecessary conversion and failures).
|
||||
"""
|
||||
is_pdf = input_path.lower().endswith(".pdf")
|
||||
with tempfile.TemporaryDirectory(prefix="soffice_profile_") as user_profile:
|
||||
with tempfile.TemporaryDirectory(prefix="soffice_convert_") as convert_tmp_dir:
|
||||
stem = splitext(basename(input_path))[0]
|
||||
pdf_path = (
|
||||
input_path
|
||||
if is_pdf
|
||||
else convert_to_pdf(input_path, user_profile, convert_tmp_dir, stem)
|
||||
)
|
||||
if not (pdf_path and exists(pdf_path)):
|
||||
raise RuntimeError("Failed to produce/read PDF for DPI computation.")
|
||||
|
||||
info = pdfinfo_from_path(pdf_path)
|
||||
size_val = info.get("Page size")
|
||||
if not size_val:
|
||||
for k, v in info.items():
|
||||
if isinstance(v, str) and "size" in k.lower() and "pts" in v:
|
||||
size_val = v
|
||||
break
|
||||
if not isinstance(size_val, str):
|
||||
raise RuntimeError("Failed to read PDF page size for DPI computation.")
|
||||
|
||||
def _parse_page_size_to_pts(s: str) -> tuple[float, float]:
|
||||
# Common formats from poppler/pdfinfo:
|
||||
# - "612 x 792 pts (letter)"
|
||||
# - "595.276 x 841.89 pts (A4)"
|
||||
# - sometimes inches: "8.5 x 11 in"
|
||||
m_pts = re.search(
|
||||
r"([0-9]+(?:\.[0-9]+)?)\s*x\s*([0-9]+(?:\.[0-9]+)?)\s*pts\b",
|
||||
s,
|
||||
)
|
||||
if m_pts:
|
||||
return float(m_pts.group(1)), float(m_pts.group(2))
|
||||
m_in = re.search(
|
||||
r"([0-9]+(?:\.[0-9]+)?)\s*x\s*([0-9]+(?:\.[0-9]+)?)\s*in\b",
|
||||
s,
|
||||
)
|
||||
if m_in:
|
||||
w_in = float(m_in.group(1))
|
||||
h_in = float(m_in.group(2))
|
||||
return w_in * 72.0, h_in * 72.0
|
||||
# Sometimes poppler returns without an explicit unit; treat as points.
|
||||
m = re.search(r"([0-9]+(?:\.[0-9]+)?)\s*x\s*([0-9]+(?:\.[0-9]+)?)\b", s)
|
||||
if m:
|
||||
return float(m.group(1)), float(m.group(2))
|
||||
raise RuntimeError(f"Unrecognized PDF page size format: {s!r}")
|
||||
|
||||
width_pts, height_pts = _parse_page_size_to_pts(size_val)
|
||||
width_in = width_pts / 72.0
|
||||
height_in = height_pts / 72.0
|
||||
if width_in <= 0 or height_in <= 0:
|
||||
raise RuntimeError("Invalid PDF page size values.")
|
||||
return round(min(max_w_px / width_in, max_h_px / height_in))
|
||||
|
||||
|
||||
def run_cmd_no_check(cmd: list[str]) -> None:
|
||||
subprocess.run(
|
||||
cmd,
|
||||
check=False,
|
||||
stdout=subprocess.DEVNULL,
|
||||
stderr=subprocess.DEVNULL,
|
||||
env=os.environ.copy(),
|
||||
)
|
||||
|
||||
|
||||
def convert_to_pdf(
|
||||
pptx_path: str,
|
||||
user_profile: str,
|
||||
convert_tmp_dir: str,
|
||||
stem: str,
|
||||
) -> str:
|
||||
# Try direct PPTX -> PDF
|
||||
cmd_pdf = [
|
||||
"soffice",
|
||||
"-env:UserInstallation=file://" + user_profile,
|
||||
"--invisible",
|
||||
"--headless",
|
||||
"--norestore",
|
||||
"--convert-to",
|
||||
"pdf",
|
||||
"--outdir",
|
||||
convert_tmp_dir,
|
||||
pptx_path,
|
||||
]
|
||||
run_cmd_no_check(cmd_pdf)
|
||||
|
||||
pdf_path = join(convert_tmp_dir, f"{stem}.pdf")
|
||||
if exists(pdf_path):
|
||||
return pdf_path
|
||||
|
||||
# Fallback: PPTX -> ODP, then ODP -> PDF
|
||||
# Rationale: Saving as ODP normalizes PPTX-specific constructs via the ODF serializer,
|
||||
# which often bypasses Impress PDF export issues on problematic decks.
|
||||
cmd_odp = [
|
||||
"soffice",
|
||||
"-env:UserInstallation=file://" + user_profile,
|
||||
"--invisible",
|
||||
"--headless",
|
||||
"--norestore",
|
||||
"--convert-to",
|
||||
"odp",
|
||||
"--outdir",
|
||||
convert_tmp_dir,
|
||||
pptx_path,
|
||||
]
|
||||
run_cmd_no_check(cmd_odp)
|
||||
|
||||
odp_path = join(convert_tmp_dir, f"{stem}.odp")
|
||||
|
||||
if exists(odp_path):
|
||||
# ODP -> PDF
|
||||
cmd_odp_pdf = [
|
||||
"soffice",
|
||||
"-env:UserInstallation=file://" + user_profile,
|
||||
"--invisible",
|
||||
"--headless",
|
||||
"--norestore",
|
||||
"--convert-to",
|
||||
"pdf",
|
||||
"--outdir",
|
||||
convert_tmp_dir,
|
||||
odp_path,
|
||||
]
|
||||
run_cmd_no_check(cmd_odp_pdf)
|
||||
if exists(pdf_path):
|
||||
return pdf_path
|
||||
|
||||
return ""
|
||||
|
||||
|
||||
def rasterize(
|
||||
input_path: str,
|
||||
out_dir: str,
|
||||
dpi: int,
|
||||
) -> Sequence[str]:
|
||||
"""Rasterise PPTX/PDF to PNG files placed in out_dir and return the image paths."""
|
||||
makedirs(out_dir, exist_ok=True)
|
||||
input_path = abspath(input_path)
|
||||
stem = splitext(basename(input_path))[0]
|
||||
|
||||
# Use a unique user profile to avoid LibreOffice profile lock when running concurrently
|
||||
with tempfile.TemporaryDirectory(prefix="soffice_profile_") as user_profile:
|
||||
# Write conversion outputs into a temp directory to avoid any IO oddities
|
||||
with tempfile.TemporaryDirectory(prefix="soffice_convert_") as convert_tmp_dir:
|
||||
is_pdf = input_path.lower().endswith(".pdf")
|
||||
pdf_path = (
|
||||
input_path
|
||||
if is_pdf
|
||||
else convert_to_pdf(input_path, user_profile, convert_tmp_dir, stem)
|
||||
)
|
||||
|
||||
if not pdf_path or not exists(pdf_path):
|
||||
raise RuntimeError(
|
||||
"Failed to produce PDF for rasterization (direct and ODP fallback)."
|
||||
)
|
||||
|
||||
# Perform rasterization while the temp PDF still exists
|
||||
paths_raw = cast(
|
||||
list[str],
|
||||
convert_from_path(
|
||||
pdf_path,
|
||||
dpi=dpi,
|
||||
fmt="png",
|
||||
thread_count=8,
|
||||
output_folder=out_dir,
|
||||
paths_only=True,
|
||||
output_file="slide",
|
||||
),
|
||||
)
|
||||
# Rename convert_from_path's output format f'slide{thread_id:04d}-{page_num:02d}.png'
|
||||
slides = []
|
||||
for src_path in paths_raw:
|
||||
base = splitext(basename(src_path))[0]
|
||||
slide_num_str = base.split("-")[-1]
|
||||
slide_num = int(slide_num_str)
|
||||
dst_path = join(out_dir, f"slide-{slide_num}.png")
|
||||
replace(src_path, dst_path)
|
||||
slides.append((slide_num, dst_path))
|
||||
slides.sort(key=lambda t: t[0])
|
||||
final_paths = [path for _, path in slides]
|
||||
return final_paths
|
||||
|
||||
|
||||
def main() -> None:
|
||||
parser = argparse.ArgumentParser(description="Render slides to images.")
|
||||
parser.add_argument(
|
||||
"input_path",
|
||||
type=str,
|
||||
help="Path to the input PowerPoint or PDF file.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--output_dir",
|
||||
type=str,
|
||||
default=None,
|
||||
help=(
|
||||
"Output directory for the rendered images. "
|
||||
"Defaults to a folder next to the input named after the input file (without extension)."
|
||||
),
|
||||
)
|
||||
parser.add_argument(
|
||||
"--width",
|
||||
type=int,
|
||||
default=1600,
|
||||
help=(
|
||||
"Approximate maximum width in pixels after isotropic scaling (default 1600). "
|
||||
"The actual value may exceed slightly."
|
||||
),
|
||||
)
|
||||
parser.add_argument(
|
||||
"--height",
|
||||
type=int,
|
||||
default=900,
|
||||
help=(
|
||||
"Approximate maximum height in pixels after isotropic scaling (default 900). "
|
||||
"The actual value may exceed slightly."
|
||||
),
|
||||
)
|
||||
args = parser.parse_args()
|
||||
|
||||
input_path = abspath(expanduser(args.input_path))
|
||||
out_dir = abspath(expanduser(args.output_dir)) if args.output_dir else splitext(input_path)[0]
|
||||
if input_path.lower().endswith((".pptx", ".ppsx", ".potx", ".pptm", ".ppsm", ".potm")):
|
||||
dpi = calc_dpi_via_ooxml(input_path, args.width, args.height)
|
||||
else:
|
||||
dpi = calc_dpi_via_pdf(input_path, args.width, args.height)
|
||||
rasterize(input_path, out_dir, dpi)
|
||||
print("Slides rendered to " + out_dir)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in new issue
Block a user