From 9eafbe3b9f069ae7fbeded2d4b0316bdba428703 Mon Sep 17 00:00:00 2001
From: Kirandkv <117249197+Kirandkv@users.noreply.github.com>
Date: Tue, 19 Nov 2024 13:16:29 -0700
Subject: [PATCH] Update step2_dlq.py

---
 code/step2_dlq.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/code/step2_dlq.py b/code/step2_dlq.py
index 6fe5cf2..4652acc 100644
--- a/code/step2_dlq.py
+++ b/code/step2_dlq.py
@@ -1,3 +1,8 @@
+# Description: 
+# Runs the Gaussian puff atmospheric dispersion model to simulate methane
+# concentrations at CMS sensor locations given wind data and potential source locations. 
+# Author: Kiran Damodaran (kiran.damodaran@mines.edu)
+# Last Updated: Nov 2024
 import os
 import pandas as pd
 import numpy as np
@@ -6,6 +11,7 @@
 import pickle
 import logging
 import traceback
+import zipfile
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -392,6 +398,18 @@ def safe_float(series):
     plt.close()
 
 def main():
+  # Unzip input data files
+input_zip_path = "./code/Input_data_step2.zip"  # Adjust this path to your zip file
+extract_path = "./code/input_data_step2"
+
+# Create extraction directory if it doesn't exist
+os.makedirs(extract_path, exist_ok=True)
+
+# Unzip the input files
+with zipfile.ZipFile(input_zip_path, 'r') as zip_ref:
+    zip_ref.extractall(extract_path)
+print(f"Input files extracted to: {extract_path}")
+
     # Configuration
     config = {
         'gap_time': 5,