change to make the perforation more flexiable

app.py

@@ -4,7 +4,7 @@ from datetime import datetime
 from streamlit_ext import download_button as down_btn
 
 
-def generate_perforations(df):
+def generate_perforations(df,plug_df):
     content = "UNITS METRIC\n\n"
 
     # Iterate through unique well names
@@ -16,12 +16,37 @@ def generate_perforations(df):
 
         # Iterate through rows of the filtered dataframe
         for index, row in well_df.iterrows():
-            date = row['date'].strftime('%d/%m/%Y')
-            top = row['top']
-            btm = row['btm']
-
-            # Write the perforation line to the content
-            content += f'{date} perforation {top} {btm} 0.1905 0 0 0\n'
+            if row['type'] == 'p':
+                date = row['date'].strftime('%d/%m/%Y')
+                top = row['top']
+                btm = row['btm']
+                # Write the perforation line to the content
+                content += f'{date} perforation {top} {btm} 0.1905 0 0 0\n'
+
+            if row['type'] == 's':
+                date = row['date'].strftime('%d/%m/%Y')
+                top = row['top']
+                btm = row['btm']
+                # Write the shut-in line to the content
+                content += f'{date} squeeze {top} {btm}\n'
+
+        # Filter the plug dataframe for the current well
+        plug_well_df = plug_df[plug_df['well'] == well_name]
+
+        # Iterate through rows of the filtered plug dataframe
+        for index, plug_row in plug_well_df.iterrows():
+            date = plug_row['date'].strftime('%d/%m/%Y')
+            depth = plug_row['depth']
+
+            # Filter the well dataframe based on the plug depth
+            filtered_well_df = well_df[well_df['top'] >= depth]
+
+            # Iterate through rows of the filtered well dataframe
+            for _, filtered_row in filtered_well_df.iterrows():
+                top = filtered_row['top']
+                btm = filtered_row['btm']
+                # Write the squeeze line to the content
+                content += f'{date} squeeze {top} {btm}\n'
 
         content += "\n"  # Add a newline between well sections
 
@@ -79,7 +104,8 @@ def main():
 
     if uploaded_file is not None:
         # Read the Excel file
-        df = pd.read_excel(uploaded_file)
+        df = pd.read_excel(uploaded_file,sheet_name='perforation')
+        plug_df = pd.read_excel(uploaded_file,sheet_name='plugs')
 
         # Create button to begin the process
     if st.sidebar.button("Generate"):
@@ -89,7 +115,7 @@ def main():
     # Use a callback to execute actions only when the button is clicked
     if st.session_state.get("generate_clicked", False):
         # Generate content for perforations.ev and tubing.tub
-        perforations_content = generate_perforations(df)
+        perforations_content = generate_perforations(df,plug_df)
         tubing_content = generate_tubing(df, tubing=tubing)
 
         # Display perforations.ev content