TradeMate Dev
62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
from fastapi import APIRouter, HTTPException
|
|
from typing import Optional, Dict, Any
|
|
from pydantic import BaseModel
|
|
from app.services.discovery import DiscoveryService
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
class SearchRequest(BaseModel):
|
|
product_description: str
|
|
target_market: str = "US"
|
|
|
|
|
|
class AnalyzeRequest(BaseModel):
|
|
company_url: str
|
|
product_description: str
|
|
|
|
|
|
class OutreachRequest(BaseModel):
|
|
company: Dict[str, Any]
|
|
product: Dict[str, Any]
|
|
|
|
|
|
@router.post("/search")
|
|
async def search_leads(req: SearchRequest):
|
|
if not req.product_description.strip():
|
|
raise HTTPException(status_code=400, detail="请填写产品描述")
|
|
svc = DiscoveryService()
|
|
try:
|
|
result = await svc.search(req.product_description, req.target_market)
|
|
return {"success": True, "data": result}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"搜索失败: {str(e)}")
|
|
|
|
|
|
@router.post("/analyze")
|
|
async def analyze_company(req: AnalyzeRequest):
|
|
if not req.company_url.strip():
|
|
raise HTTPException(status_code=400, detail="请填写公司网址")
|
|
if not req.product_description.strip():
|
|
raise HTTPException(status_code=400, detail="请填写产品描述")
|
|
svc = DiscoveryService()
|
|
try:
|
|
result = await svc.analyze(req.company_url, req.product_description)
|
|
return {"success": True, "data": result}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"分析失败: {str(e)}")
|
|
|
|
|
|
@router.post("/outreach")
|
|
async def generate_outreach(req: OutreachRequest):
|
|
if not req.company.get("name"):
|
|
raise HTTPException(status_code=400, detail="请填写公司名称")
|
|
if not req.product.get("name"):
|
|
raise HTTPException(status_code=400, detail="请填写产品名称")
|
|
svc = DiscoveryService()
|
|
try:
|
|
result = await svc.outreach(req.company, req.product)
|
|
return {"success": True, "data": result}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"生成失败: {str(e)}")
|