maj forme

app.py

@@ -1,25 +1,28 @@
+#region# import libs
 import streamlit as st
 from transformers import pipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
+#endregion
 
-# Titre de l'application
+#region# Titre de l'application et mise en page
 st.set_page_config(page_title="Expert nutrition volaille", page_icon="🤖")
 st.title("Chatbot AI avec l'expert nutrition")
+st.sidebar.image("img/avril_logo_rvb.jpg")
+st.sidebar.header("")
+#endregion
 
-st.sidebar.image("C:/Users/ROMO-ADM1/Documents/9. Avril/SIA Hackathon/poultry_nutritionist/img/avril_logo_rvb.jpg")
+#region# Elements graphiques
 
-st.sidebar.header("")
 #Choix production
 choix_prod = st.sidebar.pills(
     "Sur quelle espèce voulez-vous avoir des renseignements ?",
-    ("Pondeuse", "Chair"),
-)
+    ("Pondeuse", "Chair"),)
 
 #Niveau vulgarisation
 choix_vulgarisation = st.sidebar.pills(
     "Quel niveau de vulgarisation souhaitez-vous ?   (1- Très vulgarisé  2-Intermédiaire  3-Technique)",
-    ("1", "2", "3"),
-)
+    ("1", "2", "3"),)
+
 
 #Années de publication
 choix_annee = st.sidebar.slider("Années de publication",
@@ -27,32 +30,86 @@ choix_annee = st.sidebar.slider("Années de publication",
                                 max_value=2025,
                                 value=(2020,2025))
 
+#endregion
 
-# Chargement du modèle
+#region# Chargement du modèle
 @st.cache_data
 def load_model():
     return pipeline("text-generation", model="gpt2")
 
 model = load_model()
+#endregion
 
-# Interface utilisateur
+#region# Interface utilisateur
 st.sidebar.header("")
-user_input = st.text_input("Vous : ", "")
+user_input = st.text_area("Entrez votre question:", placeholder="E.g., quelle est la meilleure alimentation pour les poules pondeuses ?")
 
-if st.button("Envoyer"):
-    if user_input:
-        with st.spinner("Réflexion de l'IA..."):
+if st.button("Envoyer la question..."):
+    if user_input and choix_prod and choix_vulgarisation and choix_annee :
+        with st.spinner("Veuillez patienter quelques instants..."):
             # Génération de la réponse
             response = model(user_input, max_length=100, num_return_sequences=1)
             bot_response = response[0]['generated_text']
             
-            st.markdown("**Bot :** " + bot_response)
+            # st.markdown("**Bot :** \\t " + bot_response)
+            # Afficher un titre
+            st.subheader("Réponse :")
+
+            # Ajouter du texte Markdown avec un cadre
+            st.markdown(f"""
+            <div style="border: 2px solid #453103; padding: 15px; border-radius: 10px;">
+                {bot_response}
+            </div>
+            """, unsafe_allow_html=True)
+
+            # Afficher un titre
+            st.subheader("Sources :")
+
+            # Ajouter du texte Markdown avec un cadre
+            st.markdown("""
+            <div style="border: 2px solid #453103; padding: 15px; border-radius: 10px;">
+                Sources
+            </div>
+            """, unsafe_allow_html=True)
+
+            st.markdown("""
+            <div style="border: 2px solid #453103; padding: 15px; border-radius: 10px;">
+                Reviews
+            </div>
+            """, unsafe_allow_html=True)
+
+
+
     else:
-        st.warning("Veuillez entrer un message!")
+        if not user_input:
+            st.warning("Veuillez entrer un message!")
+        if not choix_prod or not choix_vulgarisation or not choix_annee:
+            st.warning("Veuillez compléter les paramètres dans le bandeau latéral de gauche!")
+#endregion
+
+# choix_prod, choix_vulgarisation, choix_annee
 
-# Styles CSS pour améliorer l'apparence
+#region# Markdown
+#  Styles CSS pour améliorer l'apparence
 st.markdown("""
     <style>
+
+        /* Changer la couleur de fond de l'ensemble de l'application */
+        body {
+            background-color: #feffb3;  /* Couleur de fond bleu clair (par exemple) */
+            color: #453103;  /* Couleur de la police en gris foncé */
+        }
+
+        /* Personnaliser la couleur des titres */
+        h1, h2, h3, h4, h5, h6 {
+            color: #4CAF50;  /* Changer la couleur des titres en vert */
+        }
+
+        /* Personnaliser la couleur des paragraphes */
+        p {
+            color: #555555;  /* Couleur du texte dans les paragraphes */
+        }
+
         .st-bx {
             background-color: #f1f1f1;
             border-radius: 10px;
@@ -66,4 +123,9 @@ st.markdown("""
             color: #555;
         }
     </style>
-""", unsafe_allow_html=True)
+""", unsafe_allow_html=True)
+
+
+
+
+#endregion