gene

server.py

@@ -2,9 +2,6 @@
 import json
 from upload_video import DownloadVideos, GetRefreshTokenFromCode
 
-REFRESH_TOKEN = None
-VIDEOS = None
-
 class MyHandler(server.BaseHTTPRequestHandler):
     def do_GET(self):
         # send index.html file to client 
@@ -50,7 +47,6 @@ def do_POST(self):
             return
         try: 
             refresh_token = GetRefreshTokenFromCode(code)
-            globals()['REFRESH_TOKEN'] = refresh_token
         except Exception as e:
             print("Error getting refresh token:", e)
             self.send_response(500)
@@ -59,16 +55,24 @@ def do_POST(self):
             self.wfile.write(b'Error getting refresh token: ' + str(e).encode('utf-8'))
             return
 
-        globals()['VIDEOS'] = videos
+        print("Downloading videos...")
         self.send_response(200)
         self.end_headers()
-        self.wfile.write(b'Videos will be uploaded shortly')
-        self.server.shutdown()
+        self.wfile.write(b'Videos downloading...')
+        # send response to client
+        self.finish()
+
+        try:
+            DownloadVideos(refresh_token, videos)
+        except Exception as e:
+            print("Error downloading videos:", e)
+
+        exit(0)
 
 
 server_address = ('', 8000)
 httpd = server.HTTPServer(server_address, MyHandler)
 httpd.serve_forever()
 
-DownloadVideos(REFRESH_TOKEN, VIDEOS)
+