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"
@@ -86,30 +86,15 @@ params = {
     'redirect_uri': REDIRECT_URI,
     'response_mode': 'query',
     'scope': 'User.Read',
-    'state': '12345'  # Optional state parameter
+    'state': '12345'
 }
 
 # Construct the login URL
 login_url = f"{AUTH_URL}?{urlencode(params)}"
 
-# Gradio interface
 def show_login_button():
     return f'<a href="{login_url}" target="_blank">Click here to login with Microsoft</a>'
 
-# Dummy function to simulate token validation (you will replace this with actual validation)
-def is_logged_in(token):
-    # Check if the token exists (or check if it's valid)
-    return token is not None
-
-# Gradio interface
-def check_login(status):
-    # If logged in, show the chatbot interface, otherwise show login link
-    if status:
-        return gr.update(visible=True), gr.update(visible=True)
-    else:
-        return gr.update(visible=False), gr.update(visible=False)
-    
-# Function to exchange authorization code for access token
 def exchange_code_for_token(auth_code):
     data = {
         'grant_type': 'authorization_code',
@@ -128,14 +113,15 @@ def exchange_code_for_token(auth_code):
     else:
         return None
 
-def login_user(auth_code):
-    # Exchange the authorization code for an access token
-    token = exchange_code_for_token(auth_code)
-    
-    if token:
-        return token
-    else:
-        return None
+def handle_redirect(url):
+    parsed_url = urlparse(url)
+    query_params = parse_qs(parsed_url.query)
+    auth_code = query_params.get('code')
+
+    if auth_code:
+        token = exchange_code_for_token(auth_code[0])
+        return token  # return the token or handle accordingly
+    return None
 
 # Function to retrieve answer using the RAG system
 @spaces.GPU(duration=60)
@@ -152,7 +138,6 @@ def test_rag(query):
     
     return corrected_text_books
 
-# Define the Gradio interface
 def chat(query, history=None):
     if history is None:
         history = []
@@ -161,38 +146,26 @@ def chat(query, history=None):
         history.append((query, answer))
     return history, ""  # Clear input after submission
 
-# Function to clear input text
-def clear_input():
-    return "",  # Return empty string to clear input field
-
 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())
     
     # 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]
+    # Add a function to handle the redirect from Microsoft
+    redirect_url_input = gr.Textbox(label="Redirect URL", visible=False)
+
+    redirect_url_input.change(
+        handle_redirect,
+        inputs=[redirect_url_input],
+        outputs=[input_box, submit_btn]  # Update visibility based on login
     )
 
-    # Input submission and chat handling
     submit_btn.click(chat, inputs=[input_box, chat_history], outputs=[chat_history, input_box])
 
 interface.launch()