update prompts et interface sans cadre refs

app.py

@@ -3,25 +3,24 @@ import streamlit as st
 import os
 from mistralai import Mistral
  
-MISTRAL_API_KEY = os.getenv("api_mistral")
+MISTRAL_API_KEY = 'DFtRafUrJnASg1cqrEWNtSMKAjZT4Njp' #os.getenv("api_mistral")
 model = 'mistral-large-latest'
 mistral_client = Mistral(api_key=MISTRAL_API_KEY)
 MAX_TOKENS = 1500
 #endregion
  
 #region# Définition des prompts
-def generate_prompts(score:str, type: str, annee_min: str, annee_max:str ):
+def generate_prompts(score:str, type: str, annee_min: str, annee_max:str ) -> dict:
     """
     Genere les prefixes et suffixes des prompts pour Mistral en fonction du score de vulgarisation, du type d'espece, et les années des documents
- 
     Args:
         score (str): 1 = vulgarisé, 2 = intermédiaire, 3 = technique
-        type (str): 'ponte' ou 'chair'
+        type (str): 'Ponte' ou 'Chair'
         annee_min (str): annee min de publication
         annee_max (str): annee max de publication
     
     """
- 
+
     if type == "Ponte":
         type_description = "volailles pondeuses"
     elif type == "Chair":
@@ -30,36 +29,41 @@ def generate_prompts(score:str, type: str, annee_min: str, annee_max:str ):
         raise ValueError("Type must be either 'Ponte' or 'Chair'")
  
     if score == "1":
-        prefix_prompt = f"""Tu es un assistant IA spécialisé en nutrition de la volaille. Ton utilisateur est un chercheur travaillant sur l'amélioration des régimes 
-        alimentaires pour optimiser la santé et la croissance des {type_description}. 
-        Réponds en fournissant des explications claires et simples. N'oublie pas de citer à la fin de ta réponse les références sur 
-        lesquelles tu t'es basé avec son année (entre {annee_min} et {annee_max})."""
-        suffix_prompt = """Réponds en français et donne une réponse directe et claire, donne les références de façon bibliographique. Sépare les explications et les reférences par ces 5 caractères suivants : µµµµµ"""
+        prefix_prompt = f"""Tu es un assistant IA spécialisé en nutrition de la volaille. Ton utilisateur est un chercheur travaillant sur
+        l'amélioration des régimes alimentaires pour optimiser la santé et la croissance des {type_description}.
+        Réponds en fournissant en vulgarisant les informations.
+        Pour fournir la réponse, tu dois te baser sur des publications/articles qui ont une date de publication entre {annee_min} et {annee_max}."""
+        suffix_prompt = """Réponds en français et donne une réponse directe et claire.
+        Fini par faire une bibliographie avec les références bibliographiquesque tu as utilisé."""
     elif score == "2":
-        prefix_prompt = f"""Tu es un assistant IA spécialisé en nutrition de la volaille. Ton utilisateur est un chercheur travaillant sur l'amélioration des régimes 
-        alimentaires pour optimiser la santé et la croissance des {type_description}. 
-        Réponds en fournissant des explications détaillées et précises, adaptées à la complexité de la question posée. 
-        N'oublie pas de citer à la fin de ta réponse les références sur lesquelles tu t'es basé avec son année (entre {annee_min} et {annee_max}). Sois concis et clair."""
-        suffix_prompt = """Réponds en français et en suivant cette structure :
-        Donne une explication scientifique détaillée (mécanismes biologiques, données chiffrées).
-        Fini par les études ou références bibliographiques si disponibles. Sépare les explications et les reférences par ces 5 caractères suivants : µµµµµ"""
+        prefix_prompt = f"""Tu es un assistant IA spécialisé en nutrition de la volaille. Ton utilisateur est un chercheur travaillant sur
+        l'amélioration des régimes alimentaires pour optimiser la santé et la croissance des {type_description}.
+        Réponds en fournissant des explications claires et concises, adaptées à la question posée.
+        Pour fournir la réponse, tu dois te baser sur des publications/articles qui ont une date de publication entre {annee_min} et {annee_max}.
+        Tes réponses doivent être structurées, complètes et adaptées aux professionnels du secteur."""
+        suffix_prompt = """Présente une réponse claire et concise, en commençant par une explication des causes du problème,
+        suivie d'une analyse des facteurs de risque, et termine par des recommandations. Utilise un ton
+        professionnel, clair et rigoureux. Réponds en français.
+        Fini par faire une bibliographie avec les références bibliographiques que tu as utilisé."""
     elif score == "3":
