added main page and pages with uploading

Author: anna.yamkovaya

main.py

@@ -4,8 +4,8 @@
 
 
 PAGES = {
-    "Annotations Upload": upload.annotations,
-    "Requirements Upload": upload.requirements,
+    "Annotations Upload": upload.annotations.show,
+    "Requirements Upload": upload.requirements.show,
 }
 
 

pages/upload/__init__.py

@@ -0,0 +1,2 @@
+from pages.upload import annotations
+from pages.upload import requirements

pages/upload/annotations.py

@@ -1,27 +1,33 @@
+import io
+
 import streamlit as st
 import csv
 
-st.title("Upload *.trace.csv file")
 
-uploaded_file = st.file_uploader("Choose a *.trace.csv file", type="csv")
+def show():
+    st.title("Upload *.trace.csv file")
+
+    uploaded_files = st.file_uploader("Choose a *.trace.csv file", type="csv", accept_multiple_files=True)
 
-if uploaded_file is not None:
-    try:
-        # Read the file as a string
-        with open(uploaded_file, "r") as trace_file:
-            reader = csv.reader(trace_file)
+    for uploaded_file in uploaded_files:
+        try:
+            # Read the file as a string
+            stringio = io.StringIO(uploaded_file.getvalue().decode("utf-8"))
+            reader = csv.reader(stringio)
             rows = list(reader)
             # run indexing
             # run trace
 
-        if rows:
-            st.success("CSV file uploaded and parsed successfully!")
-            st.subheader("Extracted Rows:")
-            # Display as a table
-            st.table(rows)
-        else:
-            st.warning("The uploaded CSV file is empty.")
-    except Exception as e:
-        st.error(f"Failed to parse CSV file: {e}")
-else:
-    st.info("Please upload a *.trace.csv file to extract rows.")
+            if rows:
+                st.success("CSV file uploaded and parsed successfully!")
+                st.subheader("Extracted Rows:")
+                # Display as a table
+                st.table(rows)
+            else:
+                st.warning("The uploaded CSV file is empty.")
+        except Exception as e:
+            st.error(f"Failed to parse CSV file: {e}")
+
+    if not uploaded_files:
+        st.info("Please upload a *.trace.csv file to extract rows.")
+        return False

pages/upload/requirements.py

@@ -1,26 +1,32 @@
+import io
+
 import streamlit as st
 import csv
 
-st.title("Upload a file with requirements.")
 
-uploaded_file = st.file_uploader("Choose a file with requirements", type="csv")
+def show():
+    st.title("Upload a file with requirements.")
+
+    uploaded_files = st.file_uploader("Choose a file with requirements", type="csv")
 
-if uploaded_file is not None:
-    try:
-        # Read the file as a string
-        with open(uploaded_file, "r") as trace_file:
-            reader = csv.reader(trace_file)
+    for uploaded_file in uploaded_files:
+        try:
+
+            stringio = io.StringIO(uploaded_file.getvalue().decode("utf-8"))
+            reader = csv.reader(stringio)
             rows = list(reader)
             # run indexing
 
-        if rows:
-            st.success("CSV file uploaded and parsed successfully!")
-            st.subheader("Extracted Rows:")
-            # Display as a table
-            st.table(rows)
-        else:
-            st.warning("The uploaded CSV file is empty.")
-    except Exception as e:
-        st.error(f"Failed to parse CSV file: {e}")
-else:
-    st.info("Please upload a file with requirements to extract rows.")
+            if rows:
+                st.success("CSV file uploaded and parsed successfully!")
+                st.subheader("Extracted Rows:")
+                # Display as a table
+                st.table(rows)
+            else:
+                st.warning("The uploaded CSV file is empty.")
+        except Exception as e:
+            st.error(f"Failed to parse CSV file: {e}")
+
+    if not uploaded_files:
+        st.info("Please upload a *.trace.csv file to extract rows.")
+        return False