Update app.py

app.py

@@ -42,6 +42,7 @@ def handle_upload():
         
         # Create a temporary directory to store uploaded files
         temp_dir = tempfile.mkdtemp()
+        logger.info(f"Temporary directory created at {temp_dir}")
         
         # Save the uploaded files to the temporary directory
         folder_path = os.path.join(temp_dir, 'uploaded_folder')
@@ -49,11 +50,13 @@ def handle_upload():
         for file_obj in files:
             file_path = os.path.join(folder_path, file_obj.filename)
             file_obj.save(file_path)
+            logger.info(f"File saved to {file_path}")
         
         # Save the script content to a file
         script_path = os.path.join(folder_path, 'user_script.py')
         with open(script_path, 'w') as script_file:
             script_file.write(script_content)
+        logger.info(f"Script content saved to {script_path}")
         
         # Run the script using MPI
         log_output = run_script_with_mpi(script_path, folder_path)
@@ -64,6 +67,7 @@ def handle_upload():
             for root, _, files in os.walk(folder_path):
                 for file in files:
                     zipf.write(os.path.join(root, file), os.path.relpath(os.path.join(root, file), folder_path))
+        logger.info(f"Output folder zipped at {zip_path}")
         
         return jsonify({"status": "success", "log_output": log_output, "download_url": f"/download/{os.path.basename(zip_path)}"})