import boto3
import botocore.config
import json
from datetime import datetime

def blog_generate_using_bedrock(blogtopic: str) -> str:
    prompt = f"""Write a blog minimum 300 to 500 words (no need to strictly follow) in a professional manner with only 5 subtopics and a conclusion about {blogtopic}"""
    body = {
        "prompt": prompt,
        "max_gen_len": 600,
        "temperature": 0.5,
        "top_p": 0.9
    }
    try:
        bedrock = boto3.client("bedrock-runtime", region_name="ap-south-1",
                              config=botocore.config.Config(read_timeout=300, retries={'max_attempts': 3}))
        response = bedrock.invoke_model(body=json.dumps(body), modelId="<Your_Model_ID>")
        response_content = response.get('body').read()
        response_data = json.loads(response_content)
        print(response_data)
        blog_details = response_data['generation']
        return blog_details
    except Exception as e:
        print(f"Error generating the blog: {e}")
        return ""

def save_blog_details_s3(s3_key, s3_bucket, generate_blog):
    s3 = boto3.client('s3')
    try:
        s3.put_object(Bucket=s3_bucket, Key=s3_key, Body=generate_blog)
        print("Code saved to S3")
    except Exception as e:
        print("Error when saving the code to S3")

def lambda_handler(event, context):
    event = json.loads(event['body'])
    blogtopic = event['blog_topic']
    generate_blog = blog_generate_using_bedrock(blogtopic=blogtopic)
    if generate_blog:
        curr_time = datetime.now().strftime('%H%M%S')
        s3_key = f"blog-output/{curr_time}.txt"
        s3_bucket = '<Your_s3_bucket>'
        save_blog_details_s3(s3_key, s3_bucket, generate_blog)
    else:
        print("No blog was generated")
    return {
        'statusCode': 200,
        'body': json.dumps('Blog Generation is Completed')
    }

s3_key = f"blog-output/{curr_time}.txt"
s3_bucket = 'YOUR_BUCKET_NAME'
s3.put_object(Bucket=s3_bucket, Key=s3_key, Body=generated_blog)