Update app.py

app.py

@@ -39,35 +39,44 @@ os.environ['AUTH_TOKEN_KEY'] = AUTH_TOKEN_KEY
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
 if not HF_TOKEN:
-    raise ValueError("🚨 HF_TOKEN is not set! Ensure you have the right permissions.")
+    raise ValueError("🚨 HF_TOKEN is not set! Ensure you have the correct permissions.")
 
-from huggingface_hub import HfApi
+# 🔹 Verify authentication works
+try:
+    user_info = api.whoami(token=HF_TOKEN)
+    print(f"✅ Authenticated as: {user_info['name']}")
+except Exception as e:
+    raise ValueError(f"🚨 Hugging Face authentication failed: {e}")
+from huggingface_hub import RepositoryNotFoundError
 
+repo_id = f"{HF_ORG_NAME}/{DATASET_NAME}"  # Use your organization dataset
 api = HfApi()
 
-# 🔹 Check if the dataset exists in the organization
+# 🔹 Verify if dataset exists
 try:
-    api.repo_info(repo_id, repo_type="dataset", token=HF_TOKEN)
-    print(f"✅ Dataset '{repo_id}' already exists in the organization.")
-except Exception:
+    repo_info = api.repo_info(repo_id, repo_type="dataset", token=HF_TOKEN)
+    print(f"✅ Dataset '{repo_id}' exists.")
+except RepositoryNotFoundError:
     print(f"🔴 Dataset '{repo_id}' not found. Creating it now...")
 
-    # Create repo inside the organization
-    api.create_repo(
-        repo_id=repo_id,
-        repo_type="dataset",
-        private=True,  # Ensure it's private
-        token=HF_TOKEN,
-        organization=HF_ORG_NAME  # Specify organization
-    )
-    print(f"✅ Dataset '{repo_id}' created successfully in the organization.")
+    try:
+        api.create_repo(
+            repo_id=repo_id,
+            repo_type="dataset",
+            private=True,  # ✅ Keep it private
+            token=HF_TOKEN,
+            organization=HF_ORG_NAME
+        )
+        print(f"✅ Dataset '{repo_id}' created successfully.")
+    except Exception as e:
+        raise ValueError(f"🚨 Failed to create dataset '{repo_id}': {e}")
 
 db_path = "chatbot.db"
 
-# 🔹 Ensure `chatbot.db` exists before uploading
+# 🔹 Check if chatbot.db exists
 if not os.path.exists(db_path):
     print("🔴 chatbot.db does not exist! Creating it now...")
-    import sqlite3
+
     conn = sqlite3.connect(db_path)
     cursor = conn.cursor()
     cursor.execute('''CREATE TABLE IF NOT EXISTS users (
@@ -77,35 +86,52 @@ if not os.path.exists(db_path):
     )''')
     conn.commit()
     conn.close()
-    print("✅ chatbot.db created successfully!")
-
-# 🔹 Upload chatbot.db as a private dataset
-api.upload_file(
-    path_or_fileobj=db_path,
-    path_in_repo="chatbot.db",
-    repo_id=repo_id,
-    repo_type="dataset",
-    token=HF_TOKEN
-)
+    
+    print("✅ chatbot.db created successfully.")
 
-print("✅ chatbot.db successfully uploaded to the organization's private dataset.")
+repo_files = api.list_repo_files(repo_id, repo_type="dataset", token=HF_TOKEN)
 
-from huggingface_hub import snapshot_download
-import os
+if "chatbot.db" in repo_files:
+    print("🟢 `chatbot.db` already exists in the dataset. Skipping upload.")
+else:
+    print("📤 Uploading `chatbot.db` to Hugging Face...")
 
-# 🔹 Download chatbot.db securely
-db_folder = snapshot_download(
-    repo_id=repo_id,
-    allow_patterns=["chatbot.db"],  # Only download chatbot.db
-    use_auth_token=HF_TOKEN
-)
+    api.upload_file(
+        path_or_fileobj=db_path,
+        path_in_repo="chatbot.db",
+        repo_id=repo_id,
+        repo_type="dataset",
+        token=HF_TOKEN
+    )
+
+    print("✅ `chatbot.db` successfully uploaded to the organization's private dataset.")
+
+DB_LOCAL_PATH = "/tmp/chatbot.db"  # Path where we save the database
+
+# 🔹 Only download if chatbot.db does not exist locally
+if not os.path.exists(DB_LOCAL_PATH):
+    print("📥 Downloading `chatbot.db` from Hugging Face...")
+
+    try:
+        db_folder = snapshot_download(
+            repo_id=repo_id,
+            repo_type="dataset",  # ✅ Ensure it's a dataset repo
+            allow_patterns=["chatbot.db"],
+            use_auth_token=HF_TOKEN
+        )
+
+        DB_PATH = os.path.join(db_folder, "chatbot.db")
+
+        if os.path.exists(DB_PATH):
+            print(f"✅ Database downloaded at {DB_PATH}")
+        else:
+            raise FileNotFoundError("🚨 Failed to download chatbot.db from the organization's dataset.")
 
-DB_PATH = os.path.join(db_folder, "chatbot.db")
+    except Exception as e:
+        raise FileNotFoundError(f"🚨 Error downloading chatbot.db: {e}")
 
-if os.path.exists(DB_PATH):
-    print(f"✅ Database downloaded at {DB_PATH}")
 else:
-    raise FileNotFoundError("🚨 Failed to download chatbot.db from the organization's dataset.")
+    print(f"🟢 `chatbot.db` already exists at {DB_LOCAL_PATH}, skipping download.")
 
 # ---- Database part ----- #
 # Database Connection