Update app.py

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 import pandas as pd
 from datetime import datetime
+import glob
+import os
 
 # Function to check user credentials
 def check_credentials(username, password):
@@ -8,17 +10,19 @@ def check_credentials(username, password):
 
 # Function to save data to CSV
 def save_data(form_data, username):
-    # Convert to DataFrame and append to CSV
+    # Convert to DataFrame and append to CSV. Here we assume that all CSV files are stored in the current directory.
     df = pd.DataFrame([form_data])
     csv_path = f'collected_data_{username}.csv'
-    df.to_csv(csv_path, mode='a', header=not pd.io.common.file_exists(csv_path), index=False)
+    df.to_csv(csv_path, mode='a', header=not os.path.exists(csv_path), index=False)
+
+# Function to list all CSV files created by the users
+def list_csv_files():
+    # List all CSV files in the current directory
+    return glob.glob('collected_data_*.csv')
 
 # Main App
 def main():
-    # Page configuration
     st.set_page_config(page_title="Frijoles", layout="wide")
-
-    # Sidebar for login
     st.sidebar.title("Common Bean Data Collection App")
     st.sidebar.header("Please Log In")
 
@@ -35,24 +39,13 @@ def main():
     if 'authenticated' not in st.session_state:
         st.session_state['authenticated'] = False
 
-    # Main Page
     if st.session_state['authenticated']:
         st.title('Welcome to Common Bean Data Collection App')
-        
-        # Display info links
-        col1, col2 = st.columns(2)
-        with col1:
-            if st.button('Learn More About Beans'):
-                # Display info about beans or include a link to more information
-                st.write("Information about beans will be displayed here or linked.")
-        with col2:
-            if st.button('Learn More About Breeding'):
-                # Display info about breeding or include a link to more information
-                st.write("Information about breeding will be displayed here or linked.")
 
+        # Form for new data collection
         st.header("Collect new data")
         with st.form("new_data_form"):
-            # Fields for the collection form
+            # Collect necessary information from user
             disease_type = st.selectbox("Select disease type", ['ALS', 'CBB'])
             region = st.selectbox("Select region", ['Selian', 'Uyole', 'Maluku'])
             genotype = st.selectbox("Select genotype", ['Genotype A', 'Genotype B'])
@@ -61,37 +54,42 @@ def main():
             leaf_number = st.selectbox("Select leaf number", list(range(1, 11)))
             scan_number = st.selectbox("Select scan number", list(range(1, 11)))
             visual_score = st.slider("Visual Score", 1, 9, 1)
+            
+            submitted = st.form_submit_button("Save")
 
-            # Submit button for form
-            submit_data = st.form_submit_button(label='Submit Data')
+            if submitted:
+                # Construct the form data to save
+                form_data = {
+                    "Date": datetime.now().strftime("%Y-%m-%d"),
+                    "Time": datetime.now().strftime("%H:%M:%S"),
+                    "Disease Type": disease_type,
+                    "Region": region,
+                    "Genotype": genotype,
+                    "Growth Level": growth_level,
+                    "Plant Number": plant_number,
+                    "Leaf Number": leaf_number,
+                    "Scan Number": scan_number,
+                    "Visual Score": visual_score
+                }
+                save_data(form_data, username)
+                st.success("Data saved successfully!")
 
-        if submit_data:
-            form_data = {
-                "Disease Type": disease_type,
-                "Region": region,
-                "Genotype": genotype,
-                "Growth Level": growth_level,
-                "Plant Number": plant_number,
-                "Leaf Number": leaf_number,
-                "Scan Number": scan_number,
-                "Visual Score": visual_score,
-                "Username": username,  # Include username in the saved data for identification
-                "Date": datetime.now().strftime("%Y-%m-%d"),
-                "Time": datetime.now().strftime("%H:%M:%S"),
-            }
+        # Button to open ScanSpectrum app (Placeholder for actual URL scheme)
+        st.markdown("Please use your mobile device to open the ScanSpectrum app and perform the scan.")
+        st.markdown("<a href='scanspectrum://'><button style='height: 50px; width: 300px; font-size: 20px;'>Open ScanSpectrum App</button></a>", unsafe_allow_html=True)
 
-            # Button to open ScanSpectrum app
-            st.markdown("<a href='scanspectrum://open' target='_blank'><button style='height: 50px; width: 300px; font-size: 20px;'>Open ScanSpectrum App</button></a>", unsafe_allow_html=True)
-            st.write("Please complete the scan and upload the results. After scanning, return here and click 'Save Data'.")
-            
-            # Placeholder for storing scan data
-            scanned_data_placeholder = {'wavelength': [680, 500, 450], 'intensity': [0.8, 0.6, 0.4]}
-            scanned_data = pd.DataFrame(scanned_data_placeholder)  # This would actually be fetched from the ScanSpectrum app
-            
-            # Save button to finalize data entry
-            if st.button('Save Data'):
-                save_data(form_data, username)
-                st.success("Data saved successfully.")
+        # Functionality to preview CSV files
+        st.header("Preview Saved Data")
+        all_csv_files = list_csv_files()
+        selected_csv = st.selectbox("Select a CSV file to preview", all_csv_files)
+        
+        if st.button('Show Selected CSV'):
+            if selected_csv:
+                data_to_show = pd.read_csv(selected_csv)
+                st.write(f"Preview of {selected_csv}:")
+                st.dataframe(data_to_show)
+            else:
+                st.error("No CSV file selected.")
 
     else:
         st.warning("You need to login to access the data collection form.")