authentication

app.py

@@ -7,6 +7,8 @@ from transformers import AutoConfig, AutoTokenizer, pipeline, AutoModelForCausal
 from langchain_community.document_loaders import DirectoryLoader
 import torch
 import re
+import requests
+from urllib.parse import urlencode, parse_qs, urlparse
 import transformers
 import spaces
 
@@ -24,20 +26,11 @@ books_db_client = books_db.as_retriever()
 
 # Initialize the model and tokenizer
 model_name = "stabilityai/stablelm-zephyr-3b"
-
-# bnb_config = transformers.BitsAndBytesConfig(
-#     load_in_4bit=True,
-#     bnb_4bit_quant_type='nf4',
-#     bnb_4bit_use_double_quant=True,
-#     bnb_4bit_compute_dtype=torch.bfloat16
-# )
-
 model_config = transformers.AutoConfig.from_pretrained(model_name, max_new_tokens=1024)
 model = transformers.AutoModelForCausalLM.from_pretrained(
     model_name,
     trust_remote_code=True,
     config=model_config,
-    # quantization_config=bnb_config,
     device_map=device,
 )
 
@@ -66,6 +59,62 @@ books_db_client_retriever = RetrievalQA.from_chain_type(
     verbose=True
 )
 
+# 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"
+
+# Function to redirect to Microsoft login
+def get_login_url():
+    params = {
+        'client_id': CLIENT_ID,
+        'response_type': 'code',
+        'redirect_uri': REDIRECT_URI,
+        'response_mode': 'query',
+        'scope': 'User.Read',
+        'state': '12345'  # Optional state parameter for CSRF protection
+    }
+    login_url = f"{AUTH_URL}?{urlencode(params)}"
+    return login_url
+
+# Function to exchange auth code for an access token
+def exchange_code_for_token(auth_code):
+    data = {
+        'grant_type': 'authorization_code',
+        'client_id': CLIENT_ID,
+        'client_secret': CLIENT_SECRET,
+        'code': auth_code,
+        'redirect_uri': REDIRECT_URI
+    }
+    response = requests.post(TOKEN_URL, data=data)
+    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):
@@ -81,7 +130,7 @@ def test_rag(query):
     
     return corrected_text_books
 
-# Define the Gradio interface
+# Function for RAG Chat
 def chat(query, history=None):
     if history is None:
         history = []
@@ -90,10 +139,6 @@ 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
-
 # Gradio interface
 with gr.Blocks() as interface:
     gr.Markdown("## RAG Chatbot")
@@ -101,10 +146,33 @@ with gr.Blocks() as interface:
     
     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")
+
+    # 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_btn.click(chat, inputs=[input_box, chat_history], outputs=[chat_history, input_box])
-    # clear_btn.click(clear_input, outputs=input_box)
 
 interface.launch()