able to get the url that the frontend is running on for callback in a…

…pp.py

backend/api/app.py

@@ -68,11 +68,18 @@ def create_text_indexes():
 
 @app.route('/login')
 def login():
-    auth_url, _ = flow.authorization_url(prompt='consent')
+    origin = request.headers.get('Origin')
+    if not origin:
+        return jsonify({"error": "Origin header missing"}), 400
+    auth_url, _ = flow.authorization_url(prompt='consent', state=origin)
     return jsonify({"auth_url": auth_url})
 
 @app.route('/callback')
 def callback():
+    # Retrieve the origin from the state parameter
+    origin = request.args.get('state')
+    redirect_uri = origin + '/home'
+    print(redirect_uri)
     flow.fetch_token(authorization_response=request.url)
     credentials = flow.credentials
     session['credentials'] = {
@@ -83,7 +90,7 @@ def callback():
         'client_secret': credentials.client_secret,
         'scopes': credentials.scopes,
     }
-    return redirect("http://127.0.0.1:5173/home")  # Redirect to React app
+    return redirect(redirect_uri)  # Redirect to React app
 
 @app.route('/get_user_calendar')
 def get_user_calendar():