Update app.py

app.py

@@ -34,15 +34,12 @@ if not HF_TOKEN:
 REPO_ID = "Pavan2k4/Building_area"
 REPO_TYPE = "space"
 
-# Define base directory for Hugging Face Spaces
-BASE_DIR = "DATA/"
-
 # Define subdirectories
-UPLOAD_DIR = os.path.join(BASE_DIR, "uploaded_images")
-MASK_DIR = os.path.join(BASE_DIR, "generated_masks")
-PATCHES_DIR = os.path.join(BASE_DIR, "patches")
-PRED_PATCHES_DIR = os.path.join(BASE_DIR, "pred_patches")
-CSV_LOG_PATH = os.path.join(BASE_DIR, "image_log.csv")
+UPLOAD_DIR = "uploaded_images"
+MASK_DIR = "generated_masks"
+PATCHES_DIR = "patches"
+PRED_PATCHES_DIR = "pred_patches"
+CSV_LOG_PATH = "image_log.csv"
 
 
 def split(image, destination = PATCHES_DIR, patch_size = 650):
@@ -117,7 +114,6 @@ def save_to_hf_repo(local_path, repo_path):
             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:")
@@ -181,7 +177,7 @@ def upload_page():
             with open(filepath, "wb") as f:
                 f.write(bytes_data)
             
-            st.success(f"Image saved to {filepath}")
+            st.success('File uploaded and saved')
             
             # Save image to Hugging Face repo
             save_to_hf_repo(filepath, f'uploaded_images/{filename}')
@@ -191,13 +187,11 @@ def upload_page():
                 st.info('Processing GeoTIFF image...')
                 rgb_image = read_pansharpened_rgb(filepath)
                 cv2.imwrite(converted_filepath, cv2.cvtColor(rgb_image, cv2.COLOR_RGB2BGR))
-                st.success(f'GeoTIFF converted to 8-bit image and saved as {converted_filename}')
                 img = Image.open(converted_filepath)
             else:
                 img = Image.open(filepath)
             
             st.image(img, caption='Uploaded Image', use_column_width=True)
-            st.success(f'Image processed and saved as {converted_filename}')
 
             # Store the full path of the converted image
             st.session_state.filename = converted_filename
@@ -238,7 +232,6 @@ def upload_page():
                     shutil.rmtree(PRED_PATCHES_DIR)
                     os.makedirs(PATCHES_DIR)  # Recreate empty folders
                     os.makedirs(PRED_PATCHES_DIR)
-                    st.success('Temporary files cleaned up')
             else:
                 # Predict on whole image
                 st.session_state.tr_img = transforms(img)
@@ -249,6 +242,8 @@ def upload_page():
                 Image.fromarray(mask).save(mask_filepath)
                 st.session_state.mask_filename = mask_filename
 
+            st.success('Mask generated and saved')
+            
             # Save mask to Hugging Face repo
             mask_filepath = os.path.join(MASK_DIR, st.session_state.mask_filename)
             save_to_hf_repo(mask_filepath, f'generated_masks/{st.session_state.mask_filename}')