Update upload_suggestions.py

upload_suggestions.py

@@ -1,25 +1,21 @@
 import os
-import subprocess
 import hashlib
 import time
 from huggingface_hub import HfApi
 
-DEST_PATH_IN_REPO = "suggestions/suggestions.db"
 REPO_ID = "axxam/LibreTranslate_Kabyle"
+DEST_PATH_IN_REPO = "suggestions/suggestions.db"
+HF_TOKEN = os.getenv("HF_TOKEN")
 CHECKSUM_FILE = "/app/.last_suggestions_checksum"
-HF_TOKEN = os.environ.get("HF_TOKEN")
 
-def find_suggestions_db():
-    try:
-        result = subprocess.check_output(["find", "/", "-type", "f", "-name", "suggestions.db"], stderr=subprocess.DEVNULL)
-        paths = result.decode().strip().split("\n")
-        for path in paths:
-            if os.path.exists(path):
-                print(f"Found suggestions.db at: {path}")
-                return path
-    except subprocess.CalledProcessError:
-        pass
-    print("suggestions.db not found.")
+def get_suggestions_db_path():
+    cwd = os.getcwd()
+    print(f"Running in CWD: {cwd}")  # Debug line
+    db_path = os.path.join(cwd, "db", "suggestions.db")
+    if os.path.exists(db_path):
+        print(f"Found suggestions.db at {db_path}")
+        return db_path
+    print(f"suggestions.db not found at expected path: {db_path}")
     return None
 
 def file_checksum(path):
@@ -44,7 +40,7 @@ def upload_if_updated():
         print("HF_TOKEN not set — skipping upload.")
         return
 
-    db_path = find_suggestions_db()
+    db_path = get_suggestions_db_path()
     if not db_path:
         return
 
@@ -52,12 +48,12 @@ def upload_if_updated():
     last_checksum = load_last_checksum()
 
     if current_checksum == last_checksum:
-        print("No change in suggestions.db — skipping upload.")
+        print("suggestions.db unchanged — skipping upload.")
         return
 
-    print("New changes detected — uploading suggestions.db...")
-    api = HfApi()
+    print("Uploading updated suggestions.db to Hugging Face...")
     try:
+        api = HfApi()
         api.upload_file(
             path_or_fileobj=db_path,
             path_in_repo=DEST_PATH_IN_REPO,
@@ -69,7 +65,7 @@ def upload_if_updated():
         save_checksum(current_checksum)
         print(f"Upload successful at {time.strftime('%Y-%m-%d %H:%M:%S')}")
     except Exception as e:
-        print("Upload failed:", e)
+        print(f"Upload failed: {e}")
 
 if __name__ == "__main__":
     upload_if_updated()