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

class CodeScanRequest(BaseModel):
    repo_url: str
    github_token: str
    # branch or commit hash optional
    branch: Optional[str] = "main"

class VulnerabilityIssue(BaseModel):
    file_path: str
    severity: str  # High, Medium, Low, Critical
    issue: str
    explanation: str
    suggested_fix: Optional[str] = None
    line_number: Optional[int] = None

class CodeScanResponse(BaseModel):
    scan_id: str
    repo_url: str
    summary: str
    issues: List[VulnerabilityIssue]

class CodeChatRequest(BaseModel):
    scan_id: str
    message: str

class CodeChatResponse(BaseModel):
    reply: str