app/models/schemas.py

from pydantic import BaseModel
from typing import Optional, List, Dict, Any
from datetime import datetime

class StockDataRequest(BaseModel):
    symbol: str

class ComplianceAnalysisRequest(BaseModel):
    financial_data: str
    company_info: str
    data_source: str
    accounting_standards: str
    regulatory_frameworks: List[str]

class StockDataResponse(BaseModel):
    success: bool
    data: Optional[Dict[str, Any]] = None
    formatted_data: Optional[str] = None
    error: Optional[str] = None

class ComplianceAnalysisResponse(BaseModel):
    success: bool
    result: Optional[Dict[str, Any]] = None
    error: Optional[str] = None

class AnalysisHistoryItem(BaseModel):
    timestamp: datetime
    data_source: str
    company_info: str
    status: str

class RiskAssessment(BaseModel):
    risk_area: str
    risk_level: str
    impact_score: int
    likelihood_score: int

# Add this new schema for CSV generation
class GenerateCSVRequest(BaseModel):
    data_source: str
    company_info: str
    accounting_standards: str
    regulatory_frameworks: List[str]
    result: Dict[str, Any]