Update app.py

app.py

@@ -1,9 +1,15 @@
 import gradio as gr
 import pandas as pd
 
-# In-memory storage
+# In-memory data store
 df = pd.DataFrame(columns=["USERNAME", "COUNTRYCODE", "SBUCODE", "BRANDCODE", "CONCEPTCODE", "CHANNEL"])
 
+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()
+
 def add_entry(username, countrycode, sbucode, brandcode, conceptcode, channel):
     global df
     new_row = {
@@ -15,12 +21,12 @@ def add_entry(username, countrycode, sbucode, brandcode, conceptcode, channel):
         "CHANNEL": channel
     }
     df = pd.concat([df, pd.DataFrame([new_row])], ignore_index=True)
-    return df
+    return show_data()
 
 def bulk_input(bulk_df):
     global df
     df = pd.concat([df, pd.DataFrame(bulk_df).dropna(how='all')], ignore_index=True)
-    return df
+    return show_data()
 
 def delete_entry(index):
     global df
@@ -28,11 +34,11 @@ def delete_entry(index):
         idx = int(index)
         if 0 <= idx < len(df):
             df = df.drop(idx).reset_index(drop=True)
-            return df
+            return show_data()
         else:
-            return f"Index {index} out of range"
+            return gr.Textbox.update(value=f"Index {index} out of range"), show_data()
     except Exception as e:
-        return f"Error: {e}"
+        return gr.Textbox.update(value=f"Error: {e}"), show_data()
 
 def update_entry(index, username, countrycode, sbucode, brandcode, conceptcode, channel):
     global df
