Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import pickle
 import pandas as pd
 import numpy as np
+from io import BytesIO
+from reportlab.pdfgen import canvas
 
 # Page Title with Style
 # st.title("🩸 Sepsis Prediction App")
@@ -126,10 +128,54 @@ if st.button("🔮 Predict Sepsis"):
         input_data_scaled_df = preprocess_input_data(input_data)
         sepsis_status = make_predictions(input_data_scaled_df)
         st.success(f"The predicted sepsis status is: {sepsis_status}")
+
+        # Add the sepsis prediction to the input DataFrame
+        input_data['Sepsis'] = sepsis_status
+
         # Convert the input data to a pandas DataFrame
         input_df = pd.DataFrame([input_data])
+
+        # Display DataFrame
         st.table(input_df)
-        
+
+        # Download Button
+        download_btn = st.button("📥 Download Prediction")
+
+        if download_btn:
+            # Prompt user for patient's name
+            patient_name = st.text_input("Enter Patient's Name:")
+
+            # Create heading
+            heading = f"Sepsis Status Prediction For {patient_name}"
+
+            # Export to Excel
+            excel_file = BytesIO()
+            with pd.ExcelWriter(excel_file, engine='openpyxl') as writer:
+                input_df.to_excel(writer, sheet_name='Sheet1', index=False)
+                writer.sheets['Sheet1'].title = heading
+            excel_file.seek(0)
+            st.download_button(
+                label="Download Excel",
+                data=excel_file,
+                file_name=f"sepsis_prediction_{patient_name}.xlsx",
+                key='excel-download',
+            )
+
+            # Export to PDF
+            pdf_file = BytesIO()
+            pdf_canvas = canvas.Canvas(pdf_file)
+            pdf_canvas.drawString(100, 800, heading)
+            table = input_df.to_string().split('\n')
+            pdf_canvas.drawString(100, 780, '\n'.join(table))
+            pdf_canvas.save()
+            pdf_file.seek(0)
+            st.download_button(
+                label="Download PDF",
+                data=pdf_file,
+                file_name=f"sepsis_prediction_{patient_name}.pdf",
+                key='pdf-download',
+            )
+
     except Exception as e:
         st.error(f"An error occurred: {e}")