save document f

routes.py

@@ -3,8 +3,7 @@ from flask_login import login_user, login_required, logout_user, current_user
 from app import app, db, bcrypt, login_manager
 from models import User, Listing, Booking, Image, Document
 from datetime import datetime
-from utils import generate_password, send_email 
-import os
+from utils import generate_password, send_email, save_document
 
 @login_manager.user_loader
 def load_user(user_id):
@@ -61,14 +60,17 @@ def upload_document():
 
     if request.method == 'POST':
         file = request.files['document']
-        if file:
-            filename = file.filename
-            filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-            file.save(filepath)
-            new_document = Document(filename=filename, user_id=current_user.id)
+        if not file:
+            return render_template('upload_document.html', message="No file selected.")
+
+        try:
+            saved_filename = save_document(file, 'documents')
+            new_document = Document(filename=saved_filename, user_id=current_user.id)
             db.session.add(new_document)
             db.session.commit()
             return redirect(url_for('upload_document'))
+        except ValueError as e:
+            return render_template('upload_document.html', message=str(e))
 
     return render_template('upload_document.html', message=None)