relay_tester.py

import customtkinter as ctk
from PIL import Image
import lgpio

# =========== GPIO Setup ===========
PIN1 = 20  # Inflation valve group 1
PIN2 = 27  # Deflation valve group 1
PIN3 = 12  # Inflation valve group 2
PIN4 = 24  # Deflation valve group 2

chip = lgpio.gpiochip_open(0)

# Claim each pin as an output and start OFF (0)
for pin in (PIN1, PIN2, PIN3, PIN4):
    lgpio.gpio_claim_output(chip, pin)
    lgpio.gpio_write(chip, pin, 0)

# =========== Valve Control Functions ===========
def set_inflation():
    """Inflation: turn on pins 1 & 3, turn off deflation pins."""
    lgpio.gpio_write(chip, PIN1, 1)
    lgpio.gpio_write(chip, PIN3, 1)
    lgpio.gpio_write(chip, PIN2, 0)
    lgpio.gpio_write(chip, PIN4, 0)

def set_deflation():
    """Deflation: turn on pins 2 & 4, turn off inflation pins."""
    lgpio.gpio_write(chip, PIN1, 0)
    lgpio.gpio_write(chip, PIN3, 0)
    lgpio.gpio_write(chip, PIN2, 1)
    lgpio.gpio_write(chip, PIN4, 1)

def set_neutral():
    """Neutral: turn all pins OFF."""
    lgpio.gpio_write(chip, PIN1, 1)
    lgpio.gpio_write(chip, PIN2, 1)
    lgpio.gpio_write(chip, PIN3, 1)
    lgpio.gpio_write(chip, PIN4, 1)

def set_group(mode, group):
    """
    Controls a single group of valves (1 or 2).
    mode can be: "Deflation", "Neutral", or "Inflation".
    """
    if group == 1:
        if mode == "Deflation":
            lgpio.gpio_write(chip, PIN1, 0)
            lgpio.gpio_write(chip, PIN2, 1)
        elif mode == "Inflation":
            lgpio.gpio_write(chip, PIN1, 1)
            lgpio.gpio_write(chip, PIN2, 0)
        else:
            lgpio.gpio_write(chip, PIN1, 1)
            lgpio.gpio_write(chip, PIN2, 1)
    elif group == 2:
        if mode == "Deflation":
            lgpio.gpio_write(chip, PIN3, 0)
            lgpio.gpio_write(chip, PIN4, 1)
        elif mode == "Inflation":
            lgpio.gpio_write(chip, PIN3, 1)
            lgpio.gpio_write(chip, PIN4, 0)
        else:
            lgpio.gpio_write(chip, PIN3, 1)
            lgpio.gpio_write(chip, PIN4, 1)

# =========== UI Setup ===========
set_neutral()
ctk.set_appearance_mode("System")        # or "Dark" / "Light"
ctk.set_default_color_theme("blue")

app = ctk.CTk()
app.title("Lake Lab: Cough/Pressure Simulation")
app.geometry("600x500")

# =========== Bind "q" Key for Quick Exit ===========
def on_key(event):
    if event.char.lower() == "q":
        on_close()
app.bind("<Key>", on_key)

# =========== Header with Logos ===========
# Load images with Pillow, then wrap in CTkImage.
try:
    meshalyzer_image_pil = Image.open("img/MeshAlyzer.jpeg")
    stl_image_pil        = Image.open("img/lakelab.jpg")
except FileNotFoundError:
    # If you don't have the actual files or they're named differently,
    # adjust the paths or handle errors as needed.
    meshalyzer_image_pil = Image.new("RGB", (200, 200), color="gray")
    stl_image_pil        = Image.new("RGB", (150, 150), color="gray")

# Create CTkImage objects (optional: adjust `size=(w, h)`).
meshalyzer_ctk_image = ctk.CTkImage(dark_image=meshalyzer_image_pil, size=(220, 80))
stl_ctk_image        = ctk.CTkImage(dark_image=stl_image_pil,        size=(120, 80))

header_frame = ctk.CTkFrame(app)
header_frame.pack(fill="x", pady=10)

meshalyzer_label = ctk.CTkLabel(header_frame, text="", image=meshalyzer_ctk_image)
meshalyzer_label.pack(side="left", padx=20)

stl_label = ctk.CTkLabel(header_frame, text="", image=stl_ctk_image)
stl_label.pack(side="right", padx=20)

# =========== Main Tabview ===========
tabview = ctk.CTkTabview(app, width=580, height=360)
tabview.pack(padx=10, pady=10, fill="both", expand=True)
tabview.add("Option 1: Independent Control")
tabview.add("Option 2: Both Groups Together")

# We’ll wrap the tabview.set so switching tabs resets everything to Neutral.
original_tabview_set = tabview.set
def new_tabview_set(tab_name):
    # Reset all segmented buttons to "Neutral" whenever tabs switch
    segmented_button_group1.set("Neutral")
    segmented_button_group2.set("Neutral")
    segmented_button_both.set("Neutral")
    set_neutral()
    original_tabview_set(tab_name)
tabview.set = new_tabview_set

# =========== Option 1: Independent Control ===========
option1_frame = tabview.tab("Option 1: Independent Control")

title_label1 = ctk.CTkLabel(
    option1_frame,
    text="Independent Control of Two Valve Groups",
    font=("Helvetica", 16, "bold")
)
title_label1.pack(pady=10)

ctk.CTkLabel(option1_frame, text="Group 1 (Pins 1 & 2):", font=("Helvetica", 14)).pack()
segmented_button_group1 = ctk.CTkSegmentedButton(
    option1_frame,
    values=["Deflation", "Neutral", "Inflation"]
)
segmented_button_group1.set("Neutral")
segmented_button_group1.pack(pady=5)

ctk.CTkLabel(option1_frame, text="Group 2 (Pins 3 & 4):", font=("Helvetica", 14)).pack(pady=(20, 0))
segmented_button_group2 = ctk.CTkSegmentedButton(
    option1_frame,
    values=["Deflation", "Neutral", "Inflation"]
)
segmented_button_group2.set("Neutral")
segmented_button_group2.pack(pady=5)

def independent_group1_callback(choice):
    set_group(choice, 1)
    # Reset Option 2's control to Neutral.
    segmented_button_both.set("Neutral")

def independent_group2_callback(choice):
    set_group(choice, 2)
    segmented_button_both.set("Neutral")

segmented_button_group1.configure(command=independent_group1_callback)
segmented_button_group2.configure(command=independent_group2_callback)

# =========== Option 2: Both Groups Together ===========
option2_frame = tabview.tab("Option 2: Both Groups Together")

title_label2 = ctk.CTkLabel(
    option2_frame,
    text="Control Both Valve Groups Together",
    font=("Helvetica", 16, "bold")
)
title_label2.pack(pady=10)

segmented_button_both = ctk.CTkSegmentedButton(
    option2_frame,
    values=["Deflation", "Neutral", "Inflation"]
)
segmented_button_both.set("Neutral")
segmented_button_both.pack(pady=5)

def both_groups_callback(choice):
    if choice == "Deflation":
        set_deflation()
    elif choice == "Inflation":
        set_inflation()
    else:
        set_neutral()
    # Reset Option 1's segmented buttons to neutral.
    segmented_button_group1.set("Neutral")
    segmented_button_group2.set("Neutral")

segmented_button_both.configure(command=both_groups_callback)

# =========== On Close Cleanup ===========
def on_close():
    set_neutral()
    lgpio.gpiochip_close(chip)
    app.destroy()

app.protocol("WM_DELETE_WINDOW", on_close)

app.mainloop()