Update app.py

app.py

@@ -4,35 +4,10 @@ import csv
 import io
 import matplotlib.pyplot as plt
 import numpy as np
+from second import double_main
+from pre import preprocess_csv, load_data, fill_missing_data
 
-def preprocess_csv(input_bytes):
-    text = input_bytes.decode()  # Decode bytes to text
-    output = io.StringIO()
-    writer = csv.writer(output)
 
-    for row in csv.reader(io.StringIO(text)):  # Read text as csv
-        if len(row) > 5:
-            row = row[0:5] + [','.join(row[5:])]  # Combine extra fields into one
-        writer.writerow(row)
-
-    output.seek(0)  # go to the start of the StringIO object
-    return output
-
-def load_data(file):
-    column_names = [
-        'Functional area',
-        'Scenario name',
-        'Start datetime',
-        'End datetime',
-        'Status',
-        'Error message'
-    ]
-    data = pd.read_csv(file, header=None, names=column_names)
-    return data
-
-def fill_missing_data(data, column_index, value):
-    data.iloc[:, column_index] = data.iloc[:, column_index].fillna(value)
-    return data
 
 
 def single_main(uploaded_file):  
@@ -41,9 +16,7 @@ def single_main(uploaded_file):
     # uploaded_file = st.file_uploader("Upload CSV file", type="csv")
 
     if uploaded_file is not None:
-        # Process the csv file
-        column_names = ["Functional area", "Scenario name", "Start datetime", "End datetime", "Status", "Error message"]
-
+       # Process the csv files with header
         filet = uploaded_file.read()
         processed_output = preprocess_csv(filet)
         processed_file = io.StringIO(processed_output.getvalue())
@@ -135,12 +108,33 @@ def single_main(uploaded_file):
     pass
 
 def main():
-    st.title('CSV Analyser')
-    
-    uploaded_file = st.file_uploader("Upload CSV file", type="csv")
+    st.title('Batch Run CSV Analyser')
+
+    # Initially we are in single file processing mode
+    if "mode" not in st.session_state:
+        st.session_state["mode"] = "single"
+
+    mode_display = f'## Current mode: {st.session_state["mode"].title()} mode'
+    st.sidebar.markdown(mode_display)
+
+    # Add a button to switch between modes
+    btn_label = "Switch to Compare mode" if st.session_state["mode"] == "single" else "Switch to Single mode"
+    if st.sidebar.button(btn_label):
+        if st.session_state["mode"] == "single":
+            st.session_state["mode"] = "compare"
+        else:
+            st.session_state["mode"] = "single"
+
+    # Only show the second file uploader in compare mode
+    if st.session_state["mode"] == "single":
+        uploaded_file_1 = st.sidebar.file_uploader("Upload CSV file", type="csv")
+        if uploaded_file_1 is not None:
+            single_main(uploaded_file_1)
+    else:
+        uploaded_file_1 = st.sidebar.file_uploader("Upload CSV file 1", type="csv")
+        uploaded_file_2 = st.sidebar.file_uploader("Upload CSV file 2", type="csv")
+        if uploaded_file_1 is not None and uploaded_file_2 is not None:
+            double_main(uploaded_file_1, uploaded_file_2)
 
-    if uploaded_file is not None:
-        single_main(uploaded_file)  # Load the main app when the file is uploaded
-        
 if __name__ == "__main__":
     main()