-        prefix_prompt = f"""Tu es un assistant IA spécialisé en nutrition de la volaille. Ton utilisateur est un chercheur travaillant sur l'amélioration des régimes 
-        alimentaires pour optimiser la santé et la croissance des {type_description}. 
+        prefix_prompt = f"""Tu es un assistant IA spécialisé en nutrition de la volaille. Ton utilisateur est un chercheur travaillant sur
+        l'amélioration des régimes alimentaires pour optimiser la santé et la croissance des {type_description}.
         Réponds en fournissant des explications détaillées et précises, adaptées à la complexité de la question posée.
         N'oublie pas de citer à la fin de ta réponse les références sur lesquelles tu t'es basé avec son année (entre {annee_min} et {annee_max}).
-        Inclus des formules, des valeurs nutritionnelles précises et des références aux normes actuelles. Sois concis et clair."""
-        suffix_prompt = """Réponds en français et en suivant cette structure : Explication scientifique détaillée (mécanismes biologiques, données chiffrées). 
-        Études ou références bibliographiques si disponibles. Sépare les explications et les reférences par ces 5 caractères suivants : µµµµµ"""
+        Tes réponses doivent être structurées, complètes et adaptées aux professionnels du secteur."""
+        suffix_prompt = """Présente une réponse détaillée et complète, en commençant par une explication des causes du problème,
+        suivie d'une analyse des facteurs de risque, et termine par des recommandations pratiques et éprouvées. Utilise un ton
+        professionnel, clair et rigoureux. Si possible, inclue des chiffres, des études ou des références pertinentes pour renforcer
+        la crédibilité de la réponse. Réponds en français.
+        Fini par faire une bibliographie avec les références bibliographiques que tu as utilisé."""
     else:
         raise ValueError("Score must be 1, 2, or 3")
  
     return prefix_prompt, suffix_prompt
- 
-def send_prompt_to_mistral(type_reponse: str, user_prompt: str, temperature:int, n_comp:int, prefix_prompt: str, suffix_prompt:str, verbose=True):
+
+def send_prompt_to_mistral(type_reponse: str, user_prompt: str, temperature:int, n_comp:int, prefix_prompt: str, suffix_prompt:str, verbose=True) -> str:
     """
     Envoie un prompt à Mistral pour obtenir une réponse
- 
     Args:
         type_reponse (str): Le rôle de l'utilisateur, peut être 'technicien' ou 'chercheur'.
                     Si le rôle ne correspond pas à l'un de ces deux, une exception sera levée.
@@ -68,14 +72,13 @@ def send_prompt_to_mistral(type_reponse: str, user_prompt: str, temperature:int,
         verbose (bool): Print la reponse avant le return
         prefixe (str): Prefixe du prompt
         suffixe (str): Suffixe du prompt
- 
     Returns:
         dict: La réponse du modèle Mistral à partir du prompt fourni.
- 
     Raises:
         ValueError: Si le rôle spécifié n'est pas 'technicien' ou 'chercheur'.
     """
- 
+
+
     # Création du message à envoyer à Mistral
     messages = [{"role": type_reponse, "content": suffix_prompt}]
     
@@ -86,7 +89,7 @@ def send_prompt_to_mistral(type_reponse: str, user_prompt: str, temperature:int,
             {"role": "system", "content": prefix_prompt},
             {"role": "user", "content": user_prompt + suffix_prompt},
         ],
- 
+
         #response_format={
         #    'type': 'json_object'
         #},
@@ -96,12 +99,12 @@ def send_prompt_to_mistral(type_reponse: str, user_prompt: str, temperature:int,
         stream=False
         #stop='\n'
     )
- 
+
     if verbose:
         print(chat_response)
- 
+
     return chat_response
