Improve AI logic

services/training/app/schemas/training.py

@@ -5,7 +5,7 @@ Includes all request/response schemas used by the API endpoints
 """
 
 from pydantic import BaseModel, Field, validator
-from typing import List, Optional, Dict, Any, Union
+from typing import List, Optional, Dict, Any, Union, Tuple
 from datetime import datetime
 from enum import Enum
 from uuid import UUID
@@ -37,6 +37,9 @@ class SingleProductTrainingRequest(BaseModel):
     daily_seasonality: bool = Field(True, description="Enable daily seasonality")
     weekly_seasonality: bool = Field(True, description="Enable weekly seasonality")
     yearly_seasonality: bool = Field(True, description="Enable yearly seasonality")
+    
+    # Location parameters
+    bakery_location: Optional[Tuple[float, float]] = Field(None, description="Bakery coordinates (latitude, longitude)")
 
 class DateRangeInfo(BaseModel):
     """Schema for date range information"""