Update main.py

main.py

@@ -5,6 +5,7 @@ import subprocess
 app = Flask(__name__)
 UPLOAD_FOLDER = 'uploads'
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 16MB max upload size
 
 # HTML template for the index page
 HTML_TEMPLATE = """
@@ -44,6 +45,10 @@ def upload_files():
     bat_filename = bat_file.filename
     bat_path = os.path.join(app.config['UPLOAD_FOLDER'], bat_filename)
 
+    # Ensure the upload directory exists
+    if not os.path.exists(app.config['UPLOAD_FOLDER']):
+        os.makedirs(app.config['UPLOAD_FOLDER'])
+
     # Save the uploaded file
     bat_file.save(bat_path)