vacuum_simulation/main.py

import sys
import random
from PyQt5.QtWidgets import QApplication
from vacuum_simulation import VacuumSimulation

if __name__ == "__main__":
    app = QApplication(sys.argv)
    
    # Default grid size or use command line arguments
    rows, cols = 15, 15
    if len(sys.argv) >= 3:
        try:
            rows, cols = int(sys.argv[1]), int(sys.argv[2])
        except ValueError:
            print("Invalid grid dimensions. Using default 15x15.")
    
    window = VacuumSimulation(rows, cols)
    window.show()
    
    sys.exit(app.exec_())