auth

app.py

@@ -8,7 +8,7 @@ from langchain_community.document_loaders import DirectoryLoader
 import torch
 import re
 import requests
-from urllib.parse import urlencode, parse_qs, urlparse
+from urllib.parse import urlencode
 import transformers
 import spaces
 
@@ -60,13 +60,12 @@ books_db_client_retriever = RetrievalQA.from_chain_type(
 )
 
 # 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/'
-AUTH_URL = f"https://login.microsoftonline.com/2b093ced-2571-463f-bc3e-b4f8bcb427ee/oauth2/v2.0/authorize"
-TOKEN_URL = f"https://login.microsoftonline.com/2b093ced-2571-463f-bc3e-b4f8bcb427ee/oauth2/v2.0/token"
-GRAPH_API_URL = "https://graph.microsoft.com/v1.0/me"
+TENANT_ID = 'your-tenant-id'
+CLIENT_ID = 'your-client-id'
+CLIENT_SECRET = 'your-client-secret'
+REDIRECT_URI = 'https://your-chatbot.hf.space/auth/callback'
+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"
 
 # Function to redirect to Microsoft login
 def get_login_url():
@@ -94,27 +93,6 @@ def exchange_code_for_token(auth_code):
     token_data = response.json()
     return token_data.get('access_token')
 
-# Step 3: Function to get user profile
-def get_user_profile(access_token):
-    headers = {
-        'Authorization': f'Bearer {access_token}'
-    }
-    response = requests.get(GRAPH_API_URL, headers=headers)
-    return response.json()
-
-# Function to handle OAuth callback
-def handle_oauth_callback(url):
-    parsed_url = urlparse(url)
-    query_params = parse_qs(parsed_url.query)
-    auth_code = query_params.get('code', [None])[0]
-    
-    if auth_code:
-        access_token = exchange_code_for_token(auth_code)
-        user_profile = get_user_profile(access_token)
-        return user_profile
-    else:
-        return "Authorization failed."
-
 # Function to retrieve answer using the RAG system
 @spaces.GPU(duration=60)
 def test_rag(query):
@@ -150,29 +128,15 @@ with gr.Blocks() as interface:
 
     # Add Microsoft OAuth Login
     auth_btn = gr.Button("Login with Microsoft")
-    
-    # OAuth callback URL input (for demonstration, replace with actual callback handler)
-    callback_url = gr.Textbox(label="OAuth Callback URL", placeholder="Paste the callback URL here...")
-
-    # Display user profile after login
-    profile_output = gr.JSON(label="User Profile")
 
     # Action for OAuth login
     def login_action():
         return gr.redirect(get_login_url())
 
-    # Action for handling OAuth callback and displaying the user profile
-    def handle_callback_action(url):
-        user_profile = handle_oauth_callback(url)
-        return user_profile
-
     # Bind login action to button
     auth_btn.click(login_action)
     
-    # Bind OAuth callback handler to the callback input
-    callback_url.change(handle_callback_action, inputs=[callback_url], outputs=[profile_output])
-    
-    # Submit action for chat
+    # Submit action
     submit_btn.click(chat, inputs=[input_box, chat_history], outputs=[chat_history, input_box])
 
 interface.launch()