generator_endpoint.py

from fastapi import FastAPI
from transformers import pipeline

app = FastAPI()
chart_generator = pipeline("text-generation", model="tmmdev/chart-generator")

@app.post("/generate")
async def generate_charts(data: dict):
    charts = chart_generator(data)
    return {
        "interactive_charts": charts.charts,
        "pattern_images": charts.patterns
    }