update

app.py

@@ -12,7 +12,7 @@ from pyproj import Proj
 app = FastAPI()
 
 PROJECT_ROOT = os.path.realpath(os.path.join(os.path.dirname(__file__), ".."))
-STATION_CSV = os.path.join(PROJECT_ROOT, "tests/stations_hawaii.csv")
+# STATION_CSV = os.path.join(PROJECT_ROOT, "tests/stations_hawaii.csv")
 # STATION_CSV = os.path.join(PROJECT_ROOT, "tests/stations.csv")  ## ridgecrest
 
 
@@ -57,14 +57,14 @@ config["bfgs_bounds"] = [list(config[x]) for x in config["dims"]] + [[None, None
 for k, v in config.items():
     print(f"{k}: {v}")
 
-## read stations
-stations = pd.read_csv(STATION_CSV, delimiter="\t")
-stations = stations.rename(columns={"station": "id"})
-stations["x(km)"] = stations["longitude"].apply(lambda x: (x - config["center"][0]) * config["degree2km"])
-stations["y(km)"] = stations["latitude"].apply(lambda x: (x - config["center"][1]) * config["degree2km"])
-stations["z(km)"] = stations["elevation(m)"].apply(lambda x: -x / 1e3)
+# ## read stations
+# stations = pd.read_csv(STATION_CSV, delimiter="\t")
+# stations = stations.rename(columns={"station": "id"})
+# stations["x(km)"] = stations["longitude"].apply(lambda x: (x - config["center"][0]) * config["degree2km"])
+# stations["y(km)"] = stations["latitude"].apply(lambda x: (x - config["center"][1]) * config["degree2km"])
+# stations["z(km)"] = stations["elevation(m)"].apply(lambda x: -x / 1e3)
 
-print(stations)
+# print(stations)
 
 
 class Data(BaseModel):
@@ -105,21 +105,16 @@ def run_gamma(picks, config, stations):
     return catalogs, picks_gamma
 
 
-@app.post("/predict_stream")
-def predict(data: Pick):
+# @app.post("/predict_stream")
+# def predict(data: Pick):
 
-    picks = pd.DataFrame(data.picks)
-    if len(picks) == 0:
-        return {"catalog": [], "picks": []}
+#     picks = pd.DataFrame(data.picks)
+#     if len(picks) == 0:
+#         return {"catalog": [], "picks": []}
 
-    catalogs, picks_gamma = run_gamma(data, config, stations)
+#     catalogs, picks_gamma = run_gamma(data, config, stations)
 
-    if use_kafka:
-        print("Push events to kafka...")
-        for event in catalogs.to_dict(orient="records"):
-            producer.send("gmma_events", key=event["time"], value=event)
-
-    return {"catalog": catalogs.to_dict(orient="records"), "picks": picks_gamma.to_dict(orient="records")}
+#     return {"catalog": catalogs.to_dict(orient="records"), "picks": picks_gamma.to_dict(orient="records")}
 
 
 @app.post("/predict")
@@ -161,11 +156,6 @@ def predict(data: Data):
 
     catalogs, picks_gamma = run_gamma(picks, config, stations)
 
-    if use_kafka:
-        print("Push events to kafka...")
-        for event in catalogs.to_dict(orient="records"):
-            producer.send("gamma_events", key=event["time"], value=event)
-
     return {"catalog": catalogs.to_dict(orient="records"), "picks": picks_gamma.to_dict(orient="records")}