auth

app.py

@@ -67,20 +67,22 @@ 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"
 
-# Function to redirect to Microsoft login
-def get_login_url():
+# Global variable to store the access token
+access_token = None
+
+# OAuth Authorization URL with parameters
+def get_auth_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
+        'state': '12345'  # Optional state parameter
     }
-    login_url = f"{AUTH_URL}?{urlencode(params)}"
-    return login_url
+    return f"{AUTH_URL}?{urlencode(params)}"
 
-# Function to exchange auth code for an access token
+# Exchange authorization code for an access token
 def exchange_code_for_token(auth_code):
     data = {
         'grant_type': 'authorization_code',
@@ -93,50 +95,90 @@ def exchange_code_for_token(auth_code):
     token_data = response.json()
     return token_data.get('access_token')
 
+# Function to fetch user profile from Microsoft Graph
+def get_user_profile(token):
+    headers = {
+        'Authorization': f'Bearer {token}'
+    }
+    response = requests.get(GRAPH_API_URL, headers=headers)
+    return response.json()
+
+# Function to check if the user is authenticated
+def is_authenticated():
+    return access_token is not None
+
 # Function to retrieve answer using the RAG system
 @spaces.GPU(duration=60)
-def test_rag(query):
-    books_retriever = books_db_client_retriever.run(query)
-    
-    # Extract the relevant answer using regex
-    corrected_text_match = re.search(r"Helpful Answer:(.*)", books_retriever, re.DOTALL)
+# Gradio app with OAuth integration
+def chat_interface():
+    global access_token
+
+    # If the user is not authenticated, redirect to Microsoft login
+    if not is_authenticated():
+        auth_url = get_auth_url()
+        return gr.Markdown(f"Please [log in]({auth_url}) to use the chatbot.")
+
+    def test_rag(query):
+        books_retriever = books_db_client_retriever.run(query)
+        
+        # Extract the relevant answer using regex
+        corrected_text_match = re.search(r"Helpful Answer:(.*)", books_retriever, re.DOTALL)
+        
+        if corrected_text_match:
+            corrected_text_books = corrected_text_match.group(1).strip()
+        else:
+            corrected_text_books = "No helpful answer found."
+        
+        return corrected_text_books
     
-    if corrected_text_match:
-        corrected_text_books = corrected_text_match.group(1).strip()
-    else:
-        corrected_text_books = "No helpful answer found."
+    # Gradio chatbot interface
+    def chat(query, history=None):
+        if history is None:
+            history = []
+        if query:
+            # Chatbot logic here
+            answer = "This is a dummy answer."
+            history.append((query, answer))
+        return history, ""  # Clear input after submission
+
+    with gr.Blocks() as interface:
+        gr.Markdown("## RAG Chatbot")
+        gr.Markdown("Ask a question and get answers based on retrieved documents.")
+        
+        input_box = gr.Textbox(label="Enter your question", placeholder="Type your question here...")
+        submit_btn = gr.Button("Submit")
+        chat_history = gr.Chatbot(label="Chat History")
+        
+        submit_btn.click(chat, inputs=[input_box, chat_history], outputs=[chat_history, input_box])
     
-    return corrected_text_books
-
-# Function for RAG Chat
-def chat(query, history=None):
-    if history is None:
-        history = []
-    if query:
-        answer = test_rag(query)
-        history.append((query, answer))
-    return history, ""  # Clear input after submission
-
-# Gradio interface
-with gr.Blocks() as interface:
-    gr.Markdown("## RAG Chatbot")
-    gr.Markdown("Ask a question and get answers based on retrieved documents.")
+    return interface
+
+# Function to handle OAuth callback and extract authorization code
+def handle_auth_callback(query_params):
+    global access_token
+
+    # Parse the query parameters to extract the authorization code
+    parsed_url = urlparse(query_params)
+    query_dict = parse_qs(parsed_url.query)
     
-    input_box = gr.Textbox(label="Enter your question", placeholder="Type your question here...")
-    submit_btn = gr.Button("Submit")
-    chat_history = gr.Chatbot(label="Chat History")
+    if 'code' in query_dict:
+        auth_code = query_dict['code'][0]
+        access_token = exchange_code_for_token(auth_code)
 
-    # Add Microsoft OAuth Login
-    auth_btn = gr.Button("Login with Microsoft")
+    return "Authentication successful. You can now use the chatbot."
 
-    # Action for OAuth login
-    def login_action():
-        return gr.redirect(get_login_url())
+# Gradio app launch
+with gr.Blocks() as app:
+    gr.Markdown("## OAuth2.0 Chatbot")
 
-    # Bind login action to button
-    auth_btn.click(login_action)
-    
-    # Submit action
-    submit_btn.click(chat, inputs=[input_box, chat_history], outputs=[chat_history, input_box])
+    # Check for the authentication callback
+    app_url = spaces.get_url()
+    if app_url.startswith(REDIRECT_URI):
+        query_params = app_url.split('?')[-1]
+        handle_auth_callback(query_params)
+
+    # Display the chat interface or authentication prompt
+    chat_interface()
+
+app.launch()
 
-interface.launch()