Upload 2 files

client/pages/nutri.py

@@ -17,13 +17,16 @@ from server.db.dbmanager import (
     delete_conversation,
     load_chatbot_suggestions,
     save_chatbot_suggestions,
+    save_recipes_with_ingredients,
+    add_ingredients_column_if_not_exists
 )
+
 import logging
 
+
 logging.basicConfig(level=logging.INFO, handlers=[logging.StreamHandler()])
 logger = logging.getLogger(__name__)
 
-
 # 🔹 Chargement des variables de session pour éviter les rechargements inutiles
 if "id_conversation" not in st.session_state:
     st.session_state.id_conversation = None
@@ -57,7 +60,7 @@ except Exception as e:
 # 🔹 Chargement de la base de données
 db_manager = st.session_state["db_manager"]
 user_id = st.session_state["user_id"]
-
+add_ingredients_column_if_not_exists(db_manager)
 if "chatbot_suggestions" not in st.session_state:
     st.session_state["chatbot_suggestions"] = load_chatbot_suggestions(
         db_manager, user_id
@@ -242,8 +245,6 @@ if prompt := st.chat_input("Dîtes quelque-chose"):
 
                         time.sleep(0.03)
 
-                # 🔹 Vérifier si la réponse contient une suggestion de recette
-
                 # 🔹 Vérifier si la réponse contient des suggestions de recettes
                 keywords = ["recette", "plat", "préparer", "ingrédients"]
 
@@ -278,18 +279,55 @@ if prompt := st.chat_input("Dîtes quelque-chose"):
                                 save_chatbot_suggestions(
                                     db_manager, user_id, new_recipes
                                 )
+
                         except Exception as e:
                             print(
                                 f"❌ Erreur lors de l'extraction des suggestions : {e}"
                             )
 
                         break  # On ne veut ajouter qu'une seule suggestion par réponse
+                time.sleep(1)   
+                # 🔹 EXTRACTION ET ENREGISTREMENT DES RECETTES
+                # 🔍 Vérifier si la réponse contient des suggestions de recettes
+                # 🔹 EXTRACTION ET ENREGISTREMENT DES RECETTES
+                # 🔍 Vérifier si la réponse contient des suggestions de recettes
+                for word in keywords:
+                    if word in response.lower():
+                        try:
+                            print("🔍 Détection de recettes, extraction en cours...")
+
+                            # 🔹 Utiliser `extract_recipes_and_ingredients` pour extraire plusieurs recettes
+                            extracted_recipes = mistral.extract_recipes_and_ingredients(response)
+
+                            if extracted_recipes:
+                                print(f"✅ {len(extracted_recipes)} recettes détectées : {extracted_recipes}")
+
+                                # 🔹 Sauvegarder les recettes et leurs ingrédients dans la base de données
+                                for recipe in extracted_recipes:
+                                    title = recipe["titre"].lstrip("-").strip()  # Supprime les tirets et espaces inutiles
+                                    ingredients = recipe["ingredients"].strip()
+
+                                    # 🔹 Supprimer une virgule finale si présente dans le titre
+                                    title_cleaned = title.rstrip(",")  # Enlève uniquement la virgule à la fin
 
-                # end_time = time.time()  # 🔹 Fin du chronomètre
-                # latency = round(end_time - start_time, 2)  # 🔹 Calcul de la latence
+                                    print(f"💾 Enregistrement de la recette '{title_cleaned}' avec les ingrédients : {ingredients}")
 
-                # print(f"✅ Réponse générée en {latency} secondes.")
-                # print(f"✅ Nombre de tokens de sortie : {output_tokens}")
+                                    save_recipes_with_ingredients(db_manager, user_id, title_cleaned, ingredients)
+
+                            else:
+                                print("⚠️ Aucune recette détectée.")
+
+                        except Exception as e:
+                            print(f"❌ Erreur lors de l'extraction des recettes et ingrédients : {e}")
+
+                        break  # Éviter d'ajouter plusieurs fois la même recette
+
+
+                end_time = time.time()  # 🔹 Fin du chronomètre
+                latency = round(end_time - start_time, 2)  # 🔹 Calcul de la latence
+
+                print(f"✅ Réponse générée en {latency} secondes.")
+                print(f"✅ Nombre de tokens de sortie : {output_tokens}")
             except Exception as e:
                 if hasattr(e, "status_code") and e.status_code == 429:
                     retries += 1
@@ -368,3 +406,19 @@ if prompt := st.chat_input("Dîtes quelque-chose"):
             "🤖 Entraînement du guardrail à reconnaître le prompt comme dangereux effectué avec succès"
         )
         st.stop()
