from machine import Pin
import time

# Configuración de pines
rodillo = Pin(4, Pin.OUT)
sellador = Pin(2, Pin.OUT)
dir_sellador = Pin(15, Pin.OUT)
enable_S = Pin(18, Pin.OUT)
enable_R = Pin(19, Pin.OUT)

# Inicialización de contador y tamaño
contador = 0
tam = 0
muestra = 200
enable_R.value(1)
enable_S.value(1)

menu = """
***Tamaño de producto disponible***
1: Muestra (2cm)
2: Chico (8cm)
3: Grande (12cm)

Ingresa el tamaño deseado: """
# Solicitar al usuario que ingrese el tamaño deseado
seleccion = int(input(menu))

# Establecer el tamaño según la selección del usuario
if seleccion == 1:
    tam = 100
elif seleccion == 2:
    tam = 300
elif seleccion == 3:
    tam = 500
else:
    print("Tamaño incorrecto")

# Ejecutar el proceso mientras el tamaño sea mayor que 0
while True:
    while tam > 0:
        for _ in range(tam):
            enable_R.value(0)
            rodillo.on()
            time.sleep_us(800)
            enable_R.value(1)
            rodillo.off()
            time.sleep_us(800)

        for _ in range(2):
            for _ in range(100):
                enable_S.value(0)
                dir_sellador.on()
                sellador.on()
                time.sleep_us(500)
                sellador.off()
                time.sleep_us(500)
                dir_sellador.off()
                enable_S.value(1)
            time.sleep(3)
            for _ in range(100):
                dir_sellador.off()
                enable_S.value(0)
                sellador.on()
                time.sleep_us(muestra)
                sellador.off()
                time.sleep_us(muestra)
                dir_sellador.on()
                enable_S.value(1)
            time.sleep(1)
        contador += 1
        print(contador)