Solaria
75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
# Simple TeX Live API Service
|
|
# Run: python texlive_api.py
|
|
|
|
from flask import Flask, request, jsonify, send_file
|
|
import subprocess
|
|
import tempfile
|
|
import os
|
|
import base64
|
|
from pathlib import Path
|
|
|
|
app = Flask(__name__)
|
|
|
|
ALLOWED_ENGINES = ['xelatex', 'pdflatex', 'lualatex']
|
|
MAX_TIMEOUT = 180
|
|
|
|
|
|
@app.route('/health', methods=['GET'])
|
|
def health():
|
|
return jsonify({"status": "ok"})
|
|
|
|
|
|
@app.route('/compile', methods=['POST'])
|
|
def compile():
|
|
data = request.get_json()
|
|
|
|
tex_content = data.get('tex')
|
|
engine = data.get('engine', 'xelatex')
|
|
timeout = min(data.get('timeout', MAX_TIMEOUT), MAX_TIMEOUT)
|
|
|
|
if not tex_content:
|
|
return jsonify({"error": "No tex content provided"}), 400
|
|
|
|
if engine not in ALLOWED_ENGINES:
|
|
return jsonify({"error": f"Invalid engine: {engine}"}), 400
|
|
|
|
# Create temp directory
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
tex_file = Path(tmpdir) / "input.tex"
|
|
tex_file.write_text(tex_content)
|
|
|
|
# Compile
|
|
try:
|
|
result = subprocess.run(
|
|
[engine, "-interaction=nonstopmode", tex_file.name],
|
|
cwd=tmpdir,
|
|
capture_output=True,
|
|
timeout=timeout,
|
|
)
|
|
|
|
pdf_file = tex_file.with_suffix('.pdf')
|
|
|
|
if pdf_file.exists():
|
|
pdf_data = base64.b64encode(pdf_file.read_bytes()).decode()
|
|
return jsonify({
|
|
"success": True,
|
|
"pdf": pdf_data,
|
|
"engine": engine,
|
|
})
|
|
else:
|
|
# Return error log
|
|
return jsonify({
|
|
"success": False,
|
|
"error": "PDF not generated",
|
|
"log": result.stderr.decode()[-2000:],
|
|
}), 500
|
|
|
|
except subprocess.TimeoutExpired:
|
|
return jsonify({"error": "Compilation timeout"}), 504
|
|
except Exception as e:
|
|
return jsonify({"error": str(e)}), 500
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run(host='0.0.0.0', port=8080)
|