from langchain.prompts import PromptTemplate
from langchain.chains import LLMChain
from langchain_openai import ChatOpenAI
from utils.encrypt import Decryptor
import os
from dotenv import load_dotenv
prompt_template = PromptTemplate.from_template("""
You're an expert startup evaluator.

Your task is to check whether the following input is a valid startup idea or not .
respond with "yes" if it is a valid startup idea, otherwise respond with "no" .
Input contains all these details:
Idea name:{name}
problem statement:{problem_statement}
solution:{solution}
target market:{target_market}
business model:{business_model}

only respond with "yes" or "no", nothing else.
""")
load_dotenv()
# validator_chain = prompt_template | llm

Decryptor=Decryptor()
class ValidatorAgent:   
    def __init__(self, api):
        self.llm = ChatOpenAI(
            model=api["model_name"], 
            temperature=api["temperature"],
            openai_api_key=Decryptor.decrypt_api_key(api["apikey"]),
            openai_api_base=api["provider_url"],
        )
        self.validator_chain = prompt_template | self.llm
    def validate_idea(self, idea: dict):
        return self.validator_chain.invoke({
            "name": idea.get("name", "not available"),
            "problem_statement": idea.get("problem_statement", "not available"),
            "solution": idea.get("solution", "not available"),
            "target_market": idea.get("target_market", "not available"),
            "business_model": idea.get("business_model", "not available")
        }).content.strip().lower()