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

tldr_prompt_template = """
Create a mind map of the given research paper along the given lines:
what is {title}?
what this paper brings to the research community including how it is different from existing work?
what novel ideas/techniques are presented in this paper?

The above sections may differ from paper to paper, hence you may need to adjust the structure accordingly by dropping / merging one or more sections.

Here is the research paper abstract: ####{paper}####

Ensure that the outline is structured in Markdown format for clarity, facilitating its integration into documents or presentations.
"""


tldr_output_parser = StrOutputParser()
tldr_prompt = PromptTemplate(
    template=tldr_prompt_template,
    input_variables=["title", "paper"],
)
tldr_chain = lambda model: tldr_prompt | model | tldr_output_parser