yukee1992's picture
Update app.py
c3b0435 verified
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)