Update main.py

main.py

@@ -4,11 +4,10 @@ import string
 
 app = Flask(__name__)
 
-# В памяти храним зарегистрированные передачи файлов
+# В памяти храним зарегистрированных пользователей и сессии передачи файлов
+users = {}
 transfers = {}
 
-BASE_URL = "https://starchik1-share.hf.space/d/"  # Постоянный серверный URL
-
 def generate_short_id(length=8):
     """
     Генерирует короткий уникальный идентификатор для передачи файла.
@@ -18,6 +17,18 @@ def generate_short_id(length=8):
         if token not in transfers:
             return token
 
+@app.route('/register', methods=['POST'])
+def register():
+    data = request.json
+    username = data.get('username')
+    password = data.get('password')
+    if not username or not password:
+        return jsonify({'error': 'Не переданы обязательные поля'}), 400
+    if username in users:
+        return jsonify({'error': 'Пользователь уже существует'}), 400
+    users[username] = {'password': password}
+    return jsonify({'message': 'Регистрация прошла успешно'})
+
 @app.route('/create_transfer', methods=['POST'])
 def create_transfer():
     data = request.json
@@ -37,8 +48,9 @@ def create_transfer():
         'filesize': filesize,
         'downloaded': False
     }
-    
-    return jsonify({'token': transfer_id})
+    # Формируем короткую ссылку для скачивания (маршрут изменён на /d/)
+    download_link = request.host_url + 'd/' + transfer_id
+    return jsonify({'download_link': download_link})
 
 @app.route('/d/<transfer_id>', methods=['GET'])
 def get_transfer(transfer_id):
@@ -48,4 +60,5 @@ def get_transfer(transfer_id):
     return jsonify(transfer)
 
 if __name__ == '__main__':
+    # Запускаем сервер на всех интерфейсах на порту 5000
     app.run(host='0.0.0.0', port=7860)