@@ -45,61 +51,87 @@ def update_entry(index, username, countrycode, sbucode, brandcode, conceptcode,
             df.at[idx, "BRANDCODE"] = brandcode
             df.at[idx, "CONCEPTCODE"] = conceptcode
             df.at[idx, "CHANNEL"] = channel
-            return df
+            return show_data()
         else:
-            return f"Index {index} out of range"
+            return gr.Textbox.update(value=f"Index {index} out of range"), show_data()
     except Exception as e:
-        return f"Error: {e}"
+        return gr.Textbox.update(value=f"Error: {e}"), show_data()
 
 def show_data():
-    return df
+    display_df = df.copy()
+    display_df.index.name = 'Index'
+    return display_df.reset_index()
 
 with gr.Blocks() as app:
-    gr.Markdown("## CRUD App with Excel-style Bulk Input")
-
-    with gr.Tab("Add Entry"):
-        with gr.Row():
-            username = gr.Textbox(label="USERNAME")
-            countrycode = gr.Textbox(label="COUNTRYCODE")
-            sbucode = gr.Textbox(label="SBUCODE")
-            brandcode = gr.Textbox(label="BRANDCODE")
-            conceptcode = gr.Textbox(label="CONCEPTCODE")
-            channel = gr.Textbox(label="CHANNEL")
-        add_btn = gr.Button("Add")
-        add_btn.click(add_entry, 
-                      inputs=[username, countrycode, sbucode, brandcode, conceptcode, channel], 
-                      outputs=gr.Dataframe())
-
-    with gr.Tab("Bulk Input (Excel-like)"):
-        bulk_df = gr.Dataframe(
-            headers=["USERNAME", "COUNTRYCODE", "SBUCODE", "BRANDCODE", "CONCEPTCODE", "CHANNEL"],
-            row_count=5,
-            col_count=(6, "fixed"),
-            label="Paste or type your data here"
-        )
-        bulk_btn = gr.Button("Submit Bulk Data")
-        bulk_btn.click(bulk_input, inputs=bulk_df, outputs=gr.Dataframe())
-
-    with gr.Tab("Update Entry"):
-        upd_index = gr.Textbox(label="Index to Update")
-        upd_username = gr.Textbox(label="USERNAME")
-        upd_country = gr.Textbox(label="COUNTRYCODE")
-        upd_sbu = gr.Textbox(label="SBUCODE")
-        upd_brand = gr.Textbox(label="BRANDCODE")
-        upd_concept = gr.Textbox(label="CONCEPTCODE")
-        upd_channel = gr.Textbox(label="CHANNEL")
-        update_btn = gr.Button("Update Entry")
-        update_btn.click(update_entry, 
-                         inputs=[upd_index, upd_username, upd_country, upd_sbu, upd_brand, upd_concept, upd_channel], 
-                         outputs=gr.Dataframe())
-
-    with gr.Tab("Delete Entry"):
-        del_index = gr.Textbox(label="Index to Delete")
-        del_btn = gr.Button("Delete")
-        del_btn.click(delete_entry, inputs=del_index, outputs=gr.Dataframe())
-
-    with gr.Tab("View Data"):
-        view_btn = gr.Button("Refresh Table")
-        view_btn.click(show_data, outputs=gr.Dataframe())
+    logged_in = gr.State(False)
+
+    with gr.Column(visible=True) as login_section:
+        gr.Markdown("## Login")
+        login_user = gr.Textbox(label="Username")
+        login_pass = gr.Textbox(label="Password", type="password")
+        login_btn = gr.Button("Login")
+        login_error = gr.Textbox(visible=False, interactive=False, show_label=False)
+
+    with gr.Column(visible=False) as main_section:
+        gr.Markdown("## CRUD App - Logged in as Admin")
+
+        with gr.Tab("Add Entry"):
+            with gr.Row():
+                username = gr.Textbox(label="USERNAME")
+                countrycode = gr.Textbox(label="COUNTRYCODE")
+                sbucode = gr.Textbox(label="SBUCODE")
+                brandcode = gr.Textbox(label="BRANDCODE")
+                conceptcode = gr.Textbox(label="CONCEPTCODE")
+                channel = gr.Textbox(label="CHANNEL")
+            add_btn = gr.Button("Add")
+            add_table = gr.Dataframe(label="Data", interactive=False)
+
+        with gr.Tab("Bulk Input"):
+            bulk_df = gr.Dataframe(
+                headers=["USERNAME", "COUNTRYCODE", "SBUCODE", "BRANDCODE", "CONCEPTCODE", "CHANNEL"],
+                row_count=5,
+                col_count=(6, "fixed"),
+                label="Paste or type your data here"
+            )
+            bulk_btn = gr.Button("Submit Bulk Data")
+            bulk_table = gr.Dataframe(label="Data", interactive=False)
+
+        with gr.Tab("Update Entry"):
+            upd_index = gr.Textbox(label="Index to Update")
+            upd_username = gr.Textbox(label="USERNAME")
+            upd_country = gr.Textbox(label="COUNTRYCODE")
+            upd_sbu = gr.Textbox(label="SBUCODE")
+            upd_brand = gr.Textbox(label="BRANDCODE")
+            upd_concept = gr.Textbox(label="CONCEPTCODE")
+            upd_channel = gr.Textbox(label="CHANNEL")
+            update_btn = gr.Button("Update Entry")
+            upd_table = gr.Dataframe(label="Data", interactive=False)
+
+        with gr.Tab("Delete Entry"):
+            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="Data", interactive=False)
+
+    login_btn.click(fn=login, inputs=[login_user, login_pass], 
+                    outputs=[login_error, main_section, logged_in, view_table])
+
+    add_btn.click(add_entry, 
+                  inputs=[username, countrycode, sbucode, brandcode, conceptcode, channel], 
+                  outputs=add_table)
+
+    bulk_btn.click(bulk_input, inputs=bulk_df, outputs=bulk_table)
+
+    update_btn.click(update_entry, 
+                     inputs=[upd_index, upd_username, upd_country, upd_sbu, upd_brand, upd_concept, upd_channel], 
+                     outputs=[upd_table])
+
+    del_btn.click(delete_entry, inputs=del_index, outputs=[del_status, del_table])
+
+    view_btn.click(show_data, outputs=view_table)
 
 app.launch()