+# def check_if_ingredients_saved(db_manager, user_id):
+#     query = "SELECT repas_suggestion, ingredients FROM suggestions_repas WHERE id_utilisateur = ?"
+#     recipes = db_manager.execute_safe(query, (user_id,), fetch=True)
+
+#     if recipes:
+#         print("✅ Recettes enregistrées en base :")
+#         for recipe in recipes:
+#             recette_nom = recipe[0]
+#             ingredients = recipe[1] if recipe[1] else "⚠️ Aucun ingrédient enregistré"
+#             print(f"📌 {recette_nom} - Ingrédients : {ingredients}")
+#     else:
+#         print("⚠️ Aucune recette enregistrée.")
+
+# # Appel pour tester
+# check_if_ingredients_saved(db_manager, user_id)
+

client/pages/user__course_list.py

@@ -1,17 +1,78 @@
 import streamlit as st
-from server.db.db import get_ingredients
+import pandas as pd
+import unicodedata
+from server.db.dbmanager import get_db_manager
 
+def normalize_text(text):
+    """Normalise un texte en supprimant les accents et en le mettant en minuscules"""
+    text = unicodedata.normalize("NFKD", text).encode("ASCII", "ignore").decode("utf-8").strip().lower()
+    return text
 
-# Page des courses
 def course_list():
-    st.write("Voici votre liste des courses.")
+    # Récupérer l'ID utilisateur
+    user_id = st.session_state.get("user_id")
+    if not user_id:
+        st.warning("⚠️ Vous devez être connecté pour voir votre liste de courses.")
+        return
 
-    ingredients_list = get_ingredients()
+    # Charger la base de données
+    db_manager = get_db_manager()
+    
+    # Charger les recettes et leurs ingrédients depuis la base de données
+    query = "SELECT repas_suggestion, ingredients FROM suggestions_repas WHERE id_utilisateur = ?"
+    raw_recipes = db_manager.execute_safe(query, (user_id,), fetch=True)
 
-    if ingredients_list:
-        for ingredient in ingredients_list:
-            st.write(f"🍎 {ingredient[0]}")
+    if not raw_recipes:
+        st.warning("⚠️ Aucune recette enregistrée pour générer une liste de courses.")
+        return
+
+    # ✅ Normalisation et suppression des doublons
+    recipes = {}
+    formatted_titles = {}
+
+    for recipe in raw_recipes:
+        title, ingredients = recipe["repas_suggestion"], recipe["ingredients"]
+        normalized_title = normalize_text(title)
+
+        if normalized_title not in recipes or (not recipes[normalized_title] and ingredients):
+            recipes[normalized_title] = ingredients
+            formatted_titles[normalized_title] = title
+
+    recipe_titles = list(formatted_titles.values())
+
+    # Sélectionner des recettes
+    selected_recipes = st.multiselect("📌 Sélectionnez une ou plusieurs recettes", recipe_titles)
+
+    # Initialiser all_ingredients pour éviter UnboundLocalError
+    all_ingredients = []
+
+    # Récupérer les ingrédients des recettes sélectionnées
+    selected_ingredients = []
+    for recipe in selected_recipes:
+        normalized_recipe = normalize_text(recipe)
+        ingredients = recipes.get(normalized_recipe)
+        if ingredients:
+            selected_ingredients.extend(ingredients.split(", "))
+
+    # ✅ Option pour afficher la liste complète de courses
+    if st.checkbox("👀 Afficher la liste complète des courses"):
+        all_ingredients = list({ing for ing_list in recipes.values() if ing_list for ing in ing_list.split(", ")})
+
+    # Affichage des ingrédients récupérés
+    ingredients_to_display = selected_ingredients if selected_ingredients else all_ingredients
+    if ingredients_to_display:
+        st.subheader("📋 Ingrédients nécessaires")
+        for ingredient in ingredients_to_display:
+            st.write(f"🛒 {ingredient}")
     else:
-        st.write("Aucun ingrédient trouvé.")
+        st.warning("⚠️ Aucun ingrédient enregistré.")
+
+    # 📥 Exporter la liste des courses
+    df = pd.DataFrame({"Ingrédients": ingredients_to_display})
 
-    st.checkbox("Ajouter un nouvel article")
+    if st.download_button(
+            label="📥 Télécharger la liste des courses",
+            data=df.to_csv(index=False, sep=";").encode("utf-8-sig"),
+            file_name="liste_courses.csv",
+            mime="text/csv"):
+        st.success("✅ Liste des courses exportée en CSV avec succès !")