|
|
from fastapi import FastAPI |
|
|
from pydantic import BaseModel |
|
|
from typing import Dict |
|
|
|
|
|
from src.core import * |
|
|
|
|
|
app = FastAPI( |
|
|
title="Insight Finder", |
|
|
description="Find relevant technologies from a problem", |
|
|
) |
|
|
|
|
|
class InputData(BaseModel): |
|
|
problem: str |
|
|
|
|
|
class InputConstraints(BaseModel): |
|
|
data: Dict[str, str] |
|
|
|
|
|
class OutputData(BaseModel): |
|
|
technologies: list |
|
|
|
|
|
@app.post("/process", response_model=OutputData) |
|
|
async def process(data: InputData): |
|
|
result = process_input(data, global_tech, global_tech_embeddings) |
|
|
return {"technologies": result} |
|
|
|
|
|
|
|
|
@app.post("/process", response_model=OutputData) |
|
|
async def process(constraints: InputConstraints): |
|
|
result = process_input(constraints, global_tech, global_tech_embeddings) |
|
|
return {"technologies": result} |
|
|
|
