Update models.py

models.py

@@ -1,46 +1,37 @@
-from typing import List, Optional, Union, Dict
-from pydantic import BaseModel, Field
+from typing import List, Optional
+from pydantic import BaseModel
 
 class Location(BaseModel):
-  name: str
-  lat: float
-  lon: float
-  relevance: str = Field(description="Why this location matters in the historical context")
-
-class TemporalZoom(BaseModel):
-  level: str = Field(description="Level of temporal detail (century/decade/year)")
-  range: List[int] = Field(description="Start and end years for the time period")
-  explanation: str = Field(description="Historical significance of this time period")
-
-class GeographicalZoom(BaseModel):
-  level: str = Field(description="Level of geographical detail (continent/country/city)")
-  locations: List[Location]
-  explanation: str = Field(description="Significance of these locations")
-
-class StyleZoom(BaseModel):
-  level: str = Field(description="Level of style detail (movement/sub_movement/specific)")
-  options: List[str] = Field(description="Available artistic styles for selection")
-  explanation: str = Field(description="Historical context for the artistic styles")
+  latitude: float
+  longitude: float
+  place_name: str
+  historical_significance: str
 
-class ZoomConfiguration(BaseModel):
-  in_zoom: Optional[Union[TemporalZoom, GeographicalZoom, StyleZoom]]
-  out_zoom: Optional[Union[TemporalZoom, GeographicalZoom, StyleZoom]]
-
-class AxisImpact(BaseModel):
-  temporal: Optional[str] = Field(default=None, description="How time period affects this axis")
-  geographical: Optional[str] = Field(default=None, description="How location affects this axis")
-  style: Optional[str] = Field(default=None, description="How artistic style affects this axis")
-
-class AxisConfiguration(BaseModel):
-  component: str
-  current_zoom: Union[TemporalZoom, GeographicalZoom, StyleZoom]
-  available_zooms: Optional[ZoomConfiguration] = None
-  impacted_by: Optional[AxisImpact] = None
+class ArtPeriod(BaseModel):
+  name: str
+  start_year: int
+  end_year: int
+  key_characteristics: List[str]
 
-class Analysis(BaseModel):
-  query_focus: str = Field(description="Main subject of the query")
-  historical_context: str = Field(description="Brief historical context")
+class Artist(BaseModel):
+  name: str
+  birth_year: Optional[int]
+  death_year: Optional[int]
+  nationality: str
+  primary_medium: List[str]
+
+class Artwork(BaseModel):
+  title: str
+  year: Optional[int]
+  medium: str
+  current_location: Optional[Location]
+  significance: str
 
 class ArtHistoryResponse(BaseModel):
-  analysis: Analysis
-  axis_configurations: Dict[str, AxisConfiguration]
+  time_period: ArtPeriod
+  locations: List[Location]
+  key_artists: List[Artist]
+  notable_works: List[Artwork]
+  cultural_context: str
+  historical_events: List[str]
+  influence_on_later_periods: str