Automatically Generate Charts from CSV Files with One Command

import csv
import sys
from pathlib import Path
import matplotlib.pyplot as plt

def generate_chart(csv_path: str) -> None:
    """Read a CSV file with headers and plot the first two numeric columns."""
    data = []
    with open(csv_path, 'r', newline='') as f:
        reader = csv.reader(f)
        headers = next(reader)
        for row in reader:
            try:
                x = float(row[0])
                y = float(row[1])
                data.append((x, y))
            except (ValueError, IndexError):
                continue
    if not data:
        print("No numeric data found.")
        return
    x_vals, y_vals = zip(*data)
    plt.plot(x_vals, y_vals, marker='o')
    plt.xlabel(headers[0] if headers else 'X')
    plt.ylabel(headers[1] if len(headers) > 1 else 'Y')
    plt.title(f"Chart from {Path(csv_path).name}")
    plt.grid(True)
    plt.savefig(Path(csv_path).with_suffix('.png'))
    print(f"Saved chart to {Path(csv_path).stem}.png")

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("Usage: python script.py <csv_file>")
        sys.exit(1)
    generate_chart(sys.argv[1])