import matplotlib.pyplot as plt
import matplotlib.patches as mpatches
def draw_piezo_battery_circuit():
fig, ax = plt.subplots(figsize = (8, 6))
# Set limits and turn off axes
ax.set_xlim(0, 10)
ax.set_ylim(0, 10)
ax.axis('off')
# Draw Battery
battery = mpatches.FancyBboxPatch((1, 7), 1, 2, boxstyle = "round,pad=0.05", edgecolor = "black", facecolor = "lightgray")
ax.add_patch(battery)
ax.text(1.5, 8.5, "Battery", ha = "center", va = "center", fontsize = 9)
# Draw TP4056 Module
tp4056 = mpatches.FancyBboxPatch((4, 7), 2, 2, boxstyle = "round,pad=0.05", edgecolor = "black", facecolor = "lightgray")
ax.add_patch(tp4056)
ax.text(5, 8.5, "TP4056", ha = "center", va = "center", fontsize = 9)
# Draw Transistor