Add orientation feature to poster generation with validation

app.py

@@ -45,8 +45,14 @@ class DistanceType(str, Enum):
     NETWORK = "network"
 
 
+class Orientation(str, Enum):
+    PORTRAIT = "portrait"
+    LANDSCAPE = "landscape"
+
+
 NETWORK_TYPES = [item.value for item in NetworkType]
 DIST_TYPES = [item.value for item in DistanceType]
+ORIENTATION_TYPES = [item.value for item in Orientation]
 
 
 _REPO_ROOT = Path(__file__).resolve().parent
@@ -60,6 +66,7 @@ class GenerateRequest(BaseModel):
     dpi: int
     network_type: NetworkType
     dist_type: DistanceType
+    orientation: Orientation
     home_point: str | None = None
     dot_size: float | None = None
 
@@ -111,6 +118,17 @@ class GenerateRequest(BaseModel):
         except ValueError as exc:
             raise ValueError("Invalid distance type.") from exc
 
+    @field_validator("orientation", mode="before")
+    @classmethod
+    def _validate_orientation(cls, value: str | Orientation) -> Orientation:
+        if isinstance(value, Orientation):
+            return value
+        value = (value or "").strip()
+        try:
+            return Orientation(value)
+        except ValueError as exc:
+            raise ValueError("Invalid orientation.") from exc
+
     @field_validator("dot_size", mode="before")
     @classmethod
     def _validate_dot_size(cls, value: str | float | None) -> float | None:
@@ -246,6 +264,7 @@ def generate(
     dpi: int,
     network_type: str,
     dist_type: str,
+    orientation: str,
     home_point: str | None,
     dot_size: float | None,
 ) -> str:
@@ -258,6 +277,7 @@ def generate(
             dpi=dpi,
             network_type=network_type,
             dist_type=dist_type,
+            orientation=orientation,
             home_point=home_point,
             dot_size=dot_size,
         )
@@ -294,6 +314,7 @@ def generate(
         dpi=request.dpi,
         dot=dot_coords,
         dot_size=request.dot_size if request.dot_size is not None else DEFAULT_DOT_SIZE,
+        orientation=request.orientation.value,
     )
     return output_path
 
@@ -469,6 +490,11 @@ def build_demo() -> gr.Blocks:
                             choices=DIST_TYPES,
                             value="bbox",
                         )
+                        orientation = gr.Dropdown(
+                            label="Orientation",
+                            choices=ORIENTATION_TYPES,
+                            value="portrait",
+                        )
                         home_point = gr.Textbox(
                             label="Home point (lat, lon)",
                             placeholder="31.3, 2.3",
@@ -500,6 +526,7 @@ def build_demo() -> gr.Blocks:
                 dpi,
                 network_type,
                 dist_type,
+                orientation,
                 home_point,
                 dot_size,
             ],

create_map_poster.py

@@ -359,6 +359,15 @@ def _coerce_coordinates(point: Coordinates | Sequence[float]) -> Coordinates:
     raise TypeError("point must be Coordinates or (lat, lon) sequence")
 
 
+def _normalize_orientation(value: str | None) -> str:
+    if not value:
+        return "portrait"
+    cleaned = str(value).strip().lower()
+    if cleaned in {"portrait", "landscape"}:
+        return cleaned
+    raise ValueError("Orientation must be 'portrait' or 'landscape'.")
+
+
 def create_poster(
     city: str,
     country: str,
@@ -371,6 +380,7 @@ def create_poster(
     dpi: int = 300,
     dot: Coordinates | Sequence[float] | None = None,
     dot_size: float = 60,
+    orientation: str = "portrait",
 ) -> None:
     print(f"\nGenerating map for {city}, {country}...")
     theme = _require_theme()
@@ -410,7 +420,9 @@ def create_poster(
     
     # 2. Setup Plot
     print("Rendering map...")
-    fig, ax = plt.subplots(figsize=(12, 16), facecolor=theme.bg)
+    orientation = _normalize_orientation(orientation)
+    figsize = (12, 16) if orientation == "portrait" else (16, 12)
+    fig, ax = plt.subplots(figsize=figsize, facecolor=theme.bg)
     ax.set_facecolor(theme.bg)
     ax.set_position([0, 0, 1, 1])