updated app.py

app.py

@@ -4,7 +4,6 @@ import webbrowser
 from http.server import BaseHTTPRequestHandler, HTTPServer
 import threading
 import spaces
-
 # OAuth Configuration
 TENANT_ID = '2b093ced-2571-463f-bc3e-b4f8bcb427ee'
 CLIENT_ID = '2a7c884c-942d-49e2-9e5d-7a29d8a0d3e5'
@@ -13,13 +12,13 @@ REDIRECT_URI = 'https://sanjeevbora-chatbot.hf.space/callback'
 AUTH_URL = f"https://login.microsoftonline.com/{TENANT_ID}/oauth2/v2.0/authorize"
 TOKEN_URL = f"https://login.microsoftonline.com/{TENANT_ID}/oauth2/v2.0/token"
 SCOPE = 'User.Read'
-access_token = None  # Global access token
+access_token = None
+lock = threading.Lock()  # Create a lock
 
 class RequestHandler(BaseHTTPRequestHandler):
     def do_GET(self):
         global access_token
         if self.path.startswith("/callback"):
-            # Capture the authorization code
             code = self.path.split("code=")[1].split("&")[0]
             response = requests.post(TOKEN_URL, data={
                 'client_id': CLIENT_ID,
@@ -29,7 +28,8 @@ class RequestHandler(BaseHTTPRequestHandler):
                 'redirect_uri': REDIRECT_URI
             })
             token_data = response.json()
-            access_token = token_data.get('access_token')
+            with lock:  # Lock access to the token
+                access_token = token_data.get('access_token')
             self.send_response(200)
             self.end_headers()
             self.wfile.write(b"Login successful! You can close this window.")
@@ -53,7 +53,7 @@ def check_login():
 def handle_login_click():
     login()
     return "Please complete the login process in the opened window."
-    
+
 @spaces.GPU(duration=60)
 def gradio_interface():
     with gr.Blocks() as demo:
@@ -66,7 +66,7 @@ def gradio_interface():
     return demo
 
 if __name__ == "__main__":
-    # Start the HTTP server in a separate thread
+    # Start the HTTP server in the main thread
     threading.Thread(target=start_http_server, daemon=True).start()
     
     # Launch Gradio app