updated app.py

app.py

@@ -4,22 +4,23 @@ 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'
 CLIENT_SECRET = 'EOF8Q~kKHCRgx8tnlLM-H8e93ifetxI6x7sU6bGW'
-REDIRECT_URI = 'https://sanjeevbora-chatbot.hf.space/callback'
+REDIRECT_URI = 'https://sanjeevbora-chatbot.hf.space/' 
 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
-lock = threading.Lock()  # Create a lock
 
 class RequestHandler(BaseHTTPRequestHandler):
     def do_GET(self):
         global access_token
         if self.path.startswith("/callback"):
-            code = self.path.split("code=")[1].split("&")[0]
+            # Capture the authorization code
+            code = self.path.split("code=")[1]
             response = requests.post(TOKEN_URL, data={
                 'client_id': CLIENT_ID,
                 'client_secret': CLIENT_SECRET,
@@ -28,8 +29,7 @@ class RequestHandler(BaseHTTPRequestHandler):
                 'redirect_uri': REDIRECT_URI
             })
             token_data = response.json()
-            with lock:  # Lock access to the token
-                access_token = token_data.get('access_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.")
@@ -52,21 +52,20 @@ def check_login():
 
 def handle_login_click():
     login()
-    return "Please complete the login process in the opened window."
-
+    return check_login()
 @spaces.GPU(duration=60)
 def gradio_interface():
     with gr.Blocks() as demo:
         gr.Markdown("### Welcome to the App")
         btn_login = gr.Button("Login with Microsoft")
-        output = gr.Textbox(label="Status", interactive=False)
+        output = gr.Textbox(label="Status")
 
         btn_login.click(handle_login_click, None, output)
 
     return demo
 
 if __name__ == "__main__":
-    # Start the HTTP server in the main thread
+    # Start the HTTP server in a separate thread
     threading.Thread(target=start_http_server, daemon=True).start()
     
     # Launch Gradio app