Update app.py

app.py

@@ -1,18 +1,33 @@
 import gradio as gr
 import pandas as pd
 
-# In-memory data store
+# === In-memory data ===
 df = pd.DataFrame(columns=["USERNAME", "COUNTRYCODE", "SBUCODE", "BRANDCODE", "CONCEPTCODE", "CHANNEL"])
+users_df = pd.DataFrame([{"USERNAME": "ADMIN", "PASSWORD": "ADMIN"}])  # default admin login
 
-def login(username, password):
-    if username == "admin" and password == "admin":
-        return gr.update(visible=False), gr.update(visible=True), True, show_data()
-    else:
-        return gr.update(visible=True, value="Invalid credentials"), gr.update(visible=False), False, pd.DataFrame()
-
+# === Utility Functions ===
 def to_upper(df):
     return df.applymap(lambda x: str(x).upper() if pd.notnull(x) else x)
 
+def show_data():
+    display_df = df.copy()
+    display_df.index.name = 'Index'
+    return display_df.reset_index()
+
+def show_users():
+    return users_df.copy()
+
+# === Login ===
+def login(username, password):
+    username = username.strip().upper()
+    password = password.strip()
+    matched = users_df[(users_df["USERNAME"] == username) & (users_df["PASSWORD"] == password)]
+    if not matched.empty:
+        return "", gr.update(visible=False), gr.update(visible=True), True, show_data()
+    else:
+        return "Invalid credentials", gr.update(visible=True), gr.update(visible=False), False, pd.DataFrame()
+
+# === Bulk Create / Update ===
 def bulk_create_or_update(bulk_df):
     global df
     bulk_df = pd.DataFrame(bulk_df).dropna(how='all')
@@ -21,66 +36,63 @@ def bulk_create_or_update(bulk_df):
 
     bulk_df = to_upper(bulk_df)
 
-    # Hanya ambil kolom penting
     bulk_df = bulk_df[["USERNAME", "COUNTRYCODE", "SBUCODE", "BRANDCODE", "CONCEPTCODE", "CHANNEL"]]
 
-    # Buat index gabungan
     merge_keys = ["USERNAME", "COUNTRYCODE", "SBUCODE", "BRANDCODE", "CONCEPTCODE", "CHANNEL"]
 
-    # 1. Remove exact duplicates (same row exists in both)
+    # Remove exact duplicates
     df_merge = pd.merge(bulk_df, df, on=merge_keys, how='inner')
     bulk_df = pd.concat([bulk_df, df_merge]).drop_duplicates(keep=False)
 
-    # 2. Pisahkan insert dan update
     existing_usernames = df["USERNAME"].unique()
     bulk_existing = bulk_df[bulk_df["USERNAME"].isin(existing_usernames)]
     bulk_new = bulk_df[~bulk_df["USERNAME"].isin(existing_usernames)]
 
-    # Update: overwrite rows with same USERNAME
     for _, row in bulk_existing.iterrows():
         df.loc[df["USERNAME"] == row["USERNAME"], ["COUNTRYCODE", "SBUCODE", "BRANDCODE", "CONCEPTCODE", "CHANNEL"]] = \
             row[["COUNTRYCODE", "SBUCODE", "BRANDCODE", "CONCEPTCODE", "CHANNEL"]].values
 
-    # Insert: append rows
     df_new = pd.DataFrame(bulk_new)
     df = pd.concat([df, df_new], ignore_index=True)
 
     return show_data()
 
+# === Delete Entry ===
 def delete_entry(index):
     global df
     try:
         idx = int(index)
         if 0 <= idx < len(df):
             df = df.drop(idx).reset_index(drop=True)
-            return gr.Textbox.update(value="Deleted."), show_data()
+            return "Deleted.", show_data()
         else:
-            return gr.Textbox.update(value="Index out of range."), show_data()
+            return "Index out of range.", show_data()
     except:
