from langchain_core.prompts import PromptTemplate
from langchain_core.output_parsers import StrOutputParser

insights_bullet_prompt_template = """
Draw the key insights about objective, method, results and conclusions from the given excerpt in the form of bullet points. Also mention the figure or tables referred to along-with the corresponding bullet points
Note: if results and conclusions are not much different, feel free to combine them to avoid duplication of information
excerpt: {paper}
"""

insights_bullet_output_parser = StrOutputParser()
insights_bullet_prompt = PromptTemplate(
    template=insights_bullet_prompt_template,
    input_variables=["paper"],
)
insights_bullet_chain = (
    lambda model: insights_bullet_prompt | model | insights_bullet_output_parser
)