Update app.py

app.py

@@ -446,9 +446,49 @@ def login():
 
 @app.route("/logout")
 def logout():
+    # Fetch the current device info
+    user_agent = request.headers.get('User-Agent')
+
+    # Load the users from the user.json file
+    url = f"https://api.github.com/repos/omarnuwrar/{repository_name}/contents/user.json"
+    headers = {
+        "Authorization": f"token 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",  # Replace with your GitHub token
+        "Accept": "application/vnd.github.v3+json"
+    }
+    response = requests.get(url, headers=headers)
+    if response.status_code == 200:
+        file_info = response.json()
+        file_content = base64.b64decode(file_info['content']).decode()
+        users = eval(file_content)  # Convert the string to a list of dictionaries
+
+        # Check if the current device info is associated with any user
+        for user in users:
+            if 'device_info' in user and user['device_info'] == user_agent:
+                # Remove the device info
+                del user['device_info']
+
+                # Convert users list back to JSON string
+                updated_file_content = json.dumps(users, indent=4)
+
+                # Update the user.json file in the repository
+                update_data = {
+                    "message": "Remove device info on logout",
+                    "content": base64.b64encode(updated_file_content.encode()).decode(),
+                    "sha": file_info['sha']
+                }
+                update_response = requests.put(url, headers=headers, json=update_data)
+                if update_response.status_code == 200:
+                    print("Device info removed on logout.")
+                else:
+                    print("Failed to update user info.")
+
+                break  # Exit loop after processing
+
+    # Clear the session and log the user out
     session.clear()
     flash("You have been logged out.")
     return redirect(url_for('login'))
+
 ##############     END LOGIN    #############################################
 
 @app.route('/get_response', methods=['POST'])