import json
from langchain.prompts import PromptTemplate
from langchain.chains import LLMChain
from pydantic import BaseModel, Field
from typing import List, Tuple
from langchain.chat_models import ChatOpenAI
from langchain.output_parsers import PydanticOutputParser


# Modèles Pydantic pour structurer la réponse
class AlternativeSuggestion(BaseModel):
    suggestion_1: str = Field(..., description="Première suggestion de reformulation.")
    suggestion_2: str = Field(..., description="Deuxième suggestion de reformulation.")
    justification: str = Field(..., description="Justification de la suggestion.")

class PatternDetail(BaseModel):
    pattern: str = Field(..., description="Mot ou expression détectée.")
    indices: List[Tuple[int, int]] = Field(..., description="Indices de début et de fin pour chaque occurrence dans le texte.")
    sentence: str = Field(..., description="Phrase complète où l'expression apparaît.")
    alternative_suggestions: AlternativeSuggestion = Field(..., description="Deux suggestions de reformulation avec justification.")

    def to_dict(self):
        return {
            "pattern": self.pattern,
            "indices": self.indices,
            "sentence": self.sentence,
            "alternative_suggestions": {
                "suggestion_1": self.alternative_suggestions.suggestion_1,
                "suggestion_2": self.alternative_suggestions.suggestion_2,
                "justification": self.alternative_suggestions.justification
            }
        }

class Summary(BaseModel):
    total_patterns_detected: int = Field(..., description="Nombre total de patterns détectés.")
    alternative_suggestions_provided: int = Field(..., description="Nombre total de suggestions fournies.")
    occurrences: int = Field(..., description="Nombre total d'occurrences trouvées dans le texte.")

    def to_dict(self):
        return {
            "total_patterns_detected": self.total_patterns_detected,
            "alternative_suggestions_provided": self.alternative_suggestions_provided,
            "occurrences": self.occurrences
        }

class DetectionResult(BaseModel):
    patterns: List[PatternDetail] = Field(..., description="Liste des patterns détectés avec détails.")
    summary: Summary = Field(..., description="Résumé de la détection.")

    def to_dict(self):
        return {
            "patterns": [pattern.to_dict() for pattern in self.patterns],
            "summary": self.summary.to_dict()
        }

def il_faut_prompt_template():
    return """En tant que rédacteur web expert, votre mission est d'analyser le texte suivant et de détecter toutes les occurrences de l'expression "il faut".

Pour chaque phrase où une ou plusieurs occurrences de l'expression "il faut" sont trouvées, veuillez :
1. Citer la phrase complète avec ses indices de début et de fin.
2. Identifier l'expression exacte utilisée dans la phrase avec ses indices.
3. Expliquer pourquoi l'expression "il faut" peut paraître trop directive ou informelle, et pourquoi elle pourrait alourdir la phrase dans un contexte formel ou professionnel.
4. Proposer cinq reformulations de la phrase, en suivant ces instructions :
   - Ne pas utiliser des expressions ou des mots qui soulignent l'importance de l'action, comme "il est nécessaire de", "il est important de", "il convient de".
   - Utilisez des formulations à l'impératif pour reformuler la phrase de manière plus professionnelle, tout en conservant le sens général.
   - Assurez-vous que la phrase reformulée soit fluide, naturelle et adaptée au contexte, sans redondance ni répétition de mots.
5. Respecter les règles suivantes lors de la reformulation :
   - Ne pas utiliser : "Crucial", "Essentiel", "Important", "Fondamental", "Le monde de", "Dans l’univers de", "À l’ère de", "Généralement", "Outre", "Cela signifie que", "Ce qui signifie que", "Voici", "Découvrez", "S’avérer être", "Il suffit de", "Permet", "Décision éclairée", "Décision avisée" et leurs variantes.
   - Ne pas utiliser "Look" et ses variantes dans un contexte de décoration.
   - Ne pas commencer une phrase par : "Il existe", "Il faut", "Cependant", "De plus", "En revanche", "Néanmoins", "Toutefois", "Imaginez" ou leurs variantes.
   - Ne pas associer : "Joue un rôle", "Il est", "Est une étape" à des mots tels que "Important", "Essentiel", "Crucial", "Primordial", "Fondamental", "Indispensable", "Déterminant".
   - Ne pas commencer une phrase par : "En plus" ou ses variantes.

6. Justifier les reformulations proposées en expliquant comment elles améliorent le style, la clarté ou la fluidité, tout en rendant la phrase plus professionnelle.

Exemples d'erreurs à corriger :
- "Il faut soumettre le rapport avant vendredi", mais on dit "Soumettez le rapport avant vendredi."
- "Il faut organiser une réunion avec l'équipe", mais on dit "Organisez une réunion avec l'équipe."
- "Il faut vérifier les données avant de les publier", mais on dit "Vérifiez les données avant de les publier."

Votre analyse doit être retournée en JSON avec la structure suivante :
{{
    "phrases_detectees": [
    {{
        "phrase": {{
        "texte": "Phrase complète contenant l'expression ou les expressions détectées",
        "indice_debut": x,
        "indice_fin": y
        }},
        "erreurs": [
        {{
            "expression": "Expression détectée",
            "indice_debut": a,
            "indice_fin": b
        }}
        ],
        "reformulations": [
        {{ "texte": "Reformulation 1" }},
        {{ "texte": "Reformulation 2" }},
        {{ "texte": "Reformulation 3" }},
        {{ "texte": "Reformulation 4" }},
        {{ "texte": "Reformulation 5" }}
        ],
        "justification": "Justification des reformulations proposées."
    }}
    ],
    "summary": {{
    "total_erreurs": nombre_total_d_expressions_detectées,
    "total_phrases": nombre_total_de_phrases_contenant_des_expressions,
    "total_reformulations": nombre_total_de_reformulations_proposées
    }}
}}


Texte à analyser :
{text}

Assurez-vous que les indices correspondent aux positions exactes dans le texte original et que le JSON est correctement formaté.
{format_instruction}
"""

