Update app.py

app.py

@@ -10,6 +10,7 @@ import numpy as np
 from PIL import Image
 import torch
 from huggingface_hub import hf_hub_download
+from huggingface_hub import HfApi
 
 
 
@@ -22,6 +23,32 @@ from Utils.area import pixel_to_sqft, process_and_overlay_image
 from Utils.convert import read_pansharpened_rgb
 
 
+hf_api = HfApi()
+HF_TOKEN = st.secrets.get("HF_TOKEN")
+if not HF_TOKEN:
+    st.error("HF_TOKEN not found.")
+    st.stop()
+
+REPO_ID = "Pavan2k4/Building_area"
+REPO_TYPE = "space"
+
+# save to dir func
+def save_to_hf_repo(local_path, repo_path):
+    try:
+        hf_api.upload_file(
+            path_or_fileobj=local_path,
+            path_in_repo=repo_path,
+            repo_id=REPO_ID,
+            repo_type=REPO_TYPE,
+            token=HF_TOKEN
+        )
+        st.success(f"File uploaded successfully to {repo_path}")
+    except Exception as e:
+        st.error(f"Error uploading file: {str(e)}")
+        st.error("Detailed error information:")
+        st.exception(e)
+
+
 BASE_DIR = os.getcwd()
 
 # Define subdirectories
@@ -174,11 +201,12 @@ def upload_page():
             
             with open(filepath, "wb") as f:
                 f.write(bytes_data)
+
             
-            st.success(f"Image saved to {filepath}")
+            #st.success(f"Image saved to {filepath}")
             
-            # Save image to Hugging Face repo
-           
+            # Save image to Hugging Face repo----------------------------------------------------------------------------------------------------------------------------------
+            save_to_hf_repo(filepath, f'uploaded_images/{filename}')
             
             # Check if the uploaded file is a GeoTIFF
             if file_extension in ['.tiff', '.tif']:
@@ -217,8 +245,8 @@ def upload_page():
             cv2.imwrite(mask_filepath, full_mask)
             st.session_state.mask_filename = mask_filename
 
-            # Save mask to Hugging Face repo
-            
+            # Save mask to Hugging Face repo---------------------------------------------------------------------------------------------
+            save_to_hf_repo(mask_filepath, f'generated_masks/{st.session_state.mask_filename}')
 
             # Log image details
             log_image_details(timestamp, converted_filename, mask_filename)