update app.py

app.py

@@ -9,6 +9,7 @@ from selenium.webdriver.chrome.service import Service as ChromeService
 from selenium.webdriver.chrome.options import Options
 from webdriver_manager.chrome import ChromeDriverManager
 from selenium.webdriver.common.by import By
+from selenium.webdriver import ActionChains
 import time
 import os
 import gspread
@@ -16,7 +17,7 @@ import pandas as pd
 from pathlib import Path
 import gradio as gr
 
-# %% slack_selenium_solution.ipynb 2
+# %% slack_selenium_solution.ipynb 1
 class SlackAnalytics():
     def __init__(self, 
                  signin_url = 'https://thriveprojectgroup.slack.com/sign_in_with_password', 
@@ -24,11 +25,13 @@ class SlackAnalytics():
                  slackbot_url= 'https://app.slack.com/client/T01BD3PFRU4/D04GAGYT34Z'):
         self.signin_url = signin_url
         self.members_stas_url = members_stas_url
-        self.slackbot_url = slackbot_url        
+        self.slackbot_url = slackbot_url
+#         self.path = Path.home()/'Downloads'
+        self.path = Path('.')
         options = Options()
         options.add_argument('--incognito')
         options.add_argument("--headless")
-        options.add_experimental_option("prefs", {"download.default_directory": os.getcwd()})
+        options.add_experimental_option("prefs", {"download.default_directory": str(self.path)})
         
         service = ChromeService(ChromeDriverManager().install())
         self.driver = webdriver.Chrome(options=options, service=service)
@@ -71,21 +74,25 @@ class SlackAnalytics():
         self.driver.get(url=self.slackbot_url)
         time.sleep(10) # wait until the file is loaded
         self.driver.find_element(By.XPATH, '//span[contains(text(), "THRIVE Project Member Analytics All time")]').click()
+        time.sleep(2)
         self.driver.find_element(By.CSS_SELECTOR, "button[data-qa='more_file_actions']").click()
+        time.sleep(2)
         self.driver.find_element(By.XPATH, '//div[contains(text(), "Delete file")]').click()
+        time.sleep(2)
         self.driver.find_element(By.XPATH, '//button[contains(text(), "Yes, Delete This File")]').click()
-
+    
         # Close the browser windows and ends the WebDriver session
-        time.sleep(1)
+        time.sleep(2)
         self.driver.quit()
  
 
-# %% slack_selenium_solution.ipynb 3
+# %% slack_selenium_solution.ipynb 2
 class GoogleSheets():
     def __init__(self, creds_file='slack-analytics-creds.json', sh_file='gs_info.txt'):
         self.creds = creds_file
         self.sh = sh_file
-        self.folder = os.getcwd()
+#         self.path = Path.home()/'Downloads'
+        self.path = Path('.')
         
     def get_sheet_info(self):
         sh_id, wk_name = open(self.sh, 'r').read().splitlines()
@@ -103,30 +110,34 @@ class GoogleSheets():
         wk.clear()
         
         # Read and remove the downoaded csv file and update the data to Google Sheets
-        for fn in os.listdir(self.folder):
+        for fn in os.listdir(self.path):
             if fn.find('THRIVE Project Member Analytics All time') != -1:
-                file = os.path.join(self.folder, fn)
+                file = os.path.join(self.path, fn)
                 df = pd.read_csv(file, low_memory=False)
                 df = df.fillna('')
                 os.remove(file)
         
                 # update the new data to Google Sheets
                 wk.update([df.columns.values.tolist()] + df.values.tolist())
+                return df[['Name', 'Messages posted', 'Last active (UTC)']].head()
 
 
-# %% slack_selenium_solution.ipynb 7
+# %% slack_selenium_solution.ipynb 6
 def run(account, creds_file, sh_file):
     SlackAnalytics().export_csv(account.name)
-    GoogleSheets(creds_file.name, sh_file.name).update_worksheet()
-    return 'Slack Analytics data update Google Sheets completed!'
+    df_sub = GoogleSheets(creds_file.name, sh_file.name).update_worksheet()
+    return ('Slack Analytics data update Google Sheets completed!', df_sub)
 
-# %% slack_selenium_solution.ipynb 8
+# %% slack_selenium_solution.ipynb 7
 iface = gr.Interface(fn=run, 
                      inputs=[gr.File(label='Slack Account File'), 
                              gr.File(label='Google Credentials Json File'), 
                              gr.File(label='Googlesheet Key File')], 
-                     outputs=gr.Text(),
+                     outputs=[gr.Text(), 
+                              gr.DataFrame(headers=['Name', 'Messages posted', 'Last active (UTC)'], 
+                                           datatype=['str', 'number', 'str'])
+                                           ],
                      allow_flagging='never', 
                      title='Slack Analytics Members Data Getter', 
                      description='Download Slack analytics data From Slack and upload it to Google Sheets')
-iface.launch(height=450, width=500, debug=True)
+iface.launch(height=450, width=500, share=True)