-        return gr.Textbox.update(value="Invalid index."), show_data()
-
-def update_entry(index, username, countrycode, sbucode, brandcode, conceptcode, channel):
-    global df
-    try:
-        idx = int(index)
-        if 0 <= idx < len(df):
-            df.at[idx, "USERNAME"] = username.upper()
-            df.at[idx, "COUNTRYCODE"] = countrycode.upper()
-            df.at[idx, "SBUCODE"] = sbucode.upper()
-            df.at[idx, "BRANDCODE"] = brandcode.upper()
-            df.at[idx, "CONCEPTCODE"] = conceptcode.upper()
-            df.at[idx, "CHANNEL"] = channel.upper()
-            return show_data()
-        else:
-            return gr.Textbox.update(value=f"Index {index} out of range"), show_data()
-    except Exception as e:
-        return gr.Textbox.update(value=f"Error: {e}"), show_data()
-
-def show_data():
-    display_df = df.copy()
-    display_df.index.name = 'Index'
-    return display_df.reset_index()
-
+        return "Invalid index.", show_data()
+
+# === User Management ===
+def add_user(new_username, new_password):
+    global users_df
+    new_username = new_username.strip().upper()
+    new_password = new_password.strip()
+    if not new_username or not new_password:
+        return "Username/password tidak boleh kosong.", show_users()
+
+    if new_username in users_df["USERNAME"].values:
+        return "Username sudah ada.", show_users()
+
+    users_df = pd.concat([users_df, pd.DataFrame([{"USERNAME": new_username, "PASSWORD": new_password}])], ignore_index=True)
+    return f"User {new_username} ditambahkan.", show_users()
+
+def delete_user(del_username):
+    global users_df
+    del_username = del_username.strip().upper()
+    if del_username == "ADMIN":
+        return "User ADMIN tidak bisa dihapus.", show_users()
+    users_df = users_df[users_df["USERNAME"] != del_username].reset_index(drop=True)
+    return f"User {del_username} dihapus.", show_users()
+
+# === UI ===
 with gr.Blocks() as app:
     logged_in = gr.State(False)
 
@@ -92,33 +104,45 @@ with gr.Blocks() as app:
         login_error = gr.Textbox(visible=False, interactive=False, show_label=False)
 
     with gr.Column(visible=False) as main_section:
-        gr.Markdown("## Bulk CRUD App (Auto Uppercase + Smart Merge)")
+        gr.Markdown("## Input Data User Brand Performance")
 
         with gr.Tab("Bulk Create/Update"):
             bulk_df = gr.Dataframe(
                 headers=["USERNAME", "COUNTRYCODE", "SBUCODE", "BRANDCODE", "CONCEPTCODE", "CHANNEL"],
                 row_count=5,
                 col_count=(6, "fixed"),
-                label="Paste or type data (seperti Excel)"
+                label="Paste or type data"
             )
             bulk_btn = gr.Button("Submit Bulk Data")
             bulk_table = gr.Dataframe(label="Resulting Data", interactive=False)
 
-        with gr.Tab("Manual Delete"):
+
+        with gr.Tab("List Data"):
+            view_btn = gr.Button("Refresh Table")
+            view_table = gr.Dataframe(label="All Data", interactive=False)
+
             del_index = gr.Textbox(label="Index to Delete")
             del_btn = gr.Button("Delete")
             del_status = gr.Textbox(visible=True, interactive=False, show_label=False)
             del_table = gr.Dataframe(label="Data", interactive=False)
 
-        with gr.Tab("View Data"):
-            view_btn = gr.Button("Refresh Table")
-            view_table = gr.Dataframe(label="All Data", interactive=False)
+        with gr.Tab("User Management"):
+            new_username = gr.Textbox(label="New Username")
+            new_password = gr.Textbox(label="New Password")
+            add_user_btn = gr.Button("Add User")
+            del_user = gr.Textbox(label="Username to Delete")
+            del_user_btn = gr.Button("Delete User")
+            user_status = gr.Textbox(visible=True, interactive=False, show_label=False)
+            user_table = gr.Dataframe(label="User List", interactive=False)
 
     login_btn.click(fn=login, inputs=[login_user, login_pass],
-                    outputs=[login_error, main_section, logged_in, view_table])
-
+                    outputs=[login_error, login_section, main_section, logged_in, view_table])
+    
     bulk_btn.click(fn=bulk_create_or_update, inputs=bulk_df, outputs=bulk_table)
     del_btn.click(fn=delete_entry, inputs=del_index, outputs=[del_status, del_table])
     view_btn.click(fn=show_data, outputs=view_table)
 
+    add_user_btn.click(fn=add_user, inputs=[new_username, new_password], outputs=[user_status, user_table])
+    del_user_btn.click(fn=delete_user, inputs=del_user, outputs=[user_status, user_table])
+
 app.launch()