Update app.py

app.py

@@ -299,64 +299,53 @@ def process_account_action(action, account_username, password, first_name=None,
 
 ################    LOGIN          #################################
 
-@app.route("/sign-up", methods=["GET", "POST"])
-def create_account():
-    if request.method == 'POST':
-        first_name = request.form['first_name']
-        last_name = request.form['last_name']
-        account_username = request.form['username']
-        password = request.form['password']
-
-        # Handling file upload
-        if 'profile_image' not in request.files:
-            flash('No file part')
-            return redirect(request.url)
-        file = request.files['profile_image']
-        if file.filename == '':
-            flash('No selected file')
-            return redirect(request.url)
-        if file:
-            filename = secure_filename(f"{account_username}.png")
-            file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-            file.save(file_path)
-
-        result = process_account_action("create", account_username, password, first_name, last_name, file_path)
-        flash(result)
-        if "successfully" in result:
-            return redirect(url_for('login'))
-
-    return render_template('sign-up.html')
-
-@app.route("/chat")
-def chat():
-    if 'account_name' not in session or session['account_name'] is None:
-        return redirect(url_for('login'))
-
-    account_name = session.get("account_name")
-    first = session.get("first")
-    last = session.get("last")
-    token2 = session.get("token2")
-    greet = session.get("greet")
-    
-    return render_template('ai-chat-bot.html', account_name=account_name, greet=greet, first=first, last=last, token2=token2)
+@app.route("/login", methods=["GET", "POST"])
+def login():
+    # Check if device info matches any stored device info
+    user_agent = request.headers.get('User-Agent')
 
+    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 device info matches any user
+        for user in users:
+            if 'device_info' in user and user['device_info'] == user_agent:
+                # Device info matches, automatically log in the user
+                session['account_name'] = user['username']
+                session['first'] = user['first_name']
+                session['last'] = user['last_name']
+                session['token2'] = user['token']
 
-@app.route("/login", methods=["GET", "POST"])
-def login():
+                # Initialize the user input and get the assistant's response
+                conversation_history = {}
+                user_input = f"user name : {user['first_name']} \n \n message : أهلا"
+                greet = get_assistant_response(user_input, conversation_history)
+
+                session['greet'] = greet
+
+                # Redirect to the chat page
+                return redirect(url_for('chat'))
+
+    # If no match is found or if the user is logging in normally
     message = None
     if request.method == "POST":
         account_name = request.form.get("username")
         password = request.form.get("password")
+        remember_me = request.form.get("remember_me")  # Check if "Remember Me" was selected
+
         message = process_account_action("login", account_name, password)
         flash(message)
         if "Login successful." in message:
             ############### GET INFO ABOUT USER ###############################################
             url = f"https://api.github.com/repos/omarnuwrar/{repository_name}/contents/user.json"
-            headers = {
-                "Authorization": f"token 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",
-                "Accept": "application/vnd.github.v3+json"
-            }
             response = requests.get(url, headers=headers)
             if response.status_code == 200:
                 file_info = response.json()
@@ -364,7 +353,6 @@ def login():
                 users = eval(file_content)  # Convert the string to a list of dictionaries
 
                 target_username = account_name
-
                 target_user = None
                 for user in users:
                     if user['username'] == target_username:
@@ -374,14 +362,35 @@ def login():
                 first = target_user['first_name']
                 last = target_user['last_name']
                 token2 = target_user['token']
-                user_name.append(first)
 
+                ############### REMEMBER ME IMPLEMENTATION ########################################
+                if remember_me:
+                    user_agent = request.headers.get('User-Agent')
+                    target_user['device_info'] = user_agent
+
+                    # Convert users list back to JSON string
+                    updated_file_content = json.dumps(users, indent=4)
+
+                    # Update the user.json file in the repository
+                    update_url = f"https://api.github.com/repos/omarnuwrar/{repository_name}/contents/user.json"
+                    update_data = {
+                        "message": f"Update device info for {account_name}",
+                        "content": base64.b64encode(updated_file_content.encode()).decode(),
+                        "sha": file_info['sha']
+                    }
+                    update_response = requests.put(update_url, headers=headers, json=update_data)
+                    if update_response.status_code == 200:
+                        print("User info updated with device info.")
+                    else:
+                        print("Failed to update user info.")
+                ############### END REMEMBER ME IMPLEMENTATION ####################################
+                
             ############## END INFO ABOUT USER #################################################
 
             # Initialize the user input and get the assistant's response
             conversation_history = {}
             user_input = f"user name : {first} \n \n message : أهلا"
-            greet = get_assistant_response(user_input , conversation_history )
+            greet = get_assistant_response(user_input, conversation_history)
 
             # Store data in the session
             session['account_name'] = account_name
@@ -395,6 +404,11 @@ def login():
     
     return render_template("sign-in.html", message=message)
 
+@app.route("/logout")
+def logout():
+    session.clear()
+    flash("You have been logged out.")
+    return redirect(url_for('login'))
 ##############     END LOGIN    #############################################
 
 @app.route('/get_response', methods=['POST'])