Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, request, redirect, session
+from flask import Flask, request, redirect, session, url_for
 import requests
 import os
 
@@ -29,11 +29,22 @@ def login():
         if access_token and user_id:
             session['access_token'] = access_token
             session['user_id'] = user_id
-            return redirect('/')
+            return redirect(url_for('protected_route'))
         else:
             return "Ошибка при получении access_token", 400
     else:
         return "Код авторизации не найден", 400
 
+def is_authenticated():
+    return 'access_token' in session and 'user_id' in session
+
+@app.route('/protected')
+def protected_route():
+    if is_authenticated():
+        user_id = session['user_id']
+        return f"Добро пожаловать, пользователь с ID {user_id}!"
+    else:
+        return redirect(url_for('login'))
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))