bolade
76 lines
2.0 KiB
Python
76 lines
2.0 KiB
Python
import io
|
|
|
|
import pandas as pd
|
|
from db.db import db_dependency, init_database
|
|
from dotenv import load_dotenv
|
|
from fastapi import FastAPI, File, Form, UploadFile
|
|
from pydantic import BaseModel
|
|
from routers import companies, investors
|
|
from schemas.router_schemas import InvestorList
|
|
from services.llm_parser import InvestorProcessor
|
|
from services.querying import QueryProcessor
|
|
|
|
load_dotenv()
|
|
init_database()
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
# Request models
|
|
class QueryRequest(BaseModel):
|
|
question: str
|
|
|
|
class Config:
|
|
json_schema_extra = {
|
|
"example": {
|
|
"question": "Find me deep tech investors that do deals in Europe under 5 million."
|
|
}
|
|
}
|
|
|
|
|
|
@app.get("/")
|
|
def health():
|
|
return {"Hello": "World"}
|
|
|
|
|
|
@app.post("/parse-csv", tags=["CSV Upload"], response_model=list[dict])
|
|
async def parse_csv(db: db_dependency, file: UploadFile = File(...), is_investor: int = Form(...)):
|
|
# Read uploaded CSV with pandas
|
|
content = await file.read()
|
|
df = pd.read_csv(io.StringIO(content.decode("utf-8")))
|
|
|
|
# Process the dataframe
|
|
processor = InvestorProcessor()
|
|
|
|
if is_investor == 1:
|
|
results = await processor.parse_investors(df)
|
|
else:
|
|
results = await processor.parse_companies(df)
|
|
|
|
# Convert Pydantic objects to dictionaries
|
|
return [r.model_dump() for r in results]
|
|
|
|
|
|
@app.post("/query", response_model=InvestorList, tags=["Querying"])
|
|
async def query_investors(request: QueryRequest):
|
|
"""
|
|
Query investors using natural language.
|
|
|
|
Supports queries like:
|
|
- "Show me seed stage investors"
|
|
- "Find fintech investors in Silicon Valley"
|
|
- "Growth stage investors with $5M+ check sizes"
|
|
- "Healthcare investors in Europe"
|
|
"""
|
|
processor = QueryProcessor()
|
|
results = processor.process_query(request.question)
|
|
return results
|
|
|
|
|
|
app.include_router(investors.router)
|
|
app.include_router(companies.router)
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
|
|
uvicorn.run(app="main:app", host="localhost", port=8000, reload=True)
|