Update app.py

app.py

@@ -276,9 +276,11 @@ def delete_post(post_id):
 
 
 
+
 @app.route('/follow/<int:user_id>', methods=['POST'])
 @login_required
-def follow(user_id, page):
+def follow(user_id):
+    
     user_to_follow = User.query.get(user_id)
     if user_to_follow is None:
         flash('User not found.', 'error')
@@ -296,12 +298,7 @@ def follow(user_id, page):
         return redirect(url_for('view_profile', user_id=user_id))
 
     flash(f"You are now following {user_to_follow.username}.", 'success')
-    if page == 'profile':
-        return redirect(url_for('view_profile', user_id=user_id))
-    elif page == 'search':
-        return redirect(url_for('search_results', query=request.form['query']))
-    else:
-        return redirect(url_for('view_profile', user_id=user_id))
+    return redirect(url_for('view_profile', user_id=user_id))
 
 
 @app.route('/unfollow/<int:user_id>', methods=['POST'])