From 34a23ab94de8fabaa5a3601273160ae22b1e91ad Mon Sep 17 00:00:00 2001
From: Henri Sellis <henrisel@ut.ee>
Date: Fri, 24 Oct 2025 10:03:16 +0300
Subject: [PATCH] make 3d model generation model folder loc configurable

---
 3d-generation-pipeline/.env.example      | 3 ++-
 3d-generation-pipeline/start_pipeline.py | 7 ++-----
 2 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/3d-generation-pipeline/.env.example b/3d-generation-pipeline/.env.example
index 997ad3dd..7d09a8f9 100644
--- a/3d-generation-pipeline/.env.example
+++ b/3d-generation-pipeline/.env.example
@@ -1,2 +1,3 @@
 CLOUDFLARE_ACCOUNT_ID=
-CLOUDFLARE_API_TOKEN=
\ No newline at end of file
+CLOUDFLARE_API_TOKEN=
+MODEL_FOLDER=
\ No newline at end of file
diff --git a/3d-generation-pipeline/start_pipeline.py b/3d-generation-pipeline/start_pipeline.py
index a7fb455c..282ec3f8 100644
--- a/3d-generation-pipeline/start_pipeline.py
+++ b/3d-generation-pipeline/start_pipeline.py
@@ -2,6 +2,7 @@ import os
 import base64
 import requests
 import argparse
+import subprocess
 from pathlib import Path
 from datetime import datetime
 from dotenv import load_dotenv
@@ -10,6 +11,7 @@ load_dotenv()
 
 ACCOUNT_ID = os.environ["CLOUDFLARE_ACCOUNT_ID"]
 API_TOKEN = os.environ["CLOUDFLARE_API_TOKEN"]
+MODEL_FOLDER = os.environ["MODEL_FOLDER"]
 
 
 def get_timestamp():
@@ -67,11 +69,6 @@ def refine_text_prompt(prompt):
     return data["result"]["response"]
 
 def image_to_3d(image_path, output_path):
-    import subprocess
-
-    # TODO: make this configurable
-    MODEL_FOLDER = r"D:\henrisel\AI-models\stable-fast-3d"
-
     venv_python = MODEL_FOLDER + r"\.venv\Scripts\python.exe"
     script_path = MODEL_FOLDER + r"\run.py"