# def il_faut_prompt_template():
#     return """
#    Analyse le texte suivant et détecte toutes les occurrences de l'expression "il faut".

#    Pour chaque occurrence trouvée, veuillez :

#    - Citer la phrase complète où l'expression apparaît.
#    - Indiquer l'expression exacte utilisée dans le texte.
#    - Expliquer pourquoi l'expression "il faut" peut paraître trop directive ou informelle dans un contexte formel ou professionnel.
#    - Proposer deux reformulations de la phrase sans utiliser d'expressions ou de mots qui soulignent l'importance, comme "il est nécessaire de", "il est important de", ou "il convient de".
#    - Utilisez des formulations à l'impératif pour rendre la phrase plus directe et fluide.
#    - Assurez-vous que la phrase reformulée est fluide, naturelle, et préserve le sens original.

#    Assurez-vous également de respecter les règles suivantes :
#    1. Les expressions comme "crucial", "essentiel", "important", "fondamental", etc., ne doivent pas être utilisées plus d'une fois dans l'introduction.
#    2. Les expressions comme "Il existe", "il faut", "cependant", "de plus", "en revanche", "néanmoins", "toutefois" ne doivent pas être utilisées en début de phrase.
#    3. Les expressions comme "joue un rôle", "il est", ou "est une étape" ne doivent pas être associées à des adjectifs comme "important", "essentiel", etc.
#    4. "En plus" ne doit pas être utilisé seul en début de phrase.

#    Le format de sortie doit être un JSON structuré avec les éléments suivants pour chaque pattern détecté :

#    - "pattern": "mot ou expression détectée",
#    - "indices": [[index de début, index de fin pour chaque occurrence]],
#    - "sentence": "phrase où l'expression est utilisée",
#    - "alternative_suggestions": ["suggestion 1", "suggestion 2"]

#    Si aucun pattern n'est trouvé, retournez un JSON avec :
#    - "pattern0": "",
#    - "occurrences": 0,

#    À la fin, ajoutez un résumé avec :
#    - "total_patterns_detected": nombre total de patterns détectés,
#    - "alternative_suggestions_provided": nombre de suggestions données,
#    - "occurrences": nombre d'occurrences.

#    Voici le texte à analyser :
#    ```{text}```

#    \n{format_instruction}
#     """

def il_faut(text):
    # Créer le template avec la variable text intégrée directement
    prompt_template = il_faut_prompt_template()

    output_parser = PydanticOutputParser(pydantic_object=DetectionResult)

    # Créer le prompt avec le texte intégré
    gen_prompt_template = PromptTemplate(
        input_variables=["text"],
        template=prompt_template,
        partial_variables={'format_instruction': output_parser.get_format_instructions()}
    )

    # Créer une instance du modèle de langage
    llm = ChatOpenAI(model='gpt-4o', temperature=0.5, api_key="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")

    # Exécuter la chaîne avec le LLM
    llm_chain = gen_prompt_template | llm | output_parser

    # Appel du modèle avec le texte fourni
    result_dict = llm_chain.invoke({"text": text}).to_dict()

    # Parsing de la réponse JSON
    json_result = json.dumps(result_dict, indent=8, ensure_ascii=False)

    # Générer le prompt final
    final_prompt = gen_prompt_template.format(text=text)

    # Retourne les suggestions ou les erreurs de parsing
    return {
        "result": json_result,
        "prompt": final_prompt
    }