Upload Preprocessing.py

Preprocessing.py

@@ -10,12 +10,15 @@ GITHUB_BASE_URL = "https://raw.githubusercontent.com/jainkavya738/Electricity-ML
 
 NUM_IMPUTER_URL = "https://github.com/jainkavya738/Electricity-ML-Artifacts/raw/refs/heads/main/numerical_imputer.pkl"
 CAT_IMPUTER_URL = "https://github.com/jainkavya738/Electricity-ML-Artifacts/raw/refs/heads/main/categorical_imputer.pkl"
-LE_STRUCTURE_TYPE_URL = "https://github.com/jainkavya738/Electricity-ML-Artifacts/raw/refs/heads/main/label_encoder_structure_type.pkl "
+LE_STRUCTURE_TYPE_URL = "https://github.com/jainkavya738/Electricity-ML-Artifacts/raw/refs/heads/main/label_encoder_structure_type.pkl"
 SCALER_URL = "https://github.com/jainkavya738/Electricity-ML-Artifacts/raw/refs/heads/main/scaler.pkl"
 MODEL_URL = "https://github.com/jainkavya738/Electricity-ML-Artifacts/raw/refs/heads/main/model.pkl" 
 
 def download_and_load_pkl(url, filename):
     print(f"Attempting to download {filename} from {url}")
+    
+    temp_filepath = None 
+    
     try:
         response = requests.get(url, stream=True)
         response.raise_for_status()
@@ -30,12 +33,22 @@ def download_and_load_pkl(url, filename):
         print(f"Successfully downloaded and loaded {filename}")
         os.remove(temp_filepath) 
         return obj
+    
     except requests.exceptions.RequestException as e:
         print(f"Error downloading {temp_filepath} from {url}: {e}")
         return None 
+    
     except Exception as e:
         print(f"Error loading {temp_filepath} after download: {e}")
         return None 
+    
+    finally: 
+        if temp_filepath and os.path.exists(temp_filepath):
+            try:
+                os.remove(temp_filepath)
+                print(f"Cleaned up temporary file: {temp_filepath}")
+            except OSError as e:
+                print(f"Warning: Could not remove temporary file {temp_filepath}: {e}")
 
 numerical_imputer = download_and_load_pkl(NUM_IMPUTER_URL, "numerical_imputer.pkl")
 categorical_imputer = download_and_load_pkl(CAT_IMPUTER_URL, "categorical_imputer.pkl")