TradeMate Dev
99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
from fastapi import APIRouter, HTTPException, Depends, Header
|
|
from typing import Optional
|
|
from pydantic import BaseModel
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from app.database import get_db
|
|
from app.services.marketing import MarketingService
|
|
from app.services.preference import UserPreferenceService
|
|
from app.core.security import decode_token
|
|
from app.api.v1.deps import get_current_user_id
|
|
from app.config import settings
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
class MarketingRequest(BaseModel):
|
|
product_name: str
|
|
description: str
|
|
category: Optional[str] = None
|
|
price: Optional[str] = None
|
|
keywords: Optional[list] = None
|
|
target: str = "US importers"
|
|
style: str = "professional"
|
|
language: str = "en"
|
|
count: int = 3
|
|
|
|
|
|
class KeywordsRequest(BaseModel):
|
|
product_name: str
|
|
description: str
|
|
category: Optional[str] = None
|
|
language: str = "en"
|
|
count: int = 10
|
|
|
|
|
|
class CompetitorRequest(BaseModel):
|
|
product_name: str
|
|
description: str
|
|
category: Optional[str] = None
|
|
market: str = "US"
|
|
|
|
|
|
@router.post("/generate")
|
|
async def generate_marketing(
|
|
data: MarketingRequest,
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
service = MarketingService()
|
|
pref_service = UserPreferenceService(db)
|
|
pref_context = await pref_service.get_preference_context(user_id, "marketing")
|
|
|
|
product_info = {
|
|
"name": data.product_name,
|
|
"description": data.description,
|
|
"category": data.category,
|
|
"price": data.price,
|
|
"keywords": data.keywords,
|
|
}
|
|
results = await service.generate(product_info, data.target, data.style, data.language, data.count, pref_context)
|
|
|
|
return {
|
|
"results": results,
|
|
"product": data.product_name,
|
|
"target": data.target,
|
|
"count": len(results),
|
|
}
|
|
|
|
|
|
@router.post("/keywords")
|
|
async def generate_keywords(data: KeywordsRequest, authorization: str = Header(None)):
|
|
if not authorization:
|
|
raise HTTPException(status_code=401, detail="Missing token")
|
|
|
|
service = MarketingService()
|
|
product_info = {
|
|
"name": data.product_name,
|
|
"description": data.description,
|
|
"category": data.category,
|
|
}
|
|
keywords = await service.generate_keywords(product_info, data.language, data.count)
|
|
|
|
return {"keywords": keywords, "product": data.product_name}
|
|
|
|
|
|
@router.post("/competitor-analysis")
|
|
async def competitor_analysis(data: CompetitorRequest, authorization: str = Header(None)):
|
|
if not authorization:
|
|
raise HTTPException(status_code=401, detail="Missing token")
|
|
|
|
service = MarketingService()
|
|
product_info = {
|
|
"name": data.product_name,
|
|
"description": data.description,
|
|
"category": data.category,
|
|
}
|
|
analysis = await service.analyze_competitors(product_info, data.market)
|
|
|
|
return {"analysis": analysis, "product": data.product_name, "market": data.market}
|