Update app.py

app.py

@@ -8,6 +8,7 @@ import leafmap
 from samgeo import SamGeo, show_image, download_file, overlay_images, tms_to_geotiff
 from zipfile import ZipFile
 
+
 #initialize the model
 sam = SamGeo(
     model_type="vit_h",
@@ -15,19 +16,6 @@ sam = SamGeo(
     sam_kwargs=None,
 )
 
-#mkdir for app
-storage='app_store'
-if os.path.exists(storage):
-    pass
-else:
-  os.mkdir(storage)
-
-zip_folder=f'{storage}/zip_folder'
-if os.path.exists(zip_folder):
-    pass
-else:
-  os.mkdir(zip_folder)
-
 def get_all_file_paths(directory):
   
     # initializing empty file paths list
@@ -45,15 +33,30 @@ def get_all_file_paths(directory):
 
 def get_shape_files(image):
   filename, file_extension = os.path.splitext(image.name)
-  if file_extension=='tif':
-    out_file=f'{storage}/{image.name}'
+  print(filename,file_extension)
+  if file_extension=='.tif':
+    #mkdir for app
+    storage='app_store'
+    if os.path.exists(storage):
+        pass
+    else:
+      os.mkdir(storage)
+
+    zip_folder=f'{storage}/zip_folder'
+    if os.path.exists(zip_folder):
+        pass
+    else:
+      os.mkdir(zip_folder)
+  
+    out_file=f'{storage}/out_files'
     if os.path.exists(out_file):
         pass
     else:
       os.mkdir(out_file)
 
-    sam.generate(image.name, output="masks.tif", foreground=True, unique=True)
-    sam.tiff_to_vector("masks.tif", out_file)
+    # load_model()
+    sam.generate(image.name, output=f"{out_file}/masks.tif", foreground=True, unique=True)
+    sam.tiff_to_vector(f"{out_file}/masks.tif", out_file)
 
     # writing files to a zipfile
     file_paths=get_all_file_paths(out_file)
@@ -61,14 +64,14 @@ def get_shape_files(image):
     # for file_name in file_paths:
     #       print(file_name)
 
-    with ZipFile(f'{zip_folder}/{image.name}.zip','w') as zip:
+    with ZipFile(f'{zip_folder}/shapefile.zip','w') as zip:
         # writing each file one by one
         for file in file_paths:
             zip.write(file)
     
     print('All files zipped successfully!') 
 
-    return f'{zip_folder}/{image.name}.zip'
+    return f'{zip_folder}/shapefile.zip'
   # get_shape_files(image)
   else:
     return "Try uploading .tif file for processing!."