Update app.py

app.py

@@ -4,31 +4,13 @@ from oauth2client.service_account import ServiceAccountCredentials
 import gradio as gr
 import plotly.express as px
 
-# ------------------ Authentication ------------------
-VALID_USERS = {
-    "[email protected]": "Pass.123",
-    "[email protected]": "Pass.123",
+# === Allowed Users ===
+allowed_users = {
     "[email protected]": "Pass.123",
+    "[email protected]": "Pass.123",
+    "[email protected]": "Pass.123"
 }
 
-def login():
-    st.title("🔐 Login Required")
-    email = st.text_input("Email")
-    password = st.text_input("Password", type="password")
-    if st.button("Login"):
-        if VALID_USERS.get(email) == password:
-            st.session_state.authenticated = True
-            st.rerun()
-        else:
-            st.error("❌ Incorrect email or password.")
-
-if "authenticated" not in st.session_state:
-    st.session_state.authenticated = False
-
-if not st.session_state.authenticated:
-    login()
-    st.stop()
-
 # === Google Sheets Auth ===
 scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
 creds = ServiceAccountCredentials.from_json_keyfile_name("tough-star.json", scope)
@@ -53,10 +35,10 @@ df = df.sort_values(by=['Rep Name', 'Timestamp'])
 df['Time Diff (min)'] = df.groupby(['Rep Name', 'Date'])['Timestamp'].diff().dt.total_seconds().div(60).fillna(0)
 df['Visit Order'] = df.groupby(['Rep Name', 'Date']).cumcount() + 1
 
-# === Helper: All unique reps in dataset ===
+# === All reps ===
 all_reps = sorted(df['Rep Name'].dropna().unique())
 
-# === Tab 1: Summary ===
+# === Summary logic ===
 def generate_summary(date_str):
     day_df = df[df['Date'] == date_str]
     active = day_df.groupby('Rep Name').size().reset_index(name='Total Visits')
@@ -65,7 +47,7 @@ def generate_summary(date_str):
     inactive_df = pd.DataFrame({'Inactive Reps': inactive_list})
     return active, inactive_df
 
-# === Tab 2: KAMs ===
+# === KAM logic ===
 def get_reps(date_str):
     reps = df[df['Date'] == date_str]['Rep Name'].dropna().unique()
     return gr.update(choices=sorted(reps))
@@ -112,7 +94,6 @@ def show_map(date_str, rep):
 
     fig.update_layout(mapbox_style="open-street-map", title=f"📍 {rep}'s Route on {date_str}")
 
-    # Final table
     table = subset[[ 
         'Visit Order', 'Dealership Name', 'Time', 'Time Diff (min)', 
         'Type of call', 'Sales or service'
@@ -124,7 +105,6 @@ def show_map(date_str, rep):
         'Sales or service': '💼 Category'
     })
 
-    # Add summary row
     total_time = round(table['⏱️ Time Spent'].sum(), 2)
     summary_row = pd.DataFrame([{
         'Visit Order': '',
@@ -138,28 +118,47 @@ def show_map(date_str, rep):
 
     return table, fig
 
-# === Gradio UI ===
+# === Gradio App ===
+def build_dashboard():
+    with gr.Blocks() as dashboard:
+        gr.Markdown("## 🗂️ Carfind Rep Tracker")
+
+        with gr.Tab("📊 Summary"):
+            date_summary = gr.Dropdown(label="Select Date", choices=sorted(df['Date'].unique(), reverse=True))
+            active_table = gr.Dataframe(label="✅ Active Reps (with total visits)")
+            inactive_table = gr.Dataframe(label="⚠️ Inactive Reps")
+            date_summary.change(fn=generate_summary, inputs=date_summary, outputs=[active_table, inactive_table])
+
+        with gr.Tab("👤 KAM's"):
+            with gr.Row():
+                with gr.Column(scale=1):
+                    date_picker = gr.Dropdown(label="Select Date", choices=sorted(df['Date'].unique(), reverse=True))
+                    rep_picker = gr.Dropdown(label="Select Rep")
+                    btn = gr.Button("Show Route")
+                with gr.Column(scale=2):
+                    table = gr.Dataframe(label="Call Table")
+
+            map_plot = gr.Plot(label="Map")
+            date_picker.change(fn=get_reps, inputs=date_picker, outputs=rep_picker)
+            btn.click(fn=show_map, inputs=[date_picker, rep_picker], outputs=[table, map_plot])
+    return dashboard
+
+def authenticate(email, password):
+    if email in allowed_users and allowed_users[email] == password:
+        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
+    else:
+        return "Invalid login", None, None
+
 with gr.Blocks() as app:
-    gr.Markdown("## 🗂️ Carfind Rep Tracker")
-
-    with gr.Tab("📊 Summary"):
-        date_summary = gr.Dropdown(label="Select Date", choices=sorted(df['Date'].unique(), reverse=True))
-        active_table = gr.Dataframe(label="✅ Active Reps (with total visits)")
-        inactive_table = gr.Dataframe(label="⚠️ Inactive Reps")
-        date_summary.change(fn=generate_summary, inputs=date_summary, outputs=[active_table, inactive_table])
-
-    with gr.Tab("👤 KAM's"):
-        with gr.Row():
-            with gr.Column(scale=1):
-                date_picker = gr.Dropdown(label="Select Date", choices=sorted(df['Date'].unique(), reverse=True))
-                rep_picker = gr.Dropdown(label="Select Rep")
-                btn = gr.Button("Show Route")
-            with gr.Column(scale=2):
-                table = gr.Dataframe(label="Call Table")
-
-        map_plot = gr.Plot(label="Map")
-
-        date_picker.change(fn=get_reps, inputs=date_picker, outputs=rep_picker)
-        btn.click(fn=show_map, inputs=[date_picker, rep_picker], outputs=[table, map_plot])
+    gr.Markdown("## 🔐 Login to Access Carfind Rep Tracker")
+    with gr.Row():
+        email_input = gr.Textbox(label="Email")
+        pass_input = gr.Textbox(label="Password", type="password")
+    login_button = gr.Button("Login")
+    error_box = gr.Markdown("", visible=False)
+    dashboard = build_dashboard()
+    dashboard.visible = False
+
+    login_button.click(fn=authenticate, inputs=[email_input, pass_input], outputs=[error_box, error_box, dashboard])
 
 app.launch()