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

class ChallengeSubmit(BaseModel):
    language: Optional[str] = None
    code: Optional[str] = None
    selected_option: Optional[int] = None
    selected_options: Optional[List[int]] = None

class ChallengeCreate(BaseModel):
    title: str
    description: str
    difficulty: str = "easy"
    category: str = "arrays"
    type: str = "coding"
    options: List[str] = []
    correct_option: Optional[int] = None
    questions: Optional[List[Dict[str, Any]]] = []
    company_tag: str = ""
    is_interview_question: bool = False
    starter_code: Dict[str, str] = {}
    test_cases: List[Dict[str, Any]] = []