Update app.py

app.py

@@ -64,8 +64,6 @@ cmap_select_2d = pn.widgets.Select(name='2D Colormap', options=["rainbow", "viri
 
 # ---------------- Core Functions ----------------
 def process_zip(event=None):
-    import tempfile
-
     if file_input.value:
         zip_path = os.path.join(MEDIA_DIR, file_input.filename)
         with open(zip_path, "wb") as f:
@@ -73,20 +71,34 @@ def process_zip(event=None):
         status.object = "✅ ZIP uploaded and saved."
     else:
         zip_path = os.path.join(MEDIA_DIR, "default_model.zip")
+        if not os.path.exists(zip_path):
+            zip_path = "default_model.zip"  # fallback to local directory
         if not os.path.exists(zip_path):
             status.object = "❌ No ZIP uploaded and default_model.zip not found."
             return
         status.object = "📦 Using default_model.zip"
 
-    # Use safe temporary output directory
-    output_dir = os.path.join(tempfile.gettempdir(), "ash_output")
+
+    try:
+        output_dir = os.path.join("./", "ash_output")
+        os.makedirs(output_dir, exist_ok=True)
+    except PermissionError:
+        output_dir = os.path.join(tempfile.gettempdir(), "name_output")
+        os.makedirs(output_dir, exist_ok=True)
     shutil.rmtree(output_dir, ignore_errors=True)
     os.makedirs(output_dir, exist_ok=True)
 
     try:
         processor = NAMEDataProcessor(output_root=output_dir)
-        processor.batch_process_zip(zip_path)  # This must also use a writable temp dir internally
+        processor.batch_process_zip(zip_path)
 
+        # animator_obj["3d"] = [xr.open_dataset(fp).load()
+        #                       for fp in sorted(glob.glob(os.path.join(output_dir, "3D", "*.nc")))]
+        
+        # animator_obj["3d"] = []
+        # for fp in sorted(glob.glob(os.path.join(output_dir, "3D", "*.nc"))):
+        #     with xr.open_dataset(fp) as ds:
+        #         animator_obj["3d"].append(ds.load())
         animator_obj["3d"] = []
         for fp in sorted(glob.glob(os.path.join(output_dir, "3D", "*.nc"))):
             with xr.open_dataset(fp) as ds:
@@ -97,12 +109,15 @@ def process_zip(event=None):
             with xr.open_dataset(fp) as ds:
                 animator_obj["2d"].append(ds.load())
 
+        
+        # animator_obj["2d"] = [xr.open_dataset(fp).load()
+        #                       for fp in sorted(glob.glob(os.path.join(output_dir, "horizontal", "*.nc")))]
+
         with open(os.path.join(MEDIA_DIR, "last_run.txt"), "w") as f:
             f.write(zip_path)
 
         status.object += f" | ✅ Loaded 3D: {len(animator_obj['3d'])} & 2D: {len(animator_obj['2d'])}"
         update_media_tabs()
-
     except Exception as e:
         logging.exception("Error during ZIP processing")
         status.object = f"❌ Processing failed: {e}"