Update src/saving_utils.py

src/saving_utils.py

@@ -1,7 +1,16 @@
 import os
 import pandas as pd
 
-def save_similarity_output(output_dict, method_name, leaderboard_path="/home/user/app/src/data/leaderboard_results.csv", similarity_path="/home/user/app/src/data/similarity_results.csv"):
+script_dir = os.path.dirname(os.path.abspath(__file__))  # Directory of the running script
+
+
+def save_similarity_output(output_dict, method_name, leaderboard_path="data/leaderboard_results.csv", similarity_path="data/similarity_results.csv"):
+    leaderboard_path = os.path.join(script_dir, leaderboard_path)
+    similarity_path = os.path.join(script_dir, similarity_path)
+
+    with open("test_write.txt", "w") as f:
+        f.write("Write test successful!")
+    
     # Load or initialize the DataFrames
     if os.path.exists(leaderboard_path):
         leaderboard_df = pd.read_csv(leaderboard_path)
@@ -68,7 +77,8 @@ def save_similarity_output(output_dict, method_name, leaderboard_path="/home/use
     # Save the updated DataFrames back to CSV
     similarity_df.to_csv(similarity_path, index=False)
     leaderboard_df.to_csv(leaderboard_path, index=False)
-
+    print(f"Updated files saved to {similarity_path} and {leaderboard_path}")
+    
     return 0
 
 def save_function_output(model_output, method_name, func_results_path="/home/user/app/src/data/function_results.csv", leaderboard_path="/home/user/app/src/data/leaderboard_results.csv"):