from flask import Flask, request, jsonify
from flask_cors import CORS
import logging
import os

app = Flask(__name__)
CORS(app)

# Configure logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

@app.route('/debug', methods=['GET'])
def debug():
    # Check if HF_TOKEN is accessible
    hf_token = os.environ.get('HF_TOKEN')
    all_env_vars = dict(os.environ)
    
    return jsonify({
        "HF_TOKEN_exists": hf_token is not None,
        "HF_TOKEN_length": len(hf_token) if hf_token else 0,
        "HF_TOKEN_start": hf_token[:10] + "..." if hf_token and len(hf_token) > 10 else hf_token,
        "all_env_vars": {k: v for k, v in all_env_vars.items() if 'TOKEN' in k or 'KEY' in k or 'SECRET' in k}
    })

@app.route('/generate_script', methods=['POST'])
def generate_script():
    try:
        # First, let's just check the token and return a mock response
        hf_token = os.environ.get('HF_TOKEN')
        
        if not hf_token:
            return jsonify({
                "error": "HF_TOKEN not found in environment variables",
                "message": "Please set HF_TOKEN in your Space settings"
            }), 500
        
        data = request.json
        topics = data.get('topics', [])
        
        if not topics:
            return jsonify({"error": "No topics provided"}), 400
        
        # Return mock response for now
        mock_script = f"""🎯 GOLDEN HOOK: "Discover how {topics[0]} is changing everything!"
        
📝 VALUABLE CONTENT:
- Scene 1: Show stunning visuals of {topics[0]} in action
- Scene 2: Explain 3 key benefits in simple terms
- Scene 3: Share real-world examples and success stories

🎬 CTA: "Like and follow for more insights on {topics[0]}!"

⏰ Duration: Exactly 60 seconds
🎤 Voiceover: Energetic and engaging tone
📱 Optimized for: TikTok/YouTube Shorts"""

        return jsonify({
            "topic": topics[0],
            "script": mock_script,
            "full_topics": topics,
            "status": "success",
            "token_status": "found",
            "token_length": len(hf_token)
        })
        
    except Exception as e:
        logger.error(f"Error: {str(e)}")
        return jsonify({"error": str(e)}), 500

@app.route('/health', methods=['GET'])
def health():
    return jsonify({"status": "healthy", "message": "Server is running"})

if __name__ == '__main__':
    port = int(os.environ.get('PORT', 7860))
    app.run(host='0.0.0.0', port=port, debug=False)