Removing validation

app.py

@@ -10,7 +10,6 @@ from staticmap import StaticMap, CircleMarker, Polygon
 from diffusers import ControlNetModel, StableDiffusionControlNetInpaintPipeline
 import spaces
 import logging
-from pydantic import BaseModel, ValidationError, Field
 from typing import List, Union
 
 # Set up logging
@@ -21,20 +20,6 @@ logger = logging.getLogger(__name__)
 openai_client = OpenAI(api_key=os.environ['OPENAI_API_KEY'])
 geolocator = Nominatim(user_agent="geoapi")
 
-# Define Pydantic models for GeoJSON validation
-class Geometry(BaseModel):
-    type: str
-    coordinates: Union[List[float], List[List[List[float]]]]
-
-class Feature(BaseModel):
-    type: str = "Feature"
-    properties: dict
-    geometry: Geometry
-
-class FeatureCollection(BaseModel):
-    type: str = "FeatureCollection"
-    features: List[Feature]
-
 # Function to fetch coordinates
 @spaces.GPU
 def get_geo_coordinates(location_name):
@@ -124,7 +109,6 @@ Generate similar JSON for the following query:
 # Generate GeoJSON from OpenAI response
 @spaces.GPU
 def generate_geojson(response):
-    #log response
     logger.info(f"OpenAI response: {response}")
     feature_type = response['output']['feature_representation']['type']
     city_names = response['output']['feature_representation']['cities']
@@ -135,7 +119,7 @@ def generate_geojson(response):
     # Fetch coordinates for cities
     for city in city_names:
         try:
-            coord = get_geo_coordinates(city)  # Function to fetch city coordinates
+            coord = get_geo_coordinates(city)
             if coord:
                 coordinates.append(coord)
             else:
@@ -165,13 +149,7 @@ def generate_geojson(response):
         ],
     }
 
-    # Validate the GeoJSON
-    try:
-        validated_geojson = FeatureCollection(**geojson_data)
-        return validated_geojson.dict()
-    except ValidationError as e:
-        logger.error(f"Invalid GeoJSON data: {e}")
-        raise ValueError("Generated GeoJSON is invalid.")
+    return geojson_data
 
 # Generate static map image
 @spaces.GPU