Update app.py

app.py

@@ -23,6 +23,51 @@ from langchain.chains.conversation.memory import ConversationBufferWindowMemory
 from langchain.agents import Tool, initialize_agent
 from huggingface_hub import login
 
+# Flask app for authentication
+auth_app = Flask(__name__)
+auth_app.secret_key = os.urandom(24)
+
+client_id = os.getenv("GOOGLE_CLIENT_ID")
+client_secret = os.getenv("GOOGLE_CLIENT_SECRET")
+authorization_base_url = 'https://accounts.google.com/o/oauth2/auth'
+token_url = 'https://accounts.google.com/o/oauth2/token'
+redirect_uri = 'http://localhost:5000/callback'
+
+scope = [
+    "https://www.googleapis.com/auth/userinfo.profile",
+    "https://www.googleapis.com/auth/userinfo.email",
+    "openid",
+]
+
+@auth_app.route("/")
+def home():
+    return '<a href="/login">Login with Google</a>'
+
+@auth_app.route("/login")
+def login():
+    google = OAuth2Session(client_id, scope=scope, redirect_uri=redirect_uri)
+    authorization_url, state = google.authorization_url(authorization_base_url, access_type="offline")
+    session['oauth_state'] = state
+    return redirect(authorization_url)
+
+@auth_app.route("/callback")
+def callback():
+    google = OAuth2Session(client_id, state=session['oauth_state'], redirect_uri=redirect_uri)
+    token = google.fetch_token(token_url, client_secret=client_secret, authorization_response=request.url)
+    session['oauth_token'] = token
+    return redirect(url_for('.profile'))
+
+@auth_app.route("/profile")
+def profile():
+    google = OAuth2Session(client_id, token=session['oauth_token'])
+    response = google.get('https://www.googleapis.com/oauth2/v1/userinfo')
+    user_info = response.json()
+    return f"Hello, {user_info['name']}! Your email is {user_info['email']}."
+
+# Check if user is authenticated
+def check_auth():
+    if 'oauth_token' not in session:
+        return redirect(url_for('login'))
 
 
 
@@ -975,5 +1020,8 @@ 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])
 
-demo.queue()
-demo.launch(share=True)
+if __name__ == "__main__":
+    from threading import Thread
+    Thread(target=auth_app.run, kwargs={'port': 5000, 'debug': True}).start()
+    demo.queue()
+    demo.launch(share=True)