Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -45,6 +45,10 @@ class StyleErrorResult(BaseModel):
|
|
45 |
}
|
46 |
|
47 |
# Création du prompt et du modèle LLMChain pour analyser les fautes de style
|
|
|
|
|
|
|
|
|
48 |
prompt_template = """
|
49 |
Tu es un détecteur de texte AI très précis. Je vais te donner un texte, et ta mission est de repérer des fautes de style.
|
50 |
Cela inclut les répétitions, les tournures de phrases maladroites, et les mauvais choix de mots.
|
@@ -56,24 +60,27 @@ Pour chaque erreur détectée, tu dois fournir les informations suivantes :
|
|
56 |
|
57 |
Le texte à analyser est le suivant :
|
58 |
```{text}```
|
|
|
|
|
|
|
59 |
"""
|
60 |
|
61 |
# Utilisation de LangChain pour la détection des erreurs
|
62 |
def get_llm_chain():
|
63 |
-
|
64 |
prompt = PromptTemplate(input_variables=["text"], template=prompt_template)
|
65 |
# Remplacez `llm` par votre modèle de langage préféré, tel que OpenAI GPT ou autre
|
66 |
# llm = None # Remplacez ceci par votre modèle LLM
|
67 |
# llm_chain = LLMChain(llm=llm, prompt=prompt, output_parser=output_parser)
|
68 |
llm = ChatOpenAI(model='gpt-4o',temperature=0.5, api_key="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", verbose = True)
|
69 |
-
llm_chain=
|
70 |
|
71 |
return llm_chain
|
72 |
|
73 |
# Fonction pour analyser les fautes de style
|
74 |
def analyze_style_errors(text: str) -> dict:
|
75 |
llm_chain = get_llm_chain()
|
76 |
-
result = llm_chain.
|
77 |
return result.to_dict()
|
78 |
|
79 |
# Interface Streamlit
|
|
|
45 |
}
|
46 |
|
47 |
# Création du prompt et du modèle LLMChain pour analyser les fautes de style
|
48 |
+
|
49 |
+
output_parser = PydanticOutputParser(pydantic_object=StyleErrorResult)
|
50 |
+
|
51 |
+
|
52 |
prompt_template = """
|
53 |
Tu es un détecteur de texte AI très précis. Je vais te donner un texte, et ta mission est de repérer des fautes de style.
|
54 |
Cela inclut les répétitions, les tournures de phrases maladroites, et les mauvais choix de mots.
|
|
|
60 |
|
61 |
Le texte à analyser est le suivant :
|
62 |
```{text}```
|
63 |
+
|
64 |
+
Voici le formattage de la sortie :
|
65 |
+
```{output_parser}```
|
66 |
"""
|
67 |
|
68 |
# Utilisation de LangChain pour la détection des erreurs
|
69 |
def get_llm_chain():
|
70 |
+
|
71 |
prompt = PromptTemplate(input_variables=["text"], template=prompt_template)
|
72 |
# Remplacez `llm` par votre modèle de langage préféré, tel que OpenAI GPT ou autre
|
73 |
# llm = None # Remplacez ceci par votre modèle LLM
|
74 |
# llm_chain = LLMChain(llm=llm, prompt=prompt, output_parser=output_parser)
|
75 |
llm = ChatOpenAI(model='gpt-4o',temperature=0.5, api_key="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", verbose = True)
|
76 |
+
llm_chain= prompt_template | llm | output_parser
|
77 |
|
78 |
return llm_chain
|
79 |
|
80 |
# Fonction pour analyser les fautes de style
|
81 |
def analyze_style_errors(text: str) -> dict:
|
82 |
llm_chain = get_llm_chain()
|
83 |
+
result = llm_chain.invoke({'text': text})
|
84 |
return result.to_dict()
|
85 |
|
86 |
# Interface Streamlit
|