import os
import time
import subprocess
from huggingface_hub import HfApi

# Initialize Hugging Face API
api = HfApi()

# Define your Hugging Face repo ID
repo_id = "mikahniehaus/YelpPredictorDeploy"  # Change this if needed

# Allowed file extensions
ALLOWED_EXTENSIONS = {".py", ".json", ".csv", ".pth", ".txt", ".md"}

# Directories to ignore (including `.env`, virtual environments, and `node_modules`)
IGNORE_DIRS = {".git", "__pycache__", ".venv", "venv", "env", ".env", "hf_env", "node_modules"}

# Get the current branch name
def get_current_branch():
    try:
        branch = subprocess.check_output(["git", "rev-parse", "--abbrev-ref", "HEAD"], text=True).strip()
        return branch
    except subprocess.CalledProcessError:
        print("⚠️ Warning: Not in a Git repository.")
        return None

# Get a list of changed files in the current branch
def get_changed_files():
    try:
        changed_files = subprocess.check_output(["git", "diff", "--name-only", "origin/main"], text=True).splitlines()
        return set(changed_files)
    except subprocess.CalledProcessError:
        print("⚠️ Warning: Could not get changed files.")
        return set()

# Get a list of files already in the repository
def get_repo_files():
    try:
        return set(api.list_repo_files(repo_id=repo_id))
    except Exception as e:
        print(f"⚠️ Failed to list repo files: {e}")
        return set()  # Return an empty set on failure

# Recursively find all valid files to upload
def get_all_files(directory="."):
    file_list = []
    for root, dirs, files in os.walk(directory):
        # Skip any directory that matches IGNORE_DIRS
        if any(ignored in root.split(os.sep) for ignored in IGNORE_DIRS):
            continue  # Skip the entire directory

        for file in files:
            file_path = os.path.join(root, file)
            file_extension = os.path.splitext(file)[1]

            if file_extension in ALLOWED_EXTENSIONS:
                repo_path = os.path.relpath(file_path, directory).replace("\\", "/")
                file_list.append((file_path, repo_path))
    
    return file_list

# Get repo file list to skip already uploaded files
repo_files = get_repo_files()

# Get the current Git branch
current_branch = get_current_branch()
if current_branch:
    print(f"📂 Current Git branch: {current_branch}")

# Get the list of changed files
changed_files = get_changed_files()

# Get local files to upload (excluding ignored directories)
files_to_upload = get_all_files()

# Upload only files that are changed or not yet in the repo
for file_path, repo_path in files_to_upload:
    if repo_path in repo_files and repo_path not in changed_files:
        print(f"⏭️ Skipping {file_path}, already uploaded and unchanged.")
        continue  # Skip if file is already uploaded and not changed

    while True:  # Infinite retries
        try:
            print(f"🚀 Uploading {file_path} to {repo_path} in {repo_id} (branch: {current_branch})...")
            api.upload_file(
                path_or_fileobj=file_path,
                path_in_repo=repo_path,
                repo_id=repo_id,
                repo_type="model"
            )
            print(f"✅ Successfully uploaded {file_path}")
            time.sleep(5)  # Wait to avoid rate limits
            break  # Exit loop if upload is successful

        except Exception as e:
            print(f"❌ Error uploading {file_path}: {e}")
            print("⏳ Waiting 1 hour before retrying...")
            time.sleep(3600)  # Wait 1 hour before retrying
            
            #remove exesive history
#git checkout --orphan latest_branch
#git add -A
#git commit -m "Initial commit with latest changes"
#git branch -D main  # Delete the old "main" branch (or replace with your branch name)
#git branch -m main  # Rename the new branch to "main"
#git push --force origin main