feat: Add team dropdown to interface for filtering by team

app.py

@@ -17,6 +17,7 @@ list_iface = gr.Interface(fn=compute_offer,
                     gr.Slider(0, 365, value=3, step=1, label="Days", info="Number of days to look back"),
                     gr.Slider(5000, 10000000, value=1000000, step=1, label="Minimum Sent", info="Minimum number of emails sent"),
                     gr.Dropdown(["Comcast", "Yahoo", "Hotmail", "Aol"], value="Yahoo", label="Domain"),
+                    gr.Dropdown(["All", "Team 1", "Team 2"], value="Team 2", label="Team"),
                     gr.Radio(["Newsletters", "Offers", 'Offers - IDs only'], label="Type", value="Offers - IDs only"),
                     gr.Textbox(label="Exclude list", info="Example: INH,MNP", value="INH,DHI,HHP,RTA,JVR,HTH,FNC,SCD,ENH,WIP")],
             outputs="dataframe")

utils/gradio_utils.py

@@ -246,20 +246,25 @@ def _limit_chars(input_string: str, limit: int = 35) -> str:
     return input_string[:limit]
 
 ### GENERATE TOP LISTS ###
-def compute_offer(csv_file, days_lookback, min_sent, domain, offer_type, x_list):
+def compute_offer(csv_file, days_lookback, min_sent, domain, team, offer_type, x_list, ):
     pd.set_option('display.max_colwidth', 10)
-    # cmp_list = ['MSP', 'HOM', 'NTU', 'HCK', 'DDS', 'MNP', 'PSC', 'DTL', 'GVS', 'ANP', 'WDR', 'BSG']   #1
 
-    #raw_df = pd.read_csv('tools/data/30.08.2023.gabriel.sabau.campanii.csv', parse_dates=['Data'])
     df_all = pd.read_csv(csv_file.name, parse_dates=['Data'])
-    # raw_df = pd.read_csv(csv_file.name, parse_dates=['Data']) #1
+    if team == "Team 1":
+        team_members = ['Ana Boros', 'Adrian Pop',
+                'Liviu Avram', 'Alexandru Popescu', 'Vlad Draghici']
+    elif team == "Team 2":
+        team_members = ['Cristi Rusu', 'Robert Rachiteanu', 'Adrian Sabau','Gabriel Sabau']
+    else:
+        team_members = [] # All
 
     cols = ['Campanie', 'Oferta', 'Nume', 'Server', 'User',
         'Lista Custom', 'Data', 'HClicks', 'Clicks', 'Unscribers', 'Openers',
         'Click Open', 'Leads', 'CLike', 'Complains', 'Traps', 'Send', 'ECPM', 'Comision', 'Domeniu']
 
     exclude_list = df_all[(df_all['Data'] > (pd.Timestamp('now') - pd.Timedelta(days=days_lookback))) \
-                            & (df_all['Domeniu'] == domain)]['Oferta'].unique()
+                            & (df_all['Domeniu'] == domain)\
+                            & (df_all['User'].isin(team_members))]['Oferta'].unique()
     df_all = df_all[~df_all['Oferta'].isin(exclude_list)]
     
     df_all = df_all[df_all['Send'] > int(min_sent)]