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

temp = """
Je te fournirai des données pertinentes. Ton objectif sera de générer un bref résumé de ces données et de proposer une légère analyse qui mettra en évidence certains points clés.

Suis ces étapes pour accomplir ta tâche:

Examine le fichier et identifie la nature des données.
Rédige un court paragraphe présentant les principales tendances et idées découlant des données.
Identifie quelques faits marquants ou statistiques importantes susceptibles d'intéresser l'audience.
Propose des pistes d'analyse complémentaire pouvant être développées ultérieurement.
Les caractéristiques de l'output souhaité sont les suivantes:

Un texte concis et clair, présentant les grandes lignes des données du fichier.
Une sélection des aspects les plus intrigants ou significatifs de ces données.
Des suggestions pour creuser davantage certaines observations et interprétations.
Si tout semble correct selon toi, tu peux maintenant démarrer!

Human : {input}
AI : """

pro= PromptTemplate(input_variables=['input'], template = temp)
summarize = pro | llm | StrOutputParser()