fix: target dir header & update make docs

Author: LVivona

makefile

@@ -1,23 +1,30 @@
 # Define variables
-TARGET_DIR=docs
-KATEX_HEADER=../docs/katex.html
-DOCS_DIR=bintensors
+ROOT_DIR := $(shell pwd)
+TARGET_DIR := $(ROOT_DIR)/../docs
+KATEX_HEADER := $(ROOT_DIR)/docs/katex.html
+DOCS_DIR := $(ROOT_DIR)/bintensors
 
 # Ensure the target directory exists
 $(TARGET_DIR):
 	mkdir -p $(TARGET_DIR)
 
-# Generate documentation for bintensors with a custom header
-docs: $(TARGET_DIR)
-	cd $(DOCS_DIR) && \
-	RUSTDOCFLAGS="--html-in-header=$(KATEX_HEADER)" cargo doc --no-deps --target-dir=$(TARGET_DIR)
+# Make sure the KaTeX header exists before trying to use it
+$(KATEX_HEADER):
+	@echo "Error: KaTeX header file not found at $(KATEX_HEADER)"
+	@exit 1
 
-docs-open: $(TARGET_DIR)
+# Base documentation command
+define docs_command
 	cd $(DOCS_DIR) && \
-	RUSTDOCFLAGS="--html-in-header=$(KATEX_HEADER)" cargo doc --no-deps --target-dir=$(TARGET_DIR) --open
+	RUSTDOCFLAGS="--html-in-header=$(KATEX_HEADER)" cargo doc --no-deps --target-dir=$(TARGET_DIR) $(1)
+endef
+
+# Generate documentation for bintensors with a custom header
+docs: $(TARGET_DIR) $(KATEX_HEADER)
+	$(call docs_command)
 
-# Clean docs
-clean:
-	rm -rf $(TARGET_DIR)
+# Generate and open documentation
+docs-open: $(TARGET_DIR) $(KATEX_HEADER)
+	$(call docs_command,--open)
 
-.PHONY: docs clean
+.PHONY: docs docs-open