Initial commit

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+chroma_db/** filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,216 @@
+import streamlit as st
+import os
+import datetime
+from fpdf import FPDF
+from openai import OpenAI
+from langchain.chat_models import ChatOpenAI
+from langchain.memory import ConversationBufferMemory
+from langchain.vectorstores import Chroma
+from langchain.embeddings import OpenAIEmbeddings
+from langchain.chains import LLMChain
+from langchain.prompts import PromptTemplate
+
+# ========== SETUP ==========
+st.set_page_config(page_title="Papill’IA - Assistant Recette", page_icon="🍳")
+
+# ========== CONFIG OPENAI ==========
+openai_api_key = "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"
+openai_client = OpenAI(api_key=openai_api_key)
+
+# ========== DOSSIERS ==========
+DATA_DIR = "data"
+os.makedirs(DATA_DIR, exist_ok=True)
+PERSO_PATH = os.path.join(DATA_DIR, "recettes_perso.txt")
+
+# ========== CHROMA & LangChain Setup ==========
+@st.cache_resource
+def load_resources():
+    vectordb = Chroma(
+        collection_name="recipes",
+        persist_directory="chroma_db",
+        embedding_function=OpenAIEmbeddings(openai_api_key=openai_api_key)
+    )
+    retriever = vectordb.as_retriever(search_kwargs={"k": 3})
+    memory = ConversationBufferMemory(
+        memory_key="chat_history",
+        return_messages=True,
+        output_key="answer"
+    )
+    llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.7, openai_api_key=openai_api_key)
+    return vectordb, retriever, memory, llm
+
+vectordb, retriever, memory, llm = load_resources()
+
+# ========== INTERFACE ==========
+st.title("👨‍🍳 Papill’IA – Ton assistant de cuisine intelligent")
+tabs = st.tabs(["💬 Discuter avec Papill’IA", "📚 Mes favoris", "📖 Mes recettes perso", "➕ Contribuer une recette"])
+
+# ===== TAB 1 : CHAT =====
+with tabs[0]:
+    st.subheader("💬 Discuter avec Papill’IA")
+
+    if st.button("🔄 Réinitialiser la conversation"):
+        st.session_state.chat_history = []
+        st.session_state.last_reply = ""
+        fav_path = os.path.join(DATA_DIR, "recettes_favorites.txt")
+        if os.path.exists(fav_path):
+            os.remove(fav_path)
+        st.success("💬 Conversation et favoris réinitialisés.")
+
+    if "chat_history" not in st.session_state:
+        st.session_state.chat_history = []
+    if "last_reply" not in st.session_state:
+        st.session_state["last_reply"] = ""
+
+    with st.expander("⚙️ Options de personnalisation"):
+        portions = st.slider("👥 Nombre de personnes", 1, 10, 2)
+        regimes = st.multiselect("📜 Régimes alimentaires", ["Végétarien", "Vegan", "Sans gluten", "Sans lactose", "Keto", "Halal"])
+
+    for msg in st.session_state.chat_history:
+        with st.chat_message(msg["role"]):
+            st.markdown(msg["content"])
+
+    user_msg = st.chat_input("Pose ta question ou demande une recette personnalisée...")
+
+    if user_msg:
+        last_recipe = st.session_state["last_reply"]
+        modif_keywords = ["sans", "retire", "enlève", "j’aime pas", "j'aime pas", "enlever", "supprime", "remplace"]
+        is_modification = any(k in user_msg.lower() for k in modif_keywords)
+
+        if is_modification and last_recipe:
+            full_query = f"""
+Tu es Papill’IA, un chef cuisinier français expert et créatif.
+Voici une recette à modifier selon les indications de l'utilisateur.
+
+🧰 Recette actuelle :
+{last_recipe}
+
+🔄 Demande de modification :
+{user_msg}
+
+Consignes :
+- Ne propose PAS une nouvelle recette sauf si expressément demandé.
+- Modifie uniquement ce qui est demandé (par exemple : juste l'accompagnement).
+- Garde le reste identique sauf si incohérent.
+- Réponds avec une recette complète modifiée.
+"""
+        else:
+            full_query = f"""
+Tu es Papill’IA, un chef cuisinier français expert, drôle et bienveillant.
+Tu proposes des recettes personnalisées en respectant les consignes suivantes :
+- Adapter aux préférences : {portions} personne(s), régimes : {', '.join(regimes) if regimes else 'aucun'}
+- Proposer un accompagnement adapté et des astuces de chef.
+- Fournir une estimation nutritionnelle.
+- Être flexible et inventif même si la demande est floue ou étrange.
+
+Demande utilisateur :
+{user_msg}
+"""
+
+        st.session_state.chat_history.append({"role": "user", "content": user_msg})
+        with st.chat_message("user"):
+            st.markdown(user_msg)
+
+        with st.spinner("Papill’IA réfléchit..."):
+            docs = retriever.get_relevant_documents(full_query)
+            if docs:
+                prompt_template = PromptTemplate.from_template("""{question}""")
+                qa_chain = LLMChain(llm=llm, prompt=prompt_template)
+                reply = qa_chain.invoke({"question": full_query})["text"]
+            else:
+                reply = llm.invoke(full_query).content
+
+        st.session_state.chat_history.append({"role": "assistant", "content": reply})
+        st.session_state["last_reply"] = reply
+
+        with st.chat_message("assistant"):
+            st.markdown(reply)
+
+# ===== PDF & FAVORIS =====
+        def save_pdf(text):
+            pdf = FPDF()
+            pdf.add_page()
+            pdf.set_font("Arial", size=12)
+            for line in text.split('\n'):
+                pdf.multi_cell(0, 10, line)
+            filename = f"recette_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf"
+            filepath = os.path.join(DATA_DIR, filename)
+            pdf.output(filepath)
+            return filepath
+
+        def save_favorite(text):
+            fav_path = os.path.join(DATA_DIR, "recettes_favorites.txt")
+            with open(fav_path, "a", encoding="utf-8") as f:
+                f.write("\n" + "="*60 + "\n")
+                f.write(text.strip() + "\n")
+            st.success("✅ Recette sauvegardée dans les favoris !")
+
+        col1, col2 = st.columns(2)
+        with col1:
+            if st.button("📄 Télécharger en PDF"):
+                pdf_path = save_pdf(reply)
+                with open(pdf_path, "rb") as file:
+                    st.download_button("📅 Télécharger le PDF", file, pdf_path, mime="application/pdf")
+
+        with col2:
+            if st.button("❤️ Ajouter aux favoris"):
+                save_favorite(reply)
+
+# ====== TAB 2 : FAVORIS ======
+with tabs[1]:
+    st.subheader("📚 Mes recettes favorites")
+    fav_path = os.path.join(DATA_DIR, "recettes_favorites.txt")
+    if os.path.exists(fav_path):
+        with open(fav_path, "r", encoding="utf-8") as f:
+            favoris = f.read().strip()
+        if favoris:
+            st.text_area("📜 Recettes sauvegardées :", favoris, height=400)
+        else:
+            st.info("Aucune recette sauvegardée pour le moment.")
+    else:
+        st.info("Le fichier des favoris n’existe pas encore. Sauvegarde une recette pour le créer.")
+
+# ====== TAB 3 : RECETTES PERSO ======
+with tabs[2]:
+    st.subheader("📖 Mes recettes perso")
+    if os.path.exists(PERSO_PATH):
+        with open(PERSO_PATH, "r", encoding="utf-8") as f:
+            perso = f.read().strip()
+        if perso:
+            st.text_area("📝 Recettes personnelles :", perso, height=400)
+        else:
+            st.info("Aucune recette personnelle ajoutée pour le moment.")
+    else:
+        st.info("Tu n’as pas encore contribué de recette.")
+
+# ====== TAB 4 : CONTRIBUTION ======
+with tabs[3]:
+    st.subheader("➕ Ajouter une recette personnelle")
+
+    option = st.radio("Méthode d'ajout", ["✍️ Écrire la recette", "📄 Importer un PDF"])
+
+    if option == "✍️ Écrire la recette":
+        titre = st.text_input("Nom de la recette")
+        contenu = st.text_area("Écris ici ta recette complète")
+        if st.button("✅ Ajouter à la base"):
+            if titre and contenu.strip():
+                vectordb.add_texts([contenu], ids=[titre])
+                with open(PERSO_PATH, "a", encoding="utf-8") as f:
+                    f.write(f"\n{'='*40}\n{titre}\n{contenu.strip()}\n")
+                st.success("🎉 Recette ajoutée avec succès à la base de données !")
+                st.chat_message("assistant").markdown(f"✅ Recette **{titre}** ajoutée !")
+            else:
+                st.warning("Merci de remplir les deux champs.")
+
+    elif option == "📄 Importer un PDF":
+        pdf_file = st.file_uploader("📄 Dépose ton fichier PDF", type=["pdf"])
+        if pdf_file and st.button("✅ Importer le PDF"):
+            from PyPDF2 import PdfReader
+            reader = PdfReader(pdf_file)
+            texte = "\n".join([page.extract_text() for page in reader.pages])
+            titre = os.path.splitext(pdf_file.name)[0]
+            vectordb.add_texts([texte], ids=[titre])
+            with open(PERSO_PATH, "a", encoding="utf-8") as f:
+                f.write(f"\n{'='*40}\n{titre}\n{texte.strip()}\n")
+            st.success(f"📚 PDF '{titre}' indexé avec succès dans la base !")
+            st.chat_message("assistant").markdown(f"✅ Recette **{titre}** ajoutée via PDF !")

chroma_db/ad4f4931-3d96-4e9c-a3c8-04cd4fbb0ce5/data_level0.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bb41acbe3c1b8607508173321e7ebad22da8c1cf81b42d99158c6cab8ab27aef
+size 106828000

chroma_db/ad4f4931-3d96-4e9c-a3c8-04cd4fbb0ce5/header.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:636ae6d5d386e835b6249d1a233a3b30d0c8e13c55225dd0aaf9581cc3850b49
+size 100

chroma_db/ad4f4931-3d96-4e9c-a3c8-04cd4fbb0ce5/index_metadata.pickle

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:199e8f3b67bc853a37095d8c6020c9b8bcf1e9cbce2c8f6a1274e620a32d1117
+size 470407

chroma_db/ad4f4931-3d96-4e9c-a3c8-04cd4fbb0ce5/length.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a81139698ef3acae3a92638a6a26b313290dab405952c60f9013c7c0d2fc9b04
+size 68000

chroma_db/ad4f4931-3d96-4e9c-a3c8-04cd4fbb0ce5/link_lists.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d0b5839e8b4a750768b4237d6680419925345f22d0656845281894a580e7bed4
+size 147492

chroma_db/chroma.sqlite3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c615b4fab26523a6b5a8afbd3da06e22d29e3b8fbb6116441be7db14436ccb52
+size 89743360

requirements.txt

@@ -0,0 +1,5 @@
+streamlit
+fpdf
+requests
+chromadb
+sentence-transformers