bolade
81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
from typing import Optional
|
|
|
|
from db.db import get_db
|
|
from db.models import InvestorTable, ProjectTable
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from schemas.insight_schema import InsightResponse
|
|
from services.compatibility_score import (
|
|
calculate_project_investor_compatibility,
|
|
generate_compatibility_explanation,
|
|
)
|
|
from services.insight import QueryProcessor
|
|
from sqlalchemy.orm import Session
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get(
|
|
"/insights/{investor_id}", response_model=InsightResponse, tags=["Insights"]
|
|
)
|
|
async def get_insights(
|
|
investor_id: int, project_id: Optional[int] = None, db: Session = Depends(get_db)
|
|
):
|
|
"""
|
|
Get investor insights including investment pattern analysis, market position,
|
|
and optionally compatibility score with a project.
|
|
|
|
Args:
|
|
investor_id: The ID of the investor to analyze
|
|
project_id: Optional project ID to calculate compatibility score
|
|
|
|
Returns:
|
|
InsightResponse with investment_pattern_analysis, market_position,
|
|
and compatibility_score (if project_id provided)
|
|
"""
|
|
# Get investor from database
|
|
investor = db.query(InvestorTable).filter(InvestorTable.id == investor_id).first()
|
|
if not investor:
|
|
raise HTTPException(
|
|
status_code=404, detail=f"Investor with id {investor_id} not found"
|
|
)
|
|
|
|
# Initialize the query processor for insights
|
|
query_processor = QueryProcessor()
|
|
|
|
# Get investment pattern analysis and market position using web search
|
|
insights = await query_processor.get_investor_insights(
|
|
investor_name=investor.name,
|
|
investor_website=investor.website,
|
|
investor_description=investor.description,
|
|
investor_headquarters=investor.headquarters,
|
|
investment_thesis=investor.investment_thesis,
|
|
portfolio_highlights=investor.portfolio_highlights,
|
|
)
|
|
|
|
# Calculate compatibility score if project_id is provided
|
|
compatibility_score = None
|
|
if project_id:
|
|
project = db.query(ProjectTable).filter(ProjectTable.id == project_id).first()
|
|
if not project:
|
|
raise HTTPException(
|
|
status_code=404, detail=f"Project with id {project_id} not found"
|
|
)
|
|
|
|
# Calculate the compatibility score
|
|
score = calculate_project_investor_compatibility(
|
|
project, investor, use_funds=True
|
|
)
|
|
|
|
# Generate detailed explanation
|
|
compatibility_score = generate_compatibility_explanation(
|
|
project, investor, score, use_funds=True
|
|
)
|
|
else:
|
|
compatibility_score = "Select a project to see compatibility analysis"
|
|
|
|
return InsightResponse(
|
|
investment_pattern_analysis=insights["investment_pattern_analysis"],
|
|
market_position=insights["market_position"],
|
|
compatibility_score=compatibility_score,
|
|
)
|