auth

app.py

@@ -123,8 +123,7 @@ def exchange_code_for_token(auth_code):
     if response.status_code == 200:
         token_data = response.json()
         return token_data.get('access_token')
-    else:
-        return None
+    return None
 
 def handle_redirect(auth_code):
     token = exchange_code_for_token(auth_code)
@@ -174,7 +173,6 @@ with gr.Blocks() as interface:
     gr.Markdown("## RAG Chatbot")
     gr.Markdown("Please log in to continue.")
     
-    login_link = gr.HTML(show_login_button())
     login_status = gr.Textbox(label="Login Status", visible=False)
     
     # Components for chat (initially hidden)
@@ -182,11 +180,13 @@ with gr.Blocks() as interface:
     submit_btn = gr.Button("Submit", visible=False)
     chat_history = gr.Chatbot(label="Chat History", visible=False)
 
-    # Handle the login link click
-    login_link.click(lambda: gr.update(visible=False), inputs=[], outputs=[login_link])
+    # Button for login
+    login_button = gr.Button("Click here to login with Microsoft")
+    
+    # Handle the login button click
+    login_button.click(lambda: gr.update(visible=False), inputs=[], outputs=[login_button])
     
-    # Example redirection logic (replace with actual URL handling)
-    @gr.Interface().on_submit
+    # Function to handle the redirect
     def on_redirect(url):
         # Extract the authorization code from the URL
         parsed_url = urlparse(url)
@@ -197,9 +197,10 @@ with gr.Blocks() as interface:
             if handle_redirect(auth_code[0]):  # Pass the first code if there are multiple
                 return gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
             else:
-                login_status.update("Login failed.")
-    
+                return gr.update(value="Login failed."),  # Show failure message
+
     # Input submission and chat handling
     submit_btn.click(chat, inputs=[input_box, chat_history], outputs=[chat_history, input_box])
 
+# Add an event listener to check for redirects in the Gradio interface (this might require running an additional server)
 interface.launch()