from fastapi import FastAPI
from pydantic import BaseModel

app = FastAPI()

class Peticion(BaseModel):
    tipo_modelo: str
    prompt: str

@app.get("/")
def home():
    return {"status": "online", "mensaje": "Servidor de IA listo"}

@app.post("/inferencia")
async def inferencia(data: Peticion):
    # Aquí es donde elegirás el modelo
    if data.tipo_modelo == "texto":
        # Aquí llamarías a tu función de texto
        respuesta = f"Procesando texto: {data.prompt}"
    else:
        respuesta = "Modelo no identificado"
        
    return {"resultado": respuesta}