Update app.py

app.py

@@ -2,46 +2,34 @@ import os
 import gradio as gr
 import pandas as pd
 
-# ======= Constants =======
 EXCEL_FILE = "database.xlsx"
-SHEET_DATA = "Data"
-SHEET_USERS = "Users"
 
-# ======= Global Data Store =======
-# Fungsi untuk load data dari Excel jika ada
+# ======= Helper functions for Excel storage =======
 def load_data():
-    global df, users_df
     if os.path.exists(EXCEL_FILE):
         try:
             xls = pd.ExcelFile(EXCEL_FILE)
-            if SHEET_DATA in xls.sheet_names:
-                df = pd.read_excel(xls, sheet_name=SHEET_DATA)
-            else:
-                df = pd.DataFrame(columns=["USERNAME", "COUNTRYCODE", "SBUCODE", "BRANDCODE", "CONCEPTCODE", "CHANNEL", "REMARK"])
-            if SHEET_USERS in xls.sheet_names:
-                users_df = pd.read_excel(xls, sheet_name=SHEET_USERS)
-            else:
-                users_df = pd.DataFrame([{"USERNAME": "ADMIN", "PASSWORD": "ADMIN"}])
+            df_data = pd.read_excel(xls, "DATA")
+            df_users = pd.read_excel(xls, "USERS")
+            return df_data, df_users
         except Exception as e:
-            print(f"Error loading Excel: {e}")
-            df = pd.DataFrame(columns=["USERNAME", "COUNTRYCODE", "SBUCODE", "BRANDCODE", "CONCEPTCODE", "CHANNEL", "REMARK"])
-            users_df = pd.DataFrame([{"USERNAME": "ADMIN", "PASSWORD": "ADMIN"}])
-    else:
-        df = pd.DataFrame(columns=["USERNAME", "COUNTRYCODE", "SBUCODE", "BRANDCODE", "CONCEPTCODE", "CHANNEL", "REMARK"])
-        users_df = pd.DataFrame([{"USERNAME": "ADMIN", "PASSWORD": "ADMIN"}])
-
-# Fungsi untuk simpan data ke Excel
-def save_data():
-    global df, users_df
-    with pd.ExcelWriter(EXCEL_FILE, engine='openpyxl', mode='w') as writer:
-        df.to_excel(writer, index=False, sheet_name=SHEET_DATA)
-        users_df.to_excel(writer, index=False, sheet_name=SHEET_USERS)
-
-# Load data di awal
-load_data()
+            print(f"Error loading Excel file: {e}")
+            # fallback empty
+    # If file does not exist or error, create empty
+    df_data = pd.DataFrame(columns=["USERNAME", "COUNTRYCODE", "SBUCODE", "BRANDCODE", "CONCEPTCODE", "CHANNEL", "REMARK"])
+    df_users = pd.DataFrame([{"USERNAME": "ADMIN", "PASSWORD": "ADMIN"}])
+    save_data(df_data, df_users)
+    return df_data, df_users
+
+def save_data(df_data, df_users):
+    with pd.ExcelWriter(EXCEL_FILE, engine="openpyxl") as writer:
+        df_data.to_excel(writer, sheet_name="DATA", index=False)
+        df_users.to_excel(writer, sheet_name="USERS", index=False)
 
-# ======= Helper Functions =======
+# ======= Global Data Store =======
+df, users_df = load_data()
 
+# ======= Your existing helper functions =======
 def to_upper(df_input):
     return df_input.applymap(lambda x: str(x).upper().strip() if pd.notnull(x) else x)
 
@@ -68,7 +56,7 @@ def bulk_submit(dataframe):
     merged = pd.merge(new_df, df[pk_cols], on=pk_cols, how="left", indicator=True)
     df_filtered = new_df[merged["_merge"] == "left_only"]
     df = pd.concat([df, df_filtered], ignore_index=True)
-    save_data()  # Simpan setelah update data
+    save_data(df, users_df)  # SAVE after submit
     return get_active_data()
 
 def delete_selected_rows(selected_df):
@@ -87,7 +75,7 @@ def delete_selected_rows(selected_df):
                 (df["REMARK"].str.upper() != "DELETE")
             )
             df.loc[mask, "REMARK"] = "DELETE"
-        save_data()  # Simpan setelah update data
+        save_data(df, users_df)  # SAVE after delete
         return "Berhasil dihapus.", get_active_data()
     except Exception as e:
         return f"Error: {e}", get_active_data()
@@ -101,7 +89,7 @@ def add_user(username, password):
     if username in users_df["USERNAME"].values:
         return "Username sudah ada", show_users()
     users_df = pd.concat([users_df, pd.DataFrame([{"USERNAME": username, "PASSWORD": password}])], ignore_index=True)
-    save_data()  # Simpan setelah update user
+    save_data(df, users_df)  # SAVE after add user
     return f"User {username} ditambahkan", show_users()
 
 def delete_user(username):
@@ -110,7 +98,7 @@ def delete_user(username):
     if username == "ADMIN":
         return "ADMIN tidak boleh dihapus", show_users()
     users_df = users_df[users_df["USERNAME"] != username].reset_index(drop=True)
-    save_data()  # Simpan setelah update user
+    save_data(df, users_df)  # SAVE after delete user
     return f"User {username} dihapus", show_users()
 
 def show_users():
@@ -118,40 +106,19 @@ def show_users():
 
 def show_guide():
     return """
-
-### Panduan Penggunaan Aplikasi
-
-
-1.  
-**Login**
-
-
-  - Gunakan username dan password untuk masuk.
-
-
-2.  
-**Input Data**
-
-
-  - Lihat data aktif.
-
-  - Input data baru dengan kolom lengkap.
-
-  - Pilih baris dan klik "Delete Selected Rows" untuk tandai delete.
-
-
-3.  
-**Manajemen User**
-
-
-  - Tambah dan hapus user (kecuali ADMIN).
-
-
-4.  
-**Data yang dihapus tidak benar-benar dihapus**, hanya ditandai `REMARK = DELETE`. """
-
-# ======= UI Layout =======
-
+    ### Panduan Penggunaan Aplikasi
+    1. **Login**
+       - Gunakan username dan password untuk masuk.
+    2. **Input Data**
+       - Lihat data aktif.
+       - Input data baru dengan kolom lengkap.
+       - Pilih baris dan klik "Delete Selected Rows" untuk tandai delete.
+    3. **Manajemen User**
+       - Tambah dan hapus user (kecuali ADMIN).
+    4. **Data yang dihapus tidak benar-benar dihapus**, hanya ditandai `REMARK = DELETE`.
+    """
+
+# ======= UI Layout and Events (unchanged) =======
 with gr.Blocks() as app:
     gr.Markdown("## Login")
     login_box = gr.Column(visible=True)