Update appStore/target.py

appStore/target.py

@@ -18,6 +18,39 @@ import xlsxwriter
 import plotly.express as px
 from utils.target_classifier import label_dict
 
+# Declare all the necessary variables
+classifier_identifier = 'vulnerability'
+params  = get_classifier_params(classifier_identifier)
+
+@st.cache_data
+def to_excel(df,sectorlist):
+    len_df = len(df)
+    output = BytesIO()
+    writer = pd.ExcelWriter(output, engine='xlsxwriter')
+    df.to_excel(writer, index=False, sheet_name='Sheet1')
+    workbook = writer.book
+    worksheet = writer.sheets['Sheet1']
+    worksheet.data_validation('S2:S{}'.format(len_df), 
+                              {'validate': 'list', 
+                               'source': ['No', 'Yes', 'Discard']})
+    worksheet.data_validation('X2:X{}'.format(len_df), 
+                              {'validate': 'list', 
+                               'source': sectorlist + ['Blank']})
+    worksheet.data_validation('T2:T{}'.format(len_df), 
+                              {'validate': 'list', 
+                               'source': sectorlist + ['Blank']})
+    worksheet.data_validation('U2:U{}'.format(len_df), 
+                              {'validate': 'list', 
+                               'source': sectorlist + ['Blank']})                               
+    worksheet.data_validation('V2:V{}'.format(len_df), 
+                              {'validate': 'list', 
+                               'source': sectorlist + ['Blank']})
+    worksheet.data_validation('W2:U{}'.format(len_df), 
+                              {'validate': 'list', 
+                               'source': sectorlist + ['Blank']})                            
+    writer.save()
+    processed_data = output.getvalue()
+    return processed_data
 
 def app():
     ### Main app code ###