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
|