Spaces:
Running
Running
File size: 2,441 Bytes
f9ab266 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
from fastapi import FastAPI, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse
from getReceiverData import analyze_telecom_data, get_summary_by_state
import uvicorn
import pandas as pd
import numpy as np
app = FastAPI()
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
def sanitize_json(data):
if isinstance(data, dict):
return {k: sanitize_json(v) for k, v in data.items()}
elif isinstance(data, list):
return [sanitize_json(v) for v in data]
elif isinstance(data, (np.integer, np.int64)):
return int(data)
elif isinstance(data, (np.floating, np.float64)):
return float(data)
elif isinstance(data, (np.bool_)):
return bool(data)
elif pd.isna(data):
return None
else:
return data
@app.get("/api/performance-summary")
def get_performance_summary():
results = analyze_telecom_data()
summary_data = sanitize_json({
"summary": {
"max_connected_state": results["max_connected_state"],
"max_connected_state_count": results["max_connected_state_count"],
"best_signal_state": results["best_signal_state"],
"best_signal_avg": results["best_signal_avg"],
"worst_call_drop_state": results["worst_call_drop_state"],
"worst_call_drop_rate": results["worst_call_drop_rate"]
},
"state_metrics": {
"device_count": results["state_device_count"],
"signal_strength": results["state_signal_strength"],
"call_drop_rate": results["state_call_drop_rate"]
},
"top_towns": {
"by_device": results["top_town_by_device"],
"by_signal": results["top_town_by_signal"],
"by_call_drop": results["worst_town_by_call_drop"]
},
"town_summary": results["town_summary"]
})
return summary_data
@app.get("/api/summary-by-state/{state_name}")
def get_state_summary(state_name):
result = get_summary_by_state(state_name)
json_data = sanitize_json(result.to_dict(orient='records'))
return {"town_summary": json_data}
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=8000)
#get_telecom_summary()
|