from qiskit.visualization import circuit_drawer
from typing import Optional
import matplotlib.pyplot as plt

class CircuitVisualizer:
    """Classe para visualização de circuitos quânticos."""
    
    @staticmethod
    def draw_circuit(circuit, filename: Optional[str] = None):
        """Desenha e opcionalmente salva o circuito."""
        fig = plt.figure(figsize=(12, 8))
        circuit_drawer(circuit, output='mpl')
        if filename:
            plt.savefig(filename)
        plt.close()