Update app.py

app.py

@@ -318,7 +318,7 @@ def register():
             flash('Пользователь уже существует!')
             return redirect(url_for('register'))
 
-        if user_type == 'seller':
+        if 'register_seller' in request.form and user_type == 'seller':
             org_name = request.form.get('org_name')
             org_phone = request.form.get('org_phone')
             is_online = request.form.get('is_online') == 'on'
@@ -347,7 +347,7 @@ def register():
             save_data(data)
             flash('Регистрация успешна! Ваша организация на проверке, мы с вами свяжемся.')
             return redirect(url_for('login'))
-        else:
+        elif 'register_buyer' in request.form and user_type == 'buyer':
             data['users'][username] = {
                 'password': password,
                 'bio': '',
@@ -359,6 +359,9 @@ def register():
             save_data(data)
             flash('Регистрация успешна! Войдите в систему.')
             return redirect(url_for('login'))
+        else:
+            flash('Выберите тип пользователя и нажмите соответствующую кнопку!')
+            return redirect(url_for('register'))
 
     is_authenticated = 'username' in session
     username = session.get('username', None)
@@ -448,7 +451,8 @@ def register():
                     <label><input type="checkbox" name="is_online" onclick="toggleAddress()"> Организация онлайн</label>
                     <input type="text" name="org_address" placeholder="Адрес организации" id="org_address">
                 </div>
-                <button type="submit" class="btn">Зарегистрироваться</button>
+                <button type="submit" name="register_buyer" class="btn" id="buyer-btn">Зарегистрироваться как покупатель</button>
+                <button type="submit" name="register_seller" class="btn" id="seller-btn" style="display: none;">Зарегистрироваться как продавец</button>
             </form>
             <p class="link"><a href="{{ url_for('login') }}">Уже есть аккаунт? Войти</a></p>
         </div>
@@ -462,6 +466,8 @@ def register():
             }
             function toggleSellerFields(show) {
                 document.getElementById('seller-fields').style.display = show ? 'block' : 'none';
+                document.getElementById('buyer-btn').style.display = show ? 'none' : 'block';
+                document.getElementById('seller-btn').style.display = show ? 'block' : 'none';
             }
             function toggleAddress() {
                 const isOnline = document.querySelector('input[name="is_online"]').checked;