label new

app.py

@@ -299,13 +299,19 @@ ha='center', va='center', fontsize=14, color='gray')
 def run_training():
     """Runs the fine-tuning script and returns the tail of the output."""
     try:
+        # Get the absolute path to the script's directory
+        script_dir = os.path.dirname(os.path.abspath(__file__))
+        triplets_path = os.path.join(script_dir, "data", "esg_triplets.csv")
+        create_triplets_script = os.path.join(script_dir, "create_triplets.py")
+        train_finetune_script = os.path.join(script_dir, "train_finetune.py")
+
         # Ensure the triplets file exists before running training
-        if not os.path.exists("data/esg_triplets.csv"):
+        if not os.path.exists(triplets_path):
             print("Triplet file not found. Running 'create_triplets.py' script...")
-            subprocess.run([sys.executable, "create_triplets.py"], check=True)
+            subprocess.run([sys.executable, create_triplets_script], check=True)
 
         result = subprocess.run(
-            [sys.executable, "train_finetune.py"],
+            [sys.executable, train_finetune_script],
             capture_output=True,
             text=True,
             check=True