chabo_whisp / utils /main_processor.py
leavoigt's picture
Update utils/main_processor.py
150dbdb verified
import json
import os
from utils.whisp_api_client import whisp_request
from utils.whisp_preprocessing import preprocess_whisp_data
# Import secrets
whisp_api_key = os.environ.get("WHISP_API_KEY")
print(f"API key loaded: {whisp_api_key is not None}")
# Define main function to fetch geojson, do API call and get statistics
def get_statistics(file):
# Open the geojson file uploaded
if file is None:
return "Geojson file empty"
try:
with open(file, 'r', encoding='utf-8') as f:
geojson_file = json.load(f)
except Exception as e:
return {"error": str(e)}
# Do the API call
whisp_response = whisp_request(geojson_file, api_key=whisp_api_key)
print(whisp_response)
# Preprocess
preprocessed_response = preprocess_whisp_data(whisp_response)
# Do preprocessing
return preprocessed_response