auth

app.py

@@ -153,32 +153,28 @@ def chat_interface():
     
     return interface
 
-# Function to handle OAuth callback and extract authorization code
-def handle_auth_callback(query_params):
+# Function to handle OAuth callback
+def handle_auth_callback(auth_code):
     global access_token
 
-    # Parse the query parameters to extract the authorization code
-    parsed_url = urlparse(query_params)
-    query_dict = parse_qs(parsed_url.query)
-    
-    if 'code' in query_dict:
-        auth_code = query_dict['code'][0]
-        access_token = exchange_code_for_token(auth_code)
-
+    # Exchange authorization code for access token
+    access_token = exchange_code_for_token(auth_code)
     return "Authentication successful. You can now use the chatbot."
 
 # Gradio app launch
 with gr.Blocks() as app:
     gr.Markdown("## OAuth2.0 Chatbot")
 
-    # 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)
+    # Add an input field to manually input the authorization code for testing
+    auth_code_input = gr.Textbox(label="Enter the OAuth Authorization Code")
+
+    # Button to handle authentication and exchange the code for the access token
+    auth_button = gr.Button("Authenticate")
+
+    # Callback for authentication
+    auth_button.click(fn=handle_auth_callback, inputs=auth_code_input, outputs="text")
 
     # Display the chat interface or authentication prompt
     chat_interface()
 
-app.launch()
-
+app.launch()