Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import gradio as gr
 from huggingface_hub import HfApi
+from pydriosm.reader import GeofabrikReader
 
 reponame="Omnibus/osm"
 save_data=f'https://huggingface.co/datasets/{reponame}/raw/main/'
@@ -8,20 +9,21 @@ token_self = os.environ['HF_TOKEN']
 api=HfApi(token=token_self)
 
 
-def save_file_data(inp):
-    try:
-        file_obj=inp
-        api.upload_file(
-        path_or_fileobj=inp,
-        path_in_repo=inp,
-        repo_id=reponame,
-        #repo_id=save_data.split('datasets/',1)[1].split('/raw',1)[0],
-        token=token_self,
-        repo_type="dataset",
-        )
-        print("Done")
-    except Exception as e:
-        print(e)
+def save_file_data(inp, key):
+    if inp == os.environ.get("PASS"):
+        try:
+            file_obj=inp
+            api.upload_file(
+            path_or_fileobj=inp,
+            path_in_repo=inp,
+            repo_id=reponame,
+            #repo_id=save_data.split('datasets/',1)[1].split('/raw',1)[0],
+            token=token_self,
+            repo_type="dataset",
+            )
+            print("Done")
+        except Exception as e:
+            print(e)
 
 def get_map(inp):
     if inp == os.environ.get("PASS"):
@@ -31,11 +33,21 @@ def get_map(inp):
         print("done")
         #out = os.path.abspath("quebec-latest.osm.pbf")
         #print (out)
-        save_file_data("north-america-latest.osm.pbf")
+        save_file_data("north-america-latest.osm.pbf", inp)
         print("saved")
+
+def process_data():
+    gfr = GeofabrikReader()
+    subrgn_name=""
+    parsed_1 = gfr.read_osm_pbf(
+        subregion_name=subrgn_name, data_dir=dat_dir, expand=True, verbose=True)
+        
 with gr.Blocks() as app:
-    with gr.Row():
-        inp_pass=gr.Textbox()
-        btn=gr.Button()
+    with gr.Tab("Process"):
+        proc_btn=gr.Button("Process")
+    with gr.Tab("Load"):
+        with gr.Row():
+            inp_pass=gr.Textbox()
+            btn=gr.Button()
     btn.click(get_map,inp_pass,None)
 app.launch()