Fixed imports

phasehunter/app.py

@@ -17,6 +17,8 @@ from obspy.taup.helper_classes import SlownessModelError
 
 from obspy.clients.fdsn.header import URL_MAPPINGS
 
+import matplotlib.pyplot as plt
+
 def make_prediction(waveform):
     waveform = np.load(waveform)
     processed_input = prepare_waveform(waveform)
@@ -83,7 +85,21 @@ def download_data(timestamp, eq_lat, eq_lon, client_name, radius_km):
     assert eq_lat - window > -90 and eq_lat + window < 90, "Latitude out of bounds"
     assert eq_lon - window > -180 and eq_lon + window < 180, "Longitude out of bounds"
 
+    starttime = obspy.UTCDateTime(timestamp)
+    endtime = startime + 120
+
+    inv = client.get_stations(network="*", station="*", location="*", channel="*H*", 
+                          starttime=obspy.UTCDateTime(starttime), endtime=endtime, 
+                          minlatitude=eq_lat-window, maxlatitude=eq_lat+window,
+                          minlongitude=eq_lon-window, maxlongitude=eq_lon+window, 
+                          level='channel')
     
+    for network in inv:
+        for station in network:
+            print(station)
+
+            # waveform = client.get_waveforms(network=network.code, station=station.code, location="*", channel="*", 
+            #                             starttime=obspy.UTCDateTime(start_date), endtime=obspy.UTCDateTime(end_date))
 
     return 0
 
@@ -152,6 +168,7 @@ with gr.Blocks() as demo:
                                     interactive=True)
             
         button = gr.Button("Predict phases")
+        button.click(mark_phases, inputs=inputs, outputs=outputs)
 
     with gr.Tab("Predict on your own waveform"):
         gr.Markdown("""
@@ -159,6 +176,6 @@ with gr.Blocks() as demo:
         Your waveform should be sampled at 100 sps and have 3 (Z, N, E) or 1 (Z) channels.
         """)
 
-    button.click(mark_phases, inputs=inputs, outputs=outputs)
+    button.click(download_data, inputs=[timestamp_inputs, eq_lat_inputs,eq_lo_inputs, radius_inputs], outputs=outputs)
 
 demo.launch()