import json
import os

def lambda_handler(event, context):
    try:
        body = json.loads(event['body'])
        prompt = body.get('prompt', 'Tell me a story about serverless.')

        # Đây là phần gọi API GenAI thực tế, ví dụ với OpenAI
        # from openai import OpenAI
        # client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
        # response = client.chat.completions.create(
        #     model="gpt-3.5-turbo",
        #     messages=[{"role": "user", "content": prompt}]
        # )
        # generated_text = response.choices[0].message.content
        
        # Phần mock response cho ví dụ
        generated_text = f"Serverless AI generated: '{prompt}' - This is a placeholder response."
        
        return {
            'statusCode': 200,
            'body': json.dumps({'generatedText': generated_text})
        }
    except Exception as e:
        return {
            'statusCode': 500,
            'body': json.dumps({'error': str(e)})
        }