Update app.py

app.py

@@ -23,6 +23,12 @@ from langchain.chains.conversation.memory import ConversationBufferWindowMemory
 from langchain.agents import Tool, initialize_agent
 from huggingface_hub import login
 
+
+import requests
+from requests_oauthlib import OAuth2Session
+from oauthlib.oauth2 import BackendApplicationClient
+import os
+
 # Check if the token is already set in the environment variables
 hf_token = os.getenv("HF_TOKEN")
 
@@ -830,6 +836,26 @@ def update_images():
     image_3 = generate_image(hardcoded_prompt_3)
     return image_1, image_2, image_3
 
+
+# OAuth 2.0 endpoints for Google
+authorization_base_url = 'https://accounts.google.com/o/oauth2/auth'
+token_url = 'https://accounts.google.com/o/oauth2/token'
+scope = ['https://www.googleapis.com/auth/userinfo.profile', 'https://www.googleapis.com/auth/userinfo.email']
+
+# Function to initiate the OAuth flow
+def login():
+    google = OAuth2Session(GOOGLE_CLIENT_ID, scope=scope, redirect_uri=REDIRECT_URI)
+    authorization_url, state = google.authorization_url(authorization_base_url, access_type="offline", prompt="select_account")
+    return f'<a href="{authorization_url}" target="_self">Login with Google</a>'
+
+# Function to handle the callback
+def oauth_callback(url):
+    google = OAuth2Session(GOOGLE_CLIENT_ID, redirect_uri=REDIRECT_URI, scope=scope)
+    google.fetch_token(token_url, client_secret=GOOGLE_CLIENT_SECRET, authorization_response=url)
+    userinfo = google.get('https://www.googleapis.com/oauth2/v1/userinfo').json()
+    return f"Logged in as: {userinfo['name']} ({userinfo['email']})"
+
+
 with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
     
     with gr.Row():
@@ -870,7 +896,12 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
 
             refresh_button = gr.Button("Refresh Images")
             refresh_button.click(fn=update_images, inputs=None, outputs=[image_output_1, image_output_2, image_output_3])
-
+        with gr.Column():
+            login_button = gr.HTML(login)
+            oauth_callback_url = gr.Textbox(label="Callback URL")
+            user_info = gr.HTML()
+            
+            oauth_callback_url.submit(oauth_callback, inputs=[oauth_callback_url], outputs=[user_info])
 demo.queue()
 demo.launch(share=True)