app

app.py

@@ -14,17 +14,7 @@ import cartopy.feature as cfeature
 import matplotlib.pyplot as plt
 from math import radians, sin, cos, sqrt, asin, exp
 from collections import defaultdict
-
-import base64
-import os 
-import pickle
-import time
-import json
-
-from googleapiclient.discovery import build
-from google_auth_oauthlib.flow import InstalledAppFlow
-from google.auth.transport.requests import Request
-from email.mime.text import MIMEText
+import wandb
 
 IMAGE_FOLDER = './select'
 CSV_FILE = './select.csv'
@@ -38,52 +28,6 @@ RULES = """# Plonk 🌍 🌎 🌏
 ### Click "start" to begin...
 """
 
-class GmailAPI(object):
-    def __init__(self):
-        self.scopes = ['https://www.googleapis.com/auth/gmail.compose']
-        self.service = self.build_service()
-
-    def build_service(self, ):
-        creds = None
-        if os.path.exists("token.json"):
-          creds = Credentials.from_authorized_user_file("token.json", self.scopes)
-
-        if not creds or not creds.valid:
-          if creds and creds.expired and creds.refresh_token:
-            creds.refresh(Request())
-          else:
-            json_file = json.loads(os.environ['json'])
-            # save json 
-            with open('credentials.json', 'w') as f:
-                json.dump(json_file, f)
-            flow = InstalledAppFlow.from_client_secrets_file("credentials.json", self.scopes)
-            creds = flow.run_local_server(port=0)
-            os.remove('credentials.json')
-    
-          with open("token.json", "w") as token:
-            token.write(creds.to_json())
-
-        return build("gmail", "v1", credentials=creds)
-
-    def __call__(self, text, subject=''):
-        self.send_message(self.create_message(text, subject))
-
-    def send_message(self, message, num_trials=10):
-        for _ in range(num_trials):
-            try:
-                self.service.users().messages().send(userId="me", body=message).execute()
-            except Exception as e:
-                print('An error occurred at send_message: %s' % e)
-                time.sleep(60)
-
-    def create_message(self, message_text, subject):
-        message = MIMEText(message_text)
-        message['to'] = "[email protected]"
-        message['from'] = "[email protected]"
-        message['subject'] = subject
-        raw_message = base64.urlsafe_b64encode(message.as_string().encode("utf-8"))
-        return {'raw': raw_message.decode("utf-8")}
-
 def haversine(lat1, lon1, lat2, lon2):
     if (lat1 is None) or (lon1 is None) or (lat2 is None) or (lon2 is None):
         return 0
@@ -250,12 +194,16 @@ class Engine(object):
         with open(join(self.cache_path, 'full.txt'), 'w') as f:
             print(f"{final_results}" + '\n Times: ' + times, file=f)
 
-        text = f"{final_results}" + '\n Times: ' + times
-        cache_uuid = os.path.split(self.cache_path)[1]
-        self.gmail(text, subject=f'Plonk Results [{cache_uuid}]')
+        wandb.init(project="plonk")
+        artifact = wandb.Artifact('results', type='results')
+        artifact.add_file(fname)
+        wandb.log_artifact(artifact)
+        wandb.finish()
 
 
 if __name__ == "__main__":
+    # login with the key from secret
+    wandb.login()
     import gradio as gr
     def click(state, evt: gr.SelectData):
         if state['clicked']:

requirements.txt

@@ -2,6 +2,4 @@ matplotlib==3.7.1
 pandas==2.0.1
 reverse_geocoder==1.5.1
 cartopy==0.22.0
-google-api-python-client 
-google-auth-httplib2 
-google-auth-oauthlib
+wandb==0.15.4