auth

app.py

@@ -22,7 +22,7 @@ import re
 import transformers
 import spaces
 import requests
-from urllib.parse import urlencode
+from urllib.parse import urlencode, urlparse, parse_qs
 
 # Initialize embeddings and ChromaDB
 model_name = "sentence-transformers/all-mpnet-base-v2"
@@ -120,14 +120,18 @@ def exchange_code_for_token(auth_code):
     }
     
     response = requests.post(TOKEN_URL, data=data)
-    
     if response.status_code == 200:
         token_data = response.json()
-        access_token = token_data.get('access_token')
-        return access_token
+        return token_data.get('access_token')
     else:
         return None
 
+def handle_redirect(auth_code):
+    token = exchange_code_for_token(auth_code)
+    if token:
+        return True  # Successful login
+    return False  # Failed login
+
 def login_user(auth_code):
     # Exchange the authorization code for an access token
     token = exchange_code_for_token(auth_code)
@@ -165,33 +169,36 @@ def chat(query, history=None):
 def clear_input():
     return "",  # Return empty string to clear input field
 
+# Gradio interface
 with gr.Blocks() as interface:
     gr.Markdown("## RAG Chatbot")
     gr.Markdown("Please log in to continue.")
     
-    # Custom HTML to show login link
-    login_link = gr.HTML(f'<a href="{login_url}" target="_blank">Click here to login with Microsoft</a>')
-
-    # Login button to simulate the login process
-    login_button = gr.Button("Login")
+    login_link = gr.HTML(show_login_button())
+    login_status = gr.Textbox(label="Login Status", visible=False)
     
     # Components for chat (initially hidden)
     input_box = gr.Textbox(label="Enter your question", placeholder="Type your question here...", visible=False)
     submit_btn = gr.Button("Submit", visible=False)
     chat_history = gr.Chatbot(label="Chat History", visible=False)
 
-    # Handle login button click
-    login_button.click(
-        login_user, 
-        inputs=[], 
-        outputs=[login_button],  # You can also update the UI to show login status
-        queue=False
-    ).then(
-        lambda token: check_login(is_logged_in(token)), 
-        inputs=[], 
-        outputs=[input_box, submit_btn]
-    )
-
+    # Handle the login link click
+    login_link.click(lambda: gr.update(visible=False), inputs=[], outputs=[login_link])
+    
+    # Example redirection logic (replace with actual URL handling)
+    @gr.Interface().on_submit
+    def on_redirect(url):
+        # Extract the authorization code from the URL
+        parsed_url = urlparse(url)
+        query_params = parse_qs(parsed_url.query)
+        auth_code = query_params.get('code')
+        
+        if auth_code:
+            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.")
+    
     # Input submission and chat handling
     submit_btn.click(chat, inputs=[input_box, chat_history], outputs=[chat_history, input_box])