from flask import Flask, request, jsonify
import os
import requests

app = Flask(__name__)

# Store your Kling API key in HF "Secrets" settings
KLING_API_KEY = os.getenv("KLING_API_KEY")

@app.route("/generate", methods=["POST"])
def generate():
    data = request.get_json()
    prompt = data.get("prompt", "")
    
    try:
        response = requests.post(
            "https://api.kling.ai/v1/images",  # Replace with actual Kling endpoint
            headers={
                "Authorization": f"Bearer {KLING_API_KEY}",
                "Content-Type": "application/json"
            },
            json={
                "model": "your-model-id",  # Put actual Kling model ID
                "prompt": prompt,
                "size": "1024x1024"
            }
        )
        return jsonify(response.json()), response.status_code
    except Exception as e:
        return jsonify({"error": str(e)}), 500

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860)