Update main.py

main.py

@@ -20,8 +20,8 @@ HTML_TEMPLATE = """
     <form action="/upload" method="post" enctype="multipart/form-data">
         <label for="bat_file">Upload BAT file:</label>
         <input type="file" name="bat_file" required><br><br>
-        <label for="icon_file">Upload ICO file:</label>
-        <input type="file" name="icon_file" required><br><br>
+        <label for="icon_file">Upload ICO file (optional):</label>
+        <input type="file" name="icon_file"><br><br>
         <input type="submit" value="Upload and Compile">
     </form>
 </body>
@@ -34,22 +34,25 @@ def index():
 
 @app.route('/upload', methods=['POST'])
 def upload_files():
-    if 'bat_file' not in request.files or 'icon_file' not in request.files:
-        return "No file part", 400
+    if 'bat_file' not in request.files:
+        return "No BAT file part", 400
 
     bat_file = request.files['bat_file']
-    icon_file = request.files['icon_file']
-
-    if bat_file.filename == '' or icon_file.filename == '':
-        return "No selected file", 400
+    if bat_file.filename == '':
+        return "No selected BAT file", 400
 
     # Define the paths to save the uploaded files
     bat_path = os.path.join(app.config['UPLOAD_FOLDER'], bat_file.filename)
-    icon_path = os.path.join(app.config['UPLOAD_FOLDER'], icon_file.filename)
 
-    # Save the uploaded files
+    # Save the uploaded BAT file
     bat_file.save(bat_path)
-    icon_file.save(icon_path)
+
+    # Check if an icon file was uploaded, and set the icon path accordingly
+    icon_path = ""
+    if 'icon_file' in request.files and request.files['icon_file'].filename != '':
+        icon_file = request.files['icon_file']
+        icon_path = os.path.join(app.config['UPLOAD_FOLDER'], icon_file.filename)
+        icon_file.save(icon_path)
 
     # Construct the NSIS script using the provided paths
     nsi_script = f"""!include "MUI2.nsh"
@@ -67,9 +70,8 @@ VIProductVersion "1.4.44.3"
 # Set the output directory and the base name of the installer file
 SetCompressor /SOLID lzma
 
-# Use a multi-resolution icon file with 16x16, 32x32, 64x64 sizes
-Icon "{icon_path}"
-RequestExecutionLevel admin
+# Set icon only if an icon file was provided
+{"Icon \"" + icon_path + "\"\n" if icon_path else ""}RequestExecutionLevel admin
 
 # Silent install settings
 SilentInstall silent