Update models.py

models.py

@@ -1,9 +1,34 @@
 from pydantic import BaseModel, Field
+from typing import List, Optional
 from datetime import datetime
-from typing import Optional
 
+# Existing User Sync Model
 class UserSync(BaseModel):
     clerk_id: str
-    email: str
+    email: Optional[str] = None
     name: Optional[str] = None
-    created_at: datetime = Field(default_factory=datetime.now)
+    created_at: Optional[datetime] = None
+
+# --- Testimony Validator & Map-Reduce Models ---
+
+class TimelineEvent(BaseModel):
+    timeframe: str = Field(description="Date, time, or period of the event")
+    event_description: str = Field(description="Description of what occurred")
+    source_quote: str = Field(description="Direct verbatim quote from the transcript supporting this event")
+
+class Timeline(BaseModel):
+    party: str = Field(description="Name or role of the party (e.g., Client, Accused)")
+    events: List[TimelineEvent] = Field(default_factory=list, description="Chronological list of events")
+
+class Discrepancy(BaseModel):
+    type: str = Field(description="Classification: 'Direct Conflict' or 'Omission'")
+    timeframe: str = Field(description="Timeframe or date when the conflicting event occurred")
+    client_version: Optional[str] = Field(default=None, description="Client's statement of the event")
+    accused_version: Optional[str] = Field(default=None, description="Accused party's statement of the event")
+    analysis: str = Field(description="Detailed forensic reasoning explaining why this is a conflict or omission")
+    severity: str = Field(description="Severity impact: 'High', 'Medium', or 'Low'")
+
+class ComparativeAnalysisResult(BaseModel):
+    client_timeline: Timeline
+    accused_timeline: Timeline
+    discrepancies: List[Discrepancy] = Field(default_factory=list)