- 
+
 def is_valid_mistral_response(response: dict) -> bool:
     """
     Vérifie si la réponse de l'API Mistral est valide.
@@ -142,33 +145,60 @@ def is_valid_mistral_response(response: dict) -> bool:
         return False
     
     return True
- 
+
 def print_pretty_response(response: dict, verbose=True):
- 
+
     pretty = response.choices[0].message.content
- 
+
     if verbose:
         print(pretty)
     
     return pretty
- 
+
 def response_details(response, verbose=True):
     """
     Envoie les details techniques du prompt
     """
- 
+
     details = {}
     details["id"] = response.id
     details["total_tokens"] = response.usage.total_tokens
     details["prefix"] = response.choices[0].message.prefix
     details["role"] = response.choices[0].message.role
- 
+
     if verbose:
         print(details)
- 
+
     return details
- 
-def prompt_pipeline(user_prompt: str, niveau_detail: str, type_reponse: str, souche: str, annee_publication_min: str, annee_publication_max: str):
+
+def prompt_pipeline(user_prompt: str, niveau_detail: str, type_reponse: str, souche: str, annee_publication_min: str, annee_publication_max: str) -> dict:
+    """
+    Fonction visible de l'application pour appeler un prompt et obtenir sa reponse
+    Args: 
+        prompt (str): Prompt utilisateur
+        niveau_detail (str): Niveau de detail de la requete : 1, 2, 3. Plus haut = plus d'infos
+        type_reponse (str): 'Ponte', 'Chair'
+    Returns:
+        Dict
+    """
+
+    prefix_prompt, suffix_prompt = generate_prompts(score=niveau_detail, type=type_reponse, annee_min=annee_publication_min, annee_max=annee_publication_max)
+
+    reponse_mistral = send_prompt_to_mistral(
+        type_reponse=type_reponse,
+        user_prompt=user_prompt,
+        temperature=0.10,
+        n_comp=1,
+        verbose=False,
+        prefix_prompt=prefix_prompt,
+        suffix_prompt=suffix_prompt
+    )
+
+    to_return = {}
+    to_return["reponse_propre"] = print_pretty_response(reponse_mistral, verbose=True)
+    to_return["details"] = response_details(reponse_mistral, verbose=False)
+
+    return to_return
     """
     Fonction visible de l'application pour appeler un prompt et obtenir sa reponse
  
@@ -241,7 +271,7 @@ if st.button("Envoyer la question..."):
     )
             print(response0["reponse_propre"])
             response = response0["reponse_propre"]
-            bot_response = response.split('µµµµµ')
+            bot_response = response
             
             # st.markdown("**Bot :** \\t " + bot_response)
             # Afficher un titre
@@ -250,19 +280,20 @@ if st.button("Envoyer la question..."):
             # Ajouter du texte Markdown avec un cadre
             st.markdown(f"""
             <div style="border: 2px solid #453103; padding: 15px; border-radius: 10px;">
-                {bot_response[0]}
+                {bot_response}
             </div>
             """, unsafe_allow_html=True)
+
+            #encadré sources
+            # # Afficher un titre
+            # st.subheader("Sources :")
  
-            # Afficher un titre
-            st.subheader("Sources :")
- 
-            # Ajouter du texte Markdown avec un cadre
-            st.markdown(f"""
-            <div style="border: 2px solid #453103; padding: 15px; border-radius: 10px;">
-            {bot_response[1]}
-            </div>
-            """, unsafe_allow_html=True)
+            # # Ajouter du texte Markdown avec un cadre
+            # st.markdown(f"""
+            # <div style="border: 2px solid #453103; padding: 15px; border-radius: 10px;">
+            # {bot_response[1]}
+            # </div>
+            # """, unsafe_allow_html=True)
 
             #encadré Reviews
             # st.markdown("""
@@ -278,28 +309,4 @@ if st.button("Envoyer la question..."):
             st.warning("Veuillez compléter les paramètres dans le bandeau latéral de gauche!")
 #endregion
  
-# choix_prod, choix_vulgarisation, choix_annee
- 
-#region# Markdown
-# st.markdown("""
-#     <style>
-#         p {
-#             color: #555555; 
-#         }
- 
-#         .st-bx {
-#             background-color: #f1f1f1;
-#             border-radius: 10px;
-#             padding: 10px;
-#             margin: 10px 0;
-#         }
-#         .st-bx h2 {
-#             color: #4CAF50;
-#         }
-#         .st-bx p {
-#             color: #555;
-#         }
-#     </style>
-# """, unsafe_allow_html=True)
- 
-#endregion
+