import openai
import time
import os

# Set your OpenAI API key 
openai.api_key = os.getenv("OPENAI_API_KEY")

def generate_response_from_document(query, retrieved_docs):
    if not retrieved_docs:  # Check if no documents were retrieved
        return "I cannot answer the question due to insufficient information in the documents."

    # context = " ".join([doc.page_content for doc in retrieved_docs]) # Now iterates over Document objects
    context = " ".join([doc for doc in retrieved_docs])
    prompt = """
        "You are an accurate and reliable AI assistant that can answer questions with the help of external documents.
         Please note that external documents may contain noisy or factually incorrect information. 
         If the information in the document contains the correct answer, you will give an accurate answer. 
         If the information in the document does not contain the answer, you will generate ’I can not answer the question because of the insufficient information in documents.‘. 
         If there are inconsistencies with the facts in some of the documents, please generate the response 'There are factual errors in the provided documents.' and provide the correct answer."

         Context or Document: {context}
         Query: {query}
    """
    # prompt = (
    #         "You are a highly intelligent assistant tasked with answering a question based strictly on the provided context. "
    #         f"Given Question: {query} \n\n"
    #         f"Context: {context} \n"
    #         f"Answer the question directly and concisely using only the information available in the context."
    #          "Do not include any other information which is not there in the context."
    #     )

    try:
        response = openai.chat.completions.create( # Use the new chat completions API
            model= "gpt-3.5-turbo", #"gpt-4", #"gpt-3.5-turbo"   Or use another suitable model like gpt-4
            messages=[{"role": "user", "content": prompt}],
            max_tokens=300,
            temperature=0.7,
        )
        return response.choices[0].message.content.strip() # Extract content from message
    except Exception as e:
        return f"Error generating response: {str(e)}"