insight-finder / app.py
heymenn's picture
Update app.py
9435ff3 verified
raw
history blame contribute delete
456 Bytes
from fastapi import FastAPI
from pydantic import BaseModel
from src.core import process_input
app = FastAPI(
title="Insight Finder",
description="Find relevant technologies from a problem",
)
class InputData(BaseModel):
problem: str
class OutputData(BaseModel):
technologies: list
@app.post("/process", response_model=OutputData)
async def process(data: InputData):
result = process_input(data)
return {"technologies": result}