Added description

app.py

@@ -5,7 +5,8 @@ import numpy as np
 from phasehunter.model import Onset_picker, Updated_onset_picker
 from phasehunter.data_preparation import prepare_waveform
 import torch
-from scipy.stats.kde import gaussian_kde
+
+from scipy.stats import gaussian_kde
 
 import obspy
 from obspy.clients.fdsn import Client
@@ -77,6 +78,8 @@ def mark_phases(waveform):
     plt.close(fig)
     return image
 
+#??
+
 def download_data(timestamp, eq_lat, eq_lon, client_name, radius_km):
     client = Client(client_name)
     window = radius_km / 111.2
@@ -84,8 +87,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 = catalog['DateTime'].apply(lambda x: pd.Timestamp(x)).min()
-    # endtime = catalog['DateTime'].apply(lambda x: pd.Timestamp(x)).max()
+    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
 
@@ -159,6 +175,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("""
@@ -166,6 +183,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()