Makefile

# Makefile for math-eval dataset generator

.PHONY: help setup test clean install run-small run-medium run-large inference-setup inference-test inference-demo

# Default target
help:
	@echo "Math-Eval Dataset Generator"
	@echo "=========================="
	@echo ""
	@echo "Available targets:"
	@echo "  setup           - Set up the environment and install dependencies"
	@echo "  test            - Run the test suite"
	@echo "  install         - Install Python dependencies only"
	@echo "  clean           - Clean up generated files and outputs"
	@echo "  run-small       - Generate small dataset (10 equations)"
	@echo "  run-medium      - Generate medium dataset (100 equations)"
	@echo "  run-large       - Generate large dataset (1000 equations)"
	@echo "  verify          - Verify sample equation files"
	@echo ""
	@echo "Inference targets:"
	@echo "  inference-setup - Set up inference environment"
	@echo "  inference-test  - Test inference setup"
	@echo "  inference-demo  - Run demo inference on sample data"
	@echo ""
	@echo "Examples:"
	@echo "  make setup"
	@echo "  make test"
	@echo "  make run-small"
	@echo "  make inference-demo"

# Setup the environment
setup:
	@echo "Setting up math-eval environment..."
	python setup.py

# Install dependencies
install:
	@echo "Installing dependencies..."
	pip install -r requirements.txt

# Run comprehensive tests
test:
	@echo "Running test suite..."
	python test_suite.py

# Clean up generated files
clean:
	@echo "Cleaning up..."
	rm -rf outputs/
	rm -rf __pycache__/
	rm -rf *.pyc
	rm -f test_*.txt
	rm -f *.log
	@echo "Cleanup complete"

# Generate small dataset (10 equations)
run-small:
	@echo "Generating small dataset (10 equations)..."
	python run_pipeline.py --num_equations 10 --num_vars 2 --task all --output_dir outputs/small

# Generate medium dataset (100 equations)
run-medium:
	@echo "Generating medium dataset (100 equations)..."
	python run_pipeline.py --num_equations 100 --num_vars 2 --task all --output_dir outputs/medium

# Generate large dataset (1000 equations)
run-large:
	@echo "Generating large dataset (1000 equations)..."
	python run_pipeline.py --num_equations 1000 --num_vars 2 --task all --output_dir outputs/large

# Verify sample files
verify:
	@echo "Verifying sample equation files..."
	@if [ -f "two-vars.txt" ]; then \
		python verifier.py --file two-vars.txt; \
	else \
		echo "two-vars.txt not found"; \
	fi
	@if [ -f "three-vars.txt" ]; then \
		python verifier.py --file three-vars.txt; \
	else \
		echo "three-vars.txt not found"; \
	fi

# Generate equations only
equations:
	@echo "Generating equations only..."
	python run_pipeline.py --num_equations 100 --task equations

# Generate visual representations only  
visual:
	@echo "Generating visual representations..."
	python run_pipeline.py --task visual

# Generate counting questions only
counting:
	@echo "Generating counting questions..."
	python run_pipeline.py --task counting

# Quick test with minimal data
quick-test:
	@echo "Running quick functionality test..."
	python equation_generator.py --output_file quick_test.txt --num 3 --vars 2
	python verifier.py --file quick_test.txt
	rm -f quick_test.txt
	@echo "Quick test passed!"

# Check code quality (if you have linting tools)
lint:
	@echo "Checking code quality..."
	@if command -v flake8 >/dev/null 2>&1; then \
		flake8 *.py; \
	else \
		echo "flake8 not installed, skipping linting"; \
	fi

# Display system info
info:
	@echo "System Information:"
	@echo "==================="
	@python -c "import sys; print(f'Python: {sys.version}')"
	@python -c "import platform; print(f'Platform: {platform.system()} {platform.release()}')"
	@echo "Required packages:"
	@pip list | grep -E "(numpy|pillow|tqdm)" || echo "Some packages may be missing"

# Inference-related targets

# Setup inference environment
inference-setup:
	@echo "Setting up inference environment..."
	python setup_inference.py --model_type all

# Setup only API models
inference-setup-api:
	@echo "Setting up API models only..."
	python setup_inference.py --model_type api

# Setup only open-source models  
inference-setup-opensource:
	@echo "Setting up open-source models only..."
	python setup_inference.py --model_type opensource

# Test inference setup
inference-test:
	@echo "Testing inference setup..."
	python setup_inference.py --validate_only --model_type all

# Run demo inference
inference-demo:
	@echo "Running demo inference..."
	@if [ ! -f "inference_config.json" ]; then \
		echo "Creating sample config..."; \
		cp inference_config.json inference_config_sample.json; \
		echo "Please edit inference_config_sample.json with your API keys"; \
	fi
	@echo "Demo inference requires datasets to be generated first"
	@echo "Run: make run-small && python run_inference.py --help"

# Clean inference results
clean-inference:
	@echo "Cleaning inference results..."
	rm -rf inference_results/
	rm -f *_results.csv
	@echo "Inference cleanup complete"

# Complete clean including inference
clean-all: clean clean-inference
	@echo "Complete cleanup finished"