#!/bin/bash

echo "🔧 Fixing dataset loading issue..."

# Copy the fixed app
if [ -f "app_fixed.py" ]; then
    cp app_fixed.py app.py
    echo "✅ Updated app.py with fixed dataset loading"
else
    echo "❌ app_fixed.py not found"
    exit 1
fi

# Add and commit changes
echo "📝 Committing dataset loading fix..."
git add app.py
git commit -m "Fix dataset loading - handle multiple data directory paths

- Try multiple possible data directory locations
- Add better error handling for JSON parsing
- Improve logging for dataset loading
- Handle different working directories on HF Spaces"

# Push to Hugging Face Space
echo "🚀 Pushing fix to Hugging Face Space..."
git push hf lora-training:main --force

echo "✅ Dataset loading fix complete!"
echo ""
echo "📋 Next steps:"
echo "1. Check your HF Space build status: https://huggingface.co/spaces/harismlnaslm/Textilindo"
echo "2. Test the /stats endpoint to verify dataset is loaded"
echo "3. Try the /chat endpoint with a question"