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):
    return username == "Josia" and password == "admin"

# Function to save data to CSV
def save_data(form_data, username):
    # 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 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():
    st.set_page_config(page_title="Frijoles", layout="wide")
    st.sidebar.title("Common Bean Data Collection App")
    st.sidebar.header("Please Log In")

    username = st.sidebar.text_input("Username")
    password = st.sidebar.text_input("Password", type="password")

    if st.sidebar.button("Login"):
        if check_credentials(username, password):
            st.session_state['authenticated'] = True
        else:
            st.session_state['authenticated'] = False
            st.sidebar.error("The username or password you entered is incorrect.")

    if 'authenticated' not in st.session_state:
        st.session_state['authenticated'] = False

    if st.session_state['authenticated']:
        st.title('Welcome to Common Bean Data Collection App')

        # Form for new data collection
        st.header("Collect new data")
        with st.form("new_data_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'])
            growth_level = st.selectbox("Select plant growth level", ['First Trifoliate', 'Second Trifoliate'])
            plant_number = st.selectbox("Select plant number", list(range(1, 11)))
            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")

            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!")

        # 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)

        # 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.")

# Run the main app
if __name__ == "__main__":
    main()