import json
import os
from typing import List
from fastapi import FastAPI, HTTPException
from fastapi.responses import HTMLResponse
from pydantic import BaseModel
from model import generate_structure

app = FastAPI()
STRUCTURE_FILE = "structures.json"
structure_store = []

class Prompt(BaseModel):
    text: str

class Structure(BaseModel):
    name: str
    description: str
    required_blocks: List[str]
    tags: List[str]

@app.post("/prompt")
async def prompt(prompt: Prompt):
    result = generate_structure(prompt.text)
    return {"response": result}