from openai import OpenAI
from dotenv import load_dotenv
import os

# Load environment variables
load_dotenv()
key = os.getenv("OPENAI_API_KEY")

# Initialize the OpenAI client
client = OpenAI(api_key=key)

def generate_paraphrase(sentences, model="gpt-4", num_paraphrases=10, max_tokens=150, temperature=0.7):
    """Generate paraphrased sentences using the OpenAI GPT-4 model."""
    
    # Ensure sentences is a list
    if isinstance(sentences, str):
        sentences = [sentences]
    
    paraphrased_sentences_list = []

    for sentence in sentences:
        full_prompt = f"Paraphrase the following text: '{sentence}'"
        
        try:
            chat_completion = client.chat.completions.create(
                messages=[{"role": "user", "content": full_prompt}],
                model=model,
                max_tokens=max_tokens,
                temperature=temperature,
                n=num_paraphrases  # Number of paraphrased sentences to generate
            )
            # Extract paraphrased sentences
            paraphrased_sentences = [choice.message.content.strip() for choice in chat_completion.choices]
            paraphrased_sentences_list.extend(paraphrased_sentences)
        except Exception as e:
            print(f"Error paraphrasing sentence '{sentence}': {e}")

    return paraphrased_sentences_list

# Example usage
result = generate_paraphrase(
    "Mayor Eric Adams did not attend the first candidate forum for the New York City mayoral race, but his record — and the criminal charges he faces — received plenty of attention on Saturday from the Democrats who are running to unseat him."
)

print(f"Number of paraphrases generated: {len(result)}")