FerdinandPyCode commited on
Commit
1f35e76
·
verified ·
1 Parent(s): c6cc554

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -3
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
- output_parser = PydanticOutputParser(pydantic_object=StyleErrorResult)
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= gen_prompt_template | llm | output_parser
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.run(text=text)
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