Update app.py

app.py

@@ -5,6 +5,7 @@ import pandas as pd
 from datetime import datetime
 import json
 import os
+import time
 
 # Konfigurera sidan
 st.set_page_config(
@@ -24,7 +25,8 @@ READONLY_COLUMNS = [
     'Tidstämpel',
     'Organisationsnummer', 
     'Lösenord',
-    'Account',  # Företagsnamn
+    'Områdeskod',
+    'Account',
     'Location Name',
     'Location - Address',
     'Long',
@@ -75,7 +77,6 @@ def get_sheet_data(sheet):
 
 def validate_password(df, password):
     """Validera lösenord och returnera organisationsnummer"""
-    # Hitta rad där lösenord matchar
     match = df[df['Lösenord'].astype(str) == password]
     
     if not match.empty:
@@ -85,25 +86,44 @@ def validate_password(df, password):
 
 def get_company_areas(df, org_nr):
     """Hämta alla områden för ett organisationsnummer"""
-    return df[df['Organisationsnummer'] == org_nr]
+    return df[df['Organisationsnummer'] == org_nr].copy()
 
 def update_cell(sheet, row_num, col_name, new_value, df):
     """Uppdatera en specifik cell i Google Sheets"""
     try:
-        # Hitta kolumnindex (1-baserad för gspread)
         col_idx = df.columns.get_loc(col_name) + 1
-        # row_num är 0-baserad, lägg till 2 för header och 1-basering
         cell_row = row_num + 2
-        
         sheet.update_cell(cell_row, col_idx, new_value)
         return True
     except Exception as e:
         st.error(f"❌ Kunde inte uppdatera {col_name}: {e}")
         return False
 
-def get_editable_columns(df):
-    """Returnera lista med editerbara kolumner"""
-    return [col for col in df.columns if col not in READONLY_COLUMNS]
+def get_column_config(df):
+    """Skapa kolumnkonfiguration för data editor"""
+    config = {}
+    for col in df.columns:
+        if col in READONLY_COLUMNS:
+            # Skrivskyddade kolumner
+            if col == 'Lösenord':
+                config[col] = st.column_config.TextColumn(
+                    col,
+                    disabled=True,
+                    help="Skrivskyddat fält"
+                )
+            else:
+                config[col] = st.column_config.TextColumn(
+                    col,
+                    disabled=True,
+                    help="Skrivskyddat fält"
+                )
+        else:
+            # Editerbara kolumner
+            config[col] = st.column_config.TextColumn(
+                col,
+                help="Klicka för att editera"
+            )
+    return config
 
 # CSS för bättre styling
 st.markdown("""
@@ -111,18 +131,16 @@ st.markdown("""
     .stButton button {
         width: 100%;
     }
-    .area-card {
-        padding: 20px;
-        border-radius: 10px;
-        background-color: #f0f2f6;
-        margin-bottom: 20px;
+    div[data-testid="stDataFrame"] {
+        width: 100%;
     }
-    .success-msg {
+    .success-banner {
         padding: 10px;
         border-radius: 5px;
         background-color: #d4edda;
         color: #155724;
         margin: 10px 0;
+        text-align: center;
     }
 </style>
 """, unsafe_allow_html=True)
@@ -136,6 +154,7 @@ if 'authenticated' not in st.session_state:
     st.session_state.org_nr = None
     st.session_state.df = None
     st.session_state.company_name = None
+    st.session_state.original_data = None
 
 # Anslut till Google Sheets
 sheet = get_google_sheet()
@@ -173,6 +192,7 @@ if not st.session_state.authenticated:
                             st.session_state.org_nr = org_nr
                             st.session_state.df = df
                             st.session_state.company_name = company_name
+                            st.session_state.original_data = company_areas.copy()
                             st.success("✅ Inloggning lyckades!")
                             st.rerun()
                         else:
@@ -190,7 +210,7 @@ else:
     company_name = st.session_state.company_name
     
     # Header med företagsinfo
-    col1, col2, col3 = st.columns([2, 1, 1])
+    col1, col2, col3 = st.columns([3, 1, 1])
     
     with col1:
         st.markdown(f"## 🏢 {company_name}")
@@ -199,9 +219,13 @@ else:
     with col2:
         if st.button("🔄 Uppdatera data", use_container_width=True):
             st.cache_resource.clear()
-            st.session_state.df = get_sheet_data(sheet)
-            st.success("✅ Data uppdaterad!")
-            st.rerun()
+            new_df = get_sheet_data(sheet)
+            if new_df is not None:
+                st.session_state.df = new_df
+                company_areas = get_company_areas(new_df, org_nr)
+                st.session_state.original_data = company_areas.copy()
+                st.success("✅ Data uppdaterad!")
+                st.rerun()
     
     with col3:
         if st.button("🚪 Logga ut", use_container_width=True):
@@ -209,6 +233,7 @@ else:
             st.session_state.org_nr = None
             st.session_state.df = None
             st.session_state.company_name = None
+            st.session_state.original_data = None
             st.rerun()
     
     st.divider()
@@ -216,96 +241,96 @@ else:
     # Hämta alla områden för detta företag
     company_areas = get_company_areas(df, org_nr)
     
-    st.markdown(f"### 📍 Dina områden ({len(company_areas)} st)")
+    st.markdown(f"### 📊 Dina områden ({len(company_areas)} st)")
+    st.caption("💡 Klicka i valfri cell för att editera. Ändringar sparas automatiskt.")
     
-    # Editerbara kolumner
-    editable_cols = get_editable_columns(df)
+    # Visa information om vilka fält som är editerbara
+    with st.expander("ℹ️ Information om fält", expanded=False):
+        col_info1, col_info2 = st.columns(2)
+        with col_info1:
+            st.markdown("**🔒 Skrivskyddade fält:**")
+            for col in READONLY_COLUMNS:
+                if col in company_areas.columns:
+                    st.markdown(f"- {col}")
+        with col_info2:
+            st.markdown("**✏️ Editerbara fält:**")
+            editable = [col for col in company_areas.columns if col not in READONLY_COLUMNS]
+            for col in editable:
+                st.markdown(f"- {col}")
     
-    # Visa varje område i en expanderbar sektion
-    for idx, (row_idx, area) in enumerate(company_areas.iterrows(), 1):
-        with st.expander(f"🗺️ Område {idx}: {area['Områdeskod']} - {area['Location Name']}", expanded=(idx == 1)):
-            
-            # Visa fast information
-            st.markdown("#### 📋 Fast information")
-            info_cols = st.columns(3)
-            
-            with info_cols[0]:
-                st.text_input("Områdeskod", value=area['Områdeskod'], disabled=True, key=f"area_{row_idx}_kod")
-                st.text_input("Location Name", value=area['Location Name'], disabled=True, key=f"area_{row_idx}_locname")
-            
-            with info_cols[1]:
-                st.text_input("Location Address", value=area['Location - Address'], disabled=True, key=f"area_{row_idx}_addr")
-                st.text_input("Location City", value=area['Location - City'], disabled=True, key=f"area_{row_idx}_city")
-            
-            with info_cols[2]:
-                st.text_input("Long", value=area['Long'], disabled=True, key=f"area_{row_idx}_long")
-                st.text_input("Lat", value=area['Lat'], disabled=True, key=f"area_{row_idx}_lat")
-            
-            st.divider()
-            
-            # Editerbara fält
-            st.markdown("#### ✏️ Editerbar information")
-            
-            # Gruppera editerbara fält
-            edit_col1, edit_col2 = st.columns(2)
-            
-            changes = {}  # Spara ändringar
-            
-            with edit_col1:
-                # Första halvan av editerbara kolumner
-                mid = len(editable_cols) // 2
-                for col in editable_cols[:mid]:
-                    current_value = str(area[col]) if pd.notna(area[col]) else ""
-                    
-                    new_value = st.text_input(
-                        col,
-                        value=current_value,
-                        key=f"edit_{row_idx}_{col}"
-                    )
-                    
-                    if new_value != current_value:
-                        changes[col] = new_value
+    st.divider()
+    
+    # Skapa kolumnkonfiguration
+    column_config = get_column_config(company_areas)
+    
+    # Visa editbar tabell
+    edited_data = st.data_editor(
+        company_areas,
+        column_config=column_config,
+        use_container_width=True,
+        num_rows="fixed",
+        hide_index=True,
+        key="data_editor"
+    )
+    
+    # Detektera ändringar och spara automatiskt
+    if st.session_state.original_data is not None:
+        # Jämför edited_data med original_data
+        changes_detected = False
+        changes_list = []
+        
+        for idx, (orig_row, edit_row) in enumerate(zip(
+            st.session_state.original_data.itertuples(index=True), 
+            edited_data.itertuples(index=True)
+        )):
+            orig_idx = orig_row.Index
             
-            with edit_col2:
-                # Andra halvan av editerbara kolumner
-                for col in editable_cols[mid:]:
-                    current_value = str(area[col]) if pd.notna(area[col]) else ""
-                    
-                    new_value = st.text_input(
-                        col,
-                        value=current_value,
-                        key=f"edit_{row_idx}_{col}"
-                    )
+            for col in company_areas.columns:
+                if col not in READONLY_COLUMNS:
+                    orig_val = str(getattr(orig_row, col.replace(' ', '_').replace('-', '_'), ''))
+                    edit_val = str(getattr(edit_row, col.replace(' ', '_').replace('-', '_'), ''))
                     
-                    if new_value != current_value:
-                        changes[col] = new_value
+                    if orig_val != edit_val:
+                        changes_detected = True
+                        changes_list.append({
+                            'row_idx': orig_idx,
+                            'column': col,
+                            'old_value': orig_val,
+                            'new_value': edit_val
+                        })
+        
+        # Om ändringar upptäckts, spara automatiskt
+        if changes_detected:
+            st.divider()
             
-            # Om det finns ändringar, visa uppdateringsknapp
-            if changes:
-                st.warning(f"⚠️ {len(changes)} fält har ändrats")
+            with st.container():
+                st.markdown(f"### 💾 Sparar {len(changes_list)} ändringar...")
                 
-                cols = st.columns([1, 1, 2])
-                with cols[0]:
-                    if st.button(f"💾 Spara alla ändringar", key=f"save_all_{row_idx}", type="primary", use_container_width=True):
-                        success_count = 0
-                        
-                        with st.spinner("Sparar ändringar..."):
-                            for col, new_value in changes.items():
-                                if update_cell(sheet, row_idx, col, new_value, df):
-                                    success_count += 1
-                        
-                        if success_count == len(changes):
-                            st.success(f"✅ Alla {success_count} ändringar sparade!")
-                            st.cache_resource.clear()
-                            st.session_state.df = get_sheet_data(sheet)
-                            st.rerun()
-                        else:
-                            st.error(f"⚠️ Kunde bara spara {success_count} av {len(changes)} ändringar")
+                progress_bar = st.progress(0)
+                success_count = 0
                 
-                with cols[1]:
-                    if st.button(f"↩️ Ångra ändringar", key=f"reset_{row_idx}", use_container_width=True):
-                        st.rerun()
+                for i, change in enumerate(changes_list):
+                    if update_cell(sheet, change['row_idx'], change['column'], change['new_value'], df):
+                        success_count += 1
+                    progress_bar.progress((i + 1) / len(changes_list))
+                    time.sleep(0.1)  # Liten fördröjning för att inte överbelasta API
+                
+                if success_count == len(changes_list):
+                    st.markdown('<div class="success-banner">✅ Alla ändringar sparade!</div>', unsafe_allow_html=True)
+                    time.sleep(1)
+                    
+                    # Uppdatera original data
+                    st.cache_resource.clear()
+                    new_df = get_sheet_data(sheet)
+                    if new_df is not None:
+                        st.session_state.df = new_df
+                        company_areas = get_company_areas(new_df, org_nr)
+                        st.session_state.original_data = company_areas.copy()
+                    st.rerun()
+                else:
+                    st.error(f"⚠️ Kunde bara spara {success_count} av {len(changes_list)} ändringar")
 
     # Footer
     st.divider()
-    st.caption(f"🔒 Säker åtkomst • Inloggad som {company_name} • {len(company_areas)} områden")
+    st.caption(f"🔒 Säker åtkomst • Inloggad som {company_name} • {len(company_areas)} områden")
+    st.caption("💡 Tips: Använd Ctrl+F (Cmd+F på Mac) för att söka i tabellen")