auth

app.py

@@ -77,10 +77,27 @@ params = {
     'state': '12345'  # Optional state parameter
 }
 
-# Redirect the user to Microsoft's OAuth endpoint
+# Construct the login URL
 login_url = f"{AUTH_URL}?{urlencode(params)}"
-print("Redirect to:", login_url)
 
+# 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',
@@ -91,9 +108,22 @@ def exchange_code_for_token(auth_code):
     }
     
     response = requests.post(TOKEN_URL, data=data)
-    token_data = response.json()
     
-    return token_data.get('access_token')
+    if response.status_code == 200:
+        token_data = response.json()
+        access_token = token_data.get('access_token')
+        return access_token
+    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
 
 # Function to retrieve answer using the RAG system
 @spaces.GPU(duration=60)
@@ -123,17 +153,34 @@ 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("Ask a question and get answers based on retrieved documents.")
+    gr.Markdown("Please log in to continue.")
     
-    input_box = gr.Textbox(label="Enter your question", placeholder="Type your question here...")
-    submit_btn = gr.Button("Submit")
-    # clear_btn = gr.Button("Clear")
-    chat_history = gr.Chatbot(label="Chat History")
+    # 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")
     
+    # 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]
+    )
+
+    # Input submission and chat handling
     submit_btn.click(chat, inputs=[input_box, chat_history], outputs=[chat_history, input_box])
-    # clear_btn.click(clear_input, outputs=input_box)
 
 interface.launch()