Convert generate_texture script from bash to python

src/texture/scripts/generate_texture.py

@@ -0,0 +1,54 @@
+import argparse
+import os
+import subprocess
+
+
+def main(input_file: str, output_dir: str, prompt: str):
+    input_file = os.path.abspath(input_file)
+    output_dir = os.path.abspath(output_dir)
+
+    print('--- Initial Setup ---')
+    print(f'Input File: {input_file}')
+    print(f'Absolute Output Dir: {output_dir}')
+    print(f'Prompt: {prompt}')
+    print(f'Current Directory: {os.getcwd()}')
+    print('---------------------')
+
+    subprocess.run(['python', 'blenderLego_toObj.py', '--in_file', input_file, '--out_file', output_dir],
+                   capture_output=True, check=True)
+
+    texture_output_dir = os.path.join(output_dir, 'texture_output')
+    texture_render_dir = os.path.join(output_dir, 'texture_render')
+    os.makedirs(texture_output_dir, exist_ok=True)
+    os.makedirs(texture_render_dir, exist_ok=True)
+
+    print('Generating texture...')
+
+    subprocess.run(
+        ['python', 'generate_texture.py',
+         '--input_mesh', os.path.join(output_dir, 'lego_structure_joint.obj'),
+         '--output', texture_output_dir, '--prompt', prompt, '--production'],
+        cwd='FlashTex', check=True
+    )
+
+    print('Rendering texture...')
+
+    subprocess.run(
+        ['python', 'blender_obj_uv_normal.py', '--data_path', texture_output_dir,
+         '--obj_name', 'output_mesh', '--albedo_map', 'texture_kd.png', '--normal_map', 'None',
+         '--scale', '1.0', '--start_rot_x', '270', '--start_rot_z', '180', '--output_path', texture_render_dir],
+        cwd='blender-render-toolkit', capture_output=True, check=True
+    )
+
+    print(f'Script finished. Check outputs in {output_dir}')
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description='Generate a texture for a LEGO model and render it.')
+    parser.add_argument('input_file', type=str, help='Path to the input LEGO, an LDraw (.ldr) file.')
+    parser.add_argument('output_dir', type=str, help='Path to the directory in which to save the output files.')
+    parser.add_argument('prompt', type=str, help='The input text prompt for texture generation.')
+
+    args = parser.parse_args()
+
+    main(args.input_file, args.output_dir, args.prompt)

src/texture/scripts/generate_texture.sh

@@ -1,54 +0,0 @@
-#!/bin/bash
-
-export INPUT_FILE="$1"
-export OUTPUT_DIR="$(realpath "$2")"
-export PROMPT="$3"
-
-# Create the base output directory if it doesn't exist
-mkdir -p "$OUTPUT_DIR"
-
-echo "--- Initial Setup ---"
-echo "Input File: $INPUT_FILE"
-echo "Absolute Output Dir: $OUTPUT_DIR"
-echo "Prompt: $PROMPT"
-echo "Current Directory: $(pwd)"
-echo "---------------------"
-
-uv run blenderLego_toObj.py --in_file "$INPUT_FILE" --out_file "$OUTPUT_DIR"
-
-
-if [ ! -d "$OUTPUT_DIR" ]; then
-    echo "Error: blenderLego_toObj.py did not create directory: $OUTPUT_DIR"
-    exit 1
-fi
-
-mkdir -p "$OUTPUT_DIR/texture_output"
-mkdir -p "$OUTPUT_DIR/texture_render"
-
-echo "Generating texture..."
-
-cd FlashTex
-
-python generate_texture.py --input_mesh "$OUTPUT_DIR/lego_structure_joint.obj"  \
-                           --output "$OUTPUT_DIR/texture_output" \
-                           --prompt "$PROMPT" --production
-
-
-echo "Rendering texture..."
-
-cd ../blender-render-toolkit
-
-
-uv run blender_obj_uv_normal.py --data_path "$OUTPUT_DIR/texture_output" \
-                               --obj_name output_mesh \
-                               --albedo_map texture_kd.png \
-                               --normal_map None \
-                               --scale 1.0 \
-                               --start_rot_x 270 \
-                               --start_rot_z 180 \
-                               --output_path "$OUTPUT_DIR/texture_render/"
-
-
-cd ../
-
-echo "Script finished. Check outputs in $OUTPUT_DIR"