#!/bin/bash

# 🚀 Dwrko-M1.0 Quick Upload Script
# HuggingFace पर model upload करने के लिए

echo "🤖 Dwrko-M1.0 HuggingFace Upload Script"
echo "======================================"

# Check if virtual environment exists
if [ ! -d "../dwrko_env" ]; then
    echo "📦 Creating virtual environment..."
    python3 -m venv ../dwrko_env
    
    echo "📥 Installing dependencies..."
    source ../dwrko_env/bin/activate
    pip install huggingface_hub transformers peft
else
    echo "✅ Virtual environment found"
    source ../dwrko_env/bin/activate
fi

# Get user inputs
echo ""
echo "📋 Upload Configuration:"
read -p "🏷️  HuggingFace Username: " USERNAME
read -p "📁 Model Path (default: ./dwrko-m1.0): " MODEL_PATH
MODEL_PATH=${MODEL_PATH:-./dwrko-m1.0}
read -p "🎯 Repository Name (default: Dwrko-M1.0): " REPO_NAME
REPO_NAME=${REPO_NAME:-Dwrko-M1.0}

echo ""
echo "🔍 Checking model files..."

# Check if model directory exists
if [ ! -d "$MODEL_PATH" ]; then
    echo "❌ Error: Model directory not found: $MODEL_PATH"
    echo "💡 Please train your model first using: python train.py"
    exit 1
fi

# Check required files
REQUIRED_FILES=("adapter_config.json" "adapter_model.safetensors")
for file in "${REQUIRED_FILES[@]}"; do
    if [ ! -f "$MODEL_PATH/$file" ]; then
        echo "❌ Missing required file: $file"
        echo "💡 Please complete training first"
        exit 1
    fi
done

echo "✅ All required files found"

# Login to HuggingFace
echo ""
echo "🔑 Logging into HuggingFace..."
echo "💡 You'll need to paste your HuggingFace token"
echo "   Get it from: https://huggingface.co/settings/tokens"

huggingface-cli login

if [ $? -ne 0 ]; then
    echo "❌ Login failed. Please check your token."
    exit 1
fi

echo "✅ Login successful"

# Upload model
echo ""
echo "🚀 Uploading $REPO_NAME to HuggingFace..."
echo "📁 From: $MODEL_PATH"
echo "🎯 To: $USERNAME/$REPO_NAME"

# Use Python script for upload
python upload_to_hf.py \
    --model_path "$MODEL_PATH" \
    --username "$USERNAME" \
    --repo_name "$REPO_NAME"

if [ $? -eq 0 ]; then
    echo ""
    echo "🎉 SUCCESS! Your model is now live!"
    echo "🔗 URL: https://huggingface.co/$USERNAME/$REPO_NAME"
    echo ""
    echo "📤 Share your model:"
    echo "   • Social media पर post करें"
    echo "   • Community में share करें"
    echo "   • README में badge add करें"
    echo ""
    echo "Badge:"
    echo "[![HuggingFace](https://img.shields.io/badge/🤗%20HuggingFace-$REPO_NAME-blue)](https://huggingface.co/$USERNAME/$REPO_NAME)"
else
    echo "❌ Upload failed. Check the error messages above."
    exit 1
fi