Update app.py

app.py

@@ -1,32 +1,21 @@
 import pandas as pd
 import streamlit as st
-import csv
-import io, os
 import matplotlib.pyplot as plt
 import numpy as np
 from second import double_main
-from pre import preprocess_csv, load_data, fill_missing_data
+from multiple import multiple_main
+from pre import preprocess_uploaded_file
 
 
 
 def single_main(uploaded_file):  
-    # st.title('Single CSV Analyzer')
+    # st.title('multi CSV Analyzer')
 
     # uploaded_file = st.file_uploader("Upload CSV file", type="csv")
 
     if uploaded_file is not None:
        # Process the csv files with header
-        filet = uploaded_file.read()
-        processed_output = preprocess_csv(filet)
-        processed_file = io.StringIO(processed_output.getvalue())
-        data = load_data(processed_file)
-
-        data = fill_missing_data(data, 4, 0)
-        data['Start datetime'] = pd.to_datetime(data['Start datetime'], errors='coerce')
-        data['End datetime'] = pd.to_datetime(data['End datetime'], errors='coerce')
-        data['Time spent'] = (data['End datetime'] - data['Start datetime']).dt.total_seconds()
-        data['Time spent(m:s)'] = pd.to_datetime(data['Time spent'], unit='s').dt.strftime('%M:%S')
-
+        data = preprocess_uploaded_file(uploaded_file)
         # st.write(data)
 
        # Display scenarios with status "failed" grouped by functional area
@@ -134,26 +123,24 @@ def single_main(uploaded_file):
 def main():
     st.title('Batch Run CSV Analyser')
 
-    # Initially we are in single file processing mode
+    # Initially we are in multi file processing mode
     if "mode" not in st.session_state:
-        st.session_state["mode"] = "single"
+        st.session_state["mode"] = "multi"
 
     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"
+    btn_label = "Switch to Compare mode" if st.session_state["mode"] == "multi" else "Switch to Multi mode"
     if st.sidebar.button(btn_label):
-        if st.session_state["mode"] == "single":
+        if st.session_state["mode"] == "multi":
             st.session_state["mode"] = "compare"
         else:
-            st.session_state["mode"] = "single"
+            st.session_state["mode"] = "multi"
 
     # 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)
+    if st.session_state["mode"] == "multi":
+        multiple_main()
     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")