feat: add Python bridge and optional FastAPI server
- Add bridge.py for clean JSON-based Python-Node communication
- Add optional FastAPI server for REST API access
- Include requirements for API server setup
🤖 Generated with Claude Code
Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
Thomas Hallock
parent
fb1b0470cf
commit
98263a79a0
|
|
@ -0,0 +1,8 @@
|
||||||
|
# API server requirements
|
||||||
|
fastapi==0.104.1
|
||||||
|
uvicorn[standard]==0.24.0
|
||||||
|
pydantic==2.5.0
|
||||||
|
python-multipart==0.0.6
|
||||||
|
|
||||||
|
# Include base requirements
|
||||||
|
-r requirements.txt
|
||||||
|
|
@ -0,0 +1,160 @@
|
||||||
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
|
from fastapi import FastAPI, HTTPException, Response
|
||||||
|
from fastapi.middleware.cors import CORSMiddleware
|
||||||
|
from pydantic import BaseModel, Field
|
||||||
|
from typing import Optional, List, Literal
|
||||||
|
import tempfile
|
||||||
|
import subprocess
|
||||||
|
from pathlib import Path
|
||||||
|
import base64
|
||||||
|
import io
|
||||||
|
|
||||||
|
from generate import parse_range, generate_typst_file
|
||||||
|
|
||||||
|
app = FastAPI(title="Soroban Flashcard Generator API")
|
||||||
|
|
||||||
|
# Enable CORS for web apps
|
||||||
|
app.add_middleware(
|
||||||
|
CORSMiddleware,
|
||||||
|
allow_origins=["*"], # Configure this for your domain in production
|
||||||
|
allow_credentials=True,
|
||||||
|
allow_methods=["*"],
|
||||||
|
allow_headers=["*"],
|
||||||
|
)
|
||||||
|
|
||||||
|
class FlashcardRequest(BaseModel):
|
||||||
|
"""Request model for generating flashcards"""
|
||||||
|
range: str = Field("0-9", description="Number range (e.g., '0-99') or list (e.g., '1,2,5,10')")
|
||||||
|
step: int = Field(1, description="Step/increment for ranges")
|
||||||
|
cards_per_page: int = Field(6, description="Cards per page")
|
||||||
|
paper_size: str = Field("us-letter", description="Paper size")
|
||||||
|
orientation: Literal["portrait", "landscape"] = Field("portrait")
|
||||||
|
margins: dict = Field(default_factory=lambda: {
|
||||||
|
"top": "0.5in",
|
||||||
|
"bottom": "0.5in",
|
||||||
|
"left": "0.5in",
|
||||||
|
"right": "0.5in"
|
||||||
|
})
|
||||||
|
gutter: str = Field("5mm", description="Space between cards")
|
||||||
|
shuffle: bool = False
|
||||||
|
seed: Optional[int] = None
|
||||||
|
show_cut_marks: bool = False
|
||||||
|
show_registration: bool = False
|
||||||
|
font_family: str = Field("DejaVu Sans")
|
||||||
|
font_size: str = Field("48pt")
|
||||||
|
columns: str = Field("auto", description="Number of soroban columns")
|
||||||
|
show_empty_columns: bool = False
|
||||||
|
hide_inactive_beads: bool = False
|
||||||
|
bead_shape: Literal["diamond", "circle", "square"] = Field("diamond")
|
||||||
|
color_scheme: Literal["monochrome", "place-value", "heaven-earth", "alternating"] = Field("monochrome")
|
||||||
|
colored_numerals: bool = False
|
||||||
|
scale_factor: float = Field(0.9, ge=0.1, le=1.0)
|
||||||
|
format: Literal["pdf", "base64"] = Field("base64", description="Return format")
|
||||||
|
|
||||||
|
@app.post("/generate")
|
||||||
|
async def generate_flashcards(request: FlashcardRequest):
|
||||||
|
"""Generate flashcards and return as PDF bytes or base64"""
|
||||||
|
|
||||||
|
try:
|
||||||
|
# Parse numbers
|
||||||
|
numbers = parse_range(request.range, request.step)
|
||||||
|
|
||||||
|
if request.shuffle:
|
||||||
|
import random
|
||||||
|
if request.seed is not None:
|
||||||
|
random.seed(request.seed)
|
||||||
|
random.shuffle(numbers)
|
||||||
|
|
||||||
|
# Build config
|
||||||
|
config = {
|
||||||
|
'cards_per_page': request.cards_per_page,
|
||||||
|
'paper_size': request.paper_size,
|
||||||
|
'orientation': request.orientation,
|
||||||
|
'margins': request.margins,
|
||||||
|
'gutter': request.gutter,
|
||||||
|
'show_cut_marks': request.show_cut_marks,
|
||||||
|
'show_registration': request.show_registration,
|
||||||
|
'font_family': request.font_family,
|
||||||
|
'font_size': request.font_size,
|
||||||
|
'columns': request.columns,
|
||||||
|
'show_empty_columns': request.show_empty_columns,
|
||||||
|
'hide_inactive_beads': request.hide_inactive_beads,
|
||||||
|
'bead_shape': request.bead_shape,
|
||||||
|
'color_scheme': request.color_scheme,
|
||||||
|
'colored_numerals': request.colored_numerals,
|
||||||
|
'scale_factor': request.scale_factor,
|
||||||
|
}
|
||||||
|
|
||||||
|
# Create temp files
|
||||||
|
with tempfile.TemporaryDirectory() as tmpdir:
|
||||||
|
tmpdir_path = Path(tmpdir)
|
||||||
|
temp_typst = tmpdir_path / "flashcards.typ"
|
||||||
|
temp_pdf = tmpdir_path / "flashcards.pdf"
|
||||||
|
|
||||||
|
# Generate Typst file
|
||||||
|
generate_typst_file(numbers, config, temp_typst)
|
||||||
|
|
||||||
|
# Find project root and copy template
|
||||||
|
project_root = Path(__file__).parent.parent
|
||||||
|
templates_dir = project_root / "templates"
|
||||||
|
|
||||||
|
# Copy templates to temp dir for imports to work
|
||||||
|
import shutil
|
||||||
|
temp_templates = tmpdir_path / "templates"
|
||||||
|
shutil.copytree(templates_dir, temp_templates)
|
||||||
|
|
||||||
|
# Compile with Typst
|
||||||
|
result = subprocess.run(
|
||||||
|
["typst", "compile", str(temp_typst), str(temp_pdf)],
|
||||||
|
capture_output=True,
|
||||||
|
text=True,
|
||||||
|
cwd=str(tmpdir_path)
|
||||||
|
)
|
||||||
|
|
||||||
|
if result.returncode != 0:
|
||||||
|
raise HTTPException(status_code=500, detail=f"Typst compilation failed: {result.stderr}")
|
||||||
|
|
||||||
|
# Read PDF
|
||||||
|
with open(temp_pdf, "rb") as f:
|
||||||
|
pdf_bytes = f.read()
|
||||||
|
|
||||||
|
# Return based on format
|
||||||
|
if request.format == "pdf":
|
||||||
|
return Response(
|
||||||
|
content=pdf_bytes,
|
||||||
|
media_type="application/pdf",
|
||||||
|
headers={
|
||||||
|
"Content-Disposition": "attachment; filename=flashcards.pdf"
|
||||||
|
}
|
||||||
|
)
|
||||||
|
else: # base64
|
||||||
|
return {
|
||||||
|
"pdf": base64.b64encode(pdf_bytes).decode('utf-8'),
|
||||||
|
"count": len(numbers),
|
||||||
|
"numbers": numbers[:100] # Limit preview
|
||||||
|
}
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
raise HTTPException(status_code=500, detail=str(e))
|
||||||
|
|
||||||
|
@app.get("/health")
|
||||||
|
async def health_check():
|
||||||
|
"""Health check endpoint"""
|
||||||
|
return {"status": "healthy"}
|
||||||
|
|
||||||
|
@app.get("/")
|
||||||
|
async def root():
|
||||||
|
"""API documentation"""
|
||||||
|
return {
|
||||||
|
"name": "Soroban Flashcard Generator API",
|
||||||
|
"endpoints": {
|
||||||
|
"/generate": "POST - Generate flashcards",
|
||||||
|
"/health": "GET - Health check",
|
||||||
|
"/docs": "GET - Interactive API documentation"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
import uvicorn
|
||||||
|
uvicorn.run(app, host="0.0.0.0", port=8000)
|
||||||
|
|
@ -0,0 +1,137 @@
|
||||||
|
#!/usr/bin/env python3
|
||||||
|
"""
|
||||||
|
Python bridge for Node.js integration
|
||||||
|
Provides a clean function interface instead of CLI
|
||||||
|
"""
|
||||||
|
|
||||||
|
import json
|
||||||
|
import sys
|
||||||
|
import base64
|
||||||
|
import tempfile
|
||||||
|
from pathlib import Path
|
||||||
|
import subprocess
|
||||||
|
|
||||||
|
# Import our existing functions
|
||||||
|
from generate import parse_range, generate_typst_file
|
||||||
|
|
||||||
|
def generate_flashcards_json(config_json):
|
||||||
|
"""
|
||||||
|
Generate flashcards from JSON config
|
||||||
|
Returns base64 encoded PDF
|
||||||
|
"""
|
||||||
|
config = json.loads(config_json)
|
||||||
|
|
||||||
|
# Parse numbers
|
||||||
|
numbers = parse_range(
|
||||||
|
config.get('range', '0-9'),
|
||||||
|
config.get('step', 1)
|
||||||
|
)
|
||||||
|
|
||||||
|
# Handle shuffle
|
||||||
|
if config.get('shuffle', False):
|
||||||
|
import random
|
||||||
|
if 'seed' in config:
|
||||||
|
random.seed(config['seed'])
|
||||||
|
random.shuffle(numbers)
|
||||||
|
|
||||||
|
# Build Typst config
|
||||||
|
typst_config = {
|
||||||
|
'cards_per_page': config.get('cardsPerPage', 6),
|
||||||
|
'paper_size': config.get('paperSize', 'us-letter'),
|
||||||
|
'orientation': config.get('orientation', 'portrait'),
|
||||||
|
'margins': config.get('margins', {
|
||||||
|
'top': '0.5in',
|
||||||
|
'bottom': '0.5in',
|
||||||
|
'left': '0.5in',
|
||||||
|
'right': '0.5in'
|
||||||
|
}),
|
||||||
|
'gutter': config.get('gutter', '5mm'),
|
||||||
|
'show_cut_marks': config.get('showCutMarks', False),
|
||||||
|
'show_registration': config.get('showRegistration', False),
|
||||||
|
'font_family': config.get('fontFamily', 'DejaVu Sans'),
|
||||||
|
'font_size': config.get('fontSize', '48pt'),
|
||||||
|
'columns': config.get('columns', 'auto'),
|
||||||
|
'show_empty_columns': config.get('showEmptyColumns', False),
|
||||||
|
'hide_inactive_beads': config.get('hideInactiveBeads', False),
|
||||||
|
'bead_shape': config.get('beadShape', 'diamond'),
|
||||||
|
'color_scheme': config.get('colorScheme', 'monochrome'),
|
||||||
|
'colored_numerals': config.get('coloredNumerals', False),
|
||||||
|
'scale_factor': config.get('scaleFactor', 0.9),
|
||||||
|
}
|
||||||
|
|
||||||
|
# Generate PDF in temp directory
|
||||||
|
with tempfile.TemporaryDirectory() as tmpdir:
|
||||||
|
tmpdir_path = Path(tmpdir)
|
||||||
|
temp_typst = tmpdir_path / 'flashcards.typ'
|
||||||
|
temp_pdf = tmpdir_path / 'flashcards.pdf'
|
||||||
|
|
||||||
|
# Generate Typst file
|
||||||
|
project_root = Path(__file__).parent.parent
|
||||||
|
|
||||||
|
# Create temp Typst with correct imports
|
||||||
|
typst_content = f'''
|
||||||
|
#import "{project_root}/templates/flashcards.typ": generate-flashcards
|
||||||
|
|
||||||
|
#generate-flashcards(
|
||||||
|
{numbers},
|
||||||
|
cards-per-page: {typst_config['cards_per_page']},
|
||||||
|
paper-size: "{typst_config['paper_size']}",
|
||||||
|
orientation: "{typst_config['orientation']}",
|
||||||
|
margins: (
|
||||||
|
top: {typst_config['margins'].get('top', '0.5in')},
|
||||||
|
bottom: {typst_config['margins'].get('bottom', '0.5in')},
|
||||||
|
left: {typst_config['margins'].get('left', '0.5in')},
|
||||||
|
right: {typst_config['margins'].get('right', '0.5in')}
|
||||||
|
),
|
||||||
|
gutter: {typst_config['gutter']},
|
||||||
|
show-cut-marks: {str(typst_config['show_cut_marks']).lower()},
|
||||||
|
show-registration: {str(typst_config['show_registration']).lower()},
|
||||||
|
font-family: "{typst_config['font_family']}",
|
||||||
|
font-size: {typst_config['font_size']},
|
||||||
|
columns: {typst_config['columns']},
|
||||||
|
show-empty-columns: {str(typst_config['show_empty_columns']).lower()},
|
||||||
|
hide-inactive-beads: {str(typst_config['hide_inactive_beads']).lower()},
|
||||||
|
bead-shape: "{typst_config['bead_shape']}",
|
||||||
|
color-scheme: "{typst_config['color_scheme']}",
|
||||||
|
colored-numerals: {str(typst_config['colored_numerals']).lower()},
|
||||||
|
scale-factor: {typst_config['scale_factor']}
|
||||||
|
)
|
||||||
|
'''
|
||||||
|
|
||||||
|
with open(temp_typst, 'w') as f:
|
||||||
|
f.write(typst_content)
|
||||||
|
|
||||||
|
# Compile with Typst
|
||||||
|
result = subprocess.run(
|
||||||
|
['typst', 'compile', str(temp_typst), str(temp_pdf)],
|
||||||
|
capture_output=True,
|
||||||
|
text=True,
|
||||||
|
cwd=str(project_root)
|
||||||
|
)
|
||||||
|
|
||||||
|
if result.returncode != 0:
|
||||||
|
return json.dumps({
|
||||||
|
'error': f'Typst compilation failed: {result.stderr}'
|
||||||
|
})
|
||||||
|
|
||||||
|
# Read and encode PDF
|
||||||
|
with open(temp_pdf, 'rb') as f:
|
||||||
|
pdf_bytes = f.read()
|
||||||
|
|
||||||
|
return json.dumps({
|
||||||
|
'pdf': base64.b64encode(pdf_bytes).decode('utf-8'),
|
||||||
|
'count': len(numbers),
|
||||||
|
'numbers': numbers[:100] # Limit preview
|
||||||
|
})
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
# Read JSON from stdin, write JSON to stdout
|
||||||
|
# This allows clean function-like communication
|
||||||
|
for line in sys.stdin:
|
||||||
|
try:
|
||||||
|
result = generate_flashcards_json(line.strip())
|
||||||
|
print(result)
|
||||||
|
sys.stdout.flush()
|
||||||
|
except Exception as e:
|
||||||
|
print(json.dumps({'error': str(e)}))
|
||||||
|
sys.stdout.flush()
|
||||||
Loading…
Reference in New Issue