import anthropic
import os
from dotenv import load_dotenv
import json
from src.prompt import general_summary_prompt,custom_template_prompt
load_dotenv()

def general_summary(transcription):
    client = anthropic.Anthropic(
        api_key=os.getenv("ANTHTROPIC_API_KEY"),
    )
    message = client.messages.create(
        model="claude-3-5-sonnet-20241022",
        max_tokens=4000,
        messages=[
            {"role": "user", "content": f"{general_summary_prompt}"},
            {"role": "user", "content": f"Transcription: {transcription}"}
        ]
    )

    text = message.content[0].text
    return json.loads(text)




def custom_summary(template, transcription):
    client = anthropic.Anthropic(
        api_key=os.getenv("ANTHTROPIC_API_KEY"),
    )
    message = client.messages.create(
        model="claude-3-5-sonnet-20241022",
        max_tokens=8000,
        messages=[
            {"role": "user", "content": f"{custom_template_prompt}"},
            {"role": "user", "content": f"TEMPLATE : {template}"},
            {"role": "user", "content": f"Transcription: {transcription}"}
        ]
    )

    text = message.content[0].text
    return json.loads(text)