helpers/models.py

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


# ────────────────────────────── Response Models ──────────────────────────────
class ProjectResponse(BaseModel):
    project_id: str
    user_id: str
    name: str
    description: str
    created_at: str
    updated_at: str

class ProjectsListResponse(BaseModel):
    projects: List[ProjectResponse]

class ChatMessageResponse(BaseModel):
    user_id: str
    project_id: str
    role: str
    content: str
    timestamp: float
    created_at: str
    sources: Optional[List[Dict[str, Any]]] = None
    is_report: Optional[bool] = False

class ChatHistoryResponse(BaseModel):
    messages: List[ChatMessageResponse]

class MessageResponse(BaseModel):
    message: str
    quiz: Optional[Dict[str, Any]] = None
    results: Optional[Dict[str, Any]] = None

class UploadResponse(BaseModel):
    job_id: str
    status: str
    total_files: Optional[int] = None

class FileSummaryResponse(BaseModel):
    filename: str
    summary: str

class ChatAnswerResponse(BaseModel):
    answer: str
    sources: List[Dict[str, Any]]
    relevant_files: Optional[List[str]] = None
    session_name: Optional[str] = None
    session_id: Optional[str] = None

class HealthResponse(BaseModel):
    ok: bool

class ReportResponse(BaseModel):
    filename: str
    report_markdown: str
    sources: List[Dict[str, Any]]

class StatusUpdateResponse(BaseModel):
    status: str
    message: str
    progress: Optional[int] = None

class QuizQuestionResponse(BaseModel):
    type: str  # "mcq" or "self_reflect"
    question: str
    options: Optional[List[str]] = None  # For MCQ questions
    correct_answer: Optional[int] = None  # For MCQ questions
    topic: str
    complexity: str

class QuizResponse(BaseModel):
    quiz_id: str
    user_id: str
    project_id: str
    questions: List[QuizQuestionResponse]
    time_limit: int
    documents: List[str]
    created_at: str
    status: str

class QuizResultResponse(BaseModel):
    questions: List[Dict[str, Any]]
    total_score: float
    correct_count: int
    partial_count: int
    incorrect_count: int