feat : Store in local db

db/db.py

@@ -0,0 +1,121 @@
+import sqlite3
+from contextlib import contextmanager
+
+DATABASE_PATH = "database.db"  # Chemin de la base de données
+
+# Gestionnaire de contexte pour gérer les connexions
+@contextmanager
+def connect_db():
+    conn = sqlite3.connect(DATABASE_PATH)
+    try:
+        yield conn
+    finally:
+        conn.commit()
+        conn.close()
+
+# Classe responsable de la base de données
+class DatabaseHandler:
+    def __init__(self, db_path=DATABASE_PATH):
+        self.db_path = db_path
+        self._initialize_db()
+
+    def _initialize_db(self):
+        with connect_db() as conn:
+            cursor = conn.cursor()
+            cursor.execute('''CREATE TABLE IF NOT EXISTS prompts (
+                                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                                type TEXT NOT NULL CHECK(type IN ('installation', 'difficult')),
+                                num INTEGER NOT NULL,
+                                title TEXT NOT NULL,
+                                prompt TEXT NOT NULL
+                              );''')
+            if not self._is_data_present(cursor):
+                self._load_default_data(cursor)
+
+    def _is_data_present(self, cursor):
+        cursor.execute("SELECT COUNT(*) FROM prompts;")
+        return cursor.fetchone()[0] > 0
+
+    def _load_default_data(self, cursor):
+        data = [
+            {"type": "installation", "num": 1, "title": "Contexte et objectifs", "prompt": "context_objectives"},
+            {"type": "installation", "num": 2, "title": "Statut social, juridique et fiscal", "prompt": "social_legal_fiscal_status"},
+            {"type": "installation", "num": 3, "title": "Moyens humains", "prompt": "human_resources"},
+            {"type": "installation", "num": 4, "title": "Moyens de production", "prompt": "production_resources"},
+            {"type": "installation", "num": 5, "title": "Production par atelier", "prompt": "workshop_production"},
+            {"type": "difficult", "num": 11, "title": "Contexte et objectifs", "prompt": "difficult  context_objectives"},
+            {"type": "difficult", "num": 12, "title": "Statut social, juridique et fiscal", "prompt": "difficult  social_legal_fiscal_status"},
+            {"type": "difficult", "num": 13, "title": "Moyens humains", "prompt": "difficult  human_resources"},
+            {"type": "difficult", "num": 14, "title": "Moyens de production", "prompt": "difficult  production_resources"},
+            {"type": "difficult", "num": 15, "title": "Production par atelier", "prompt": "difficult  workshop_production"}
+        ]
+        for record in data:
+            cursor.execute(
+                "INSERT INTO prompts (type, num, title, prompt) VALUES (:type, :num, :title, :prompt);",
+                record
+            )
+        cursor.connection.commit()
+
+    def _rows_to_dicts(self, rows):
+        result = []
+        for r in rows:
+            result.append({
+                "id": r[0],
+                "type": r[1],
+                "num": r[2],
+                "title": r[3],
+                "prompt_system": r[4],
+            })
+        return result
+
+    def get_prompts(self):
+        with connect_db() as conn:
+            cursor = conn.cursor()
+            cursor.execute("SELECT * FROM prompts;")
+            rows = cursor.fetchall()
+            return self._rows_to_dicts(rows)
+
+    def get_prompt_by_filters(self, type_=None, num=None):
+        with connect_db() as conn:
+            cursor = conn.cursor()
+            query = "SELECT * FROM prompts"
+            conditions = []
+            params = []
+            
+            if type_:
+                conditions.append("type = ?")
+                params.append(type_)
+            if num:
+                conditions.append("num = ?")
+                params.append(num)
+            
+            if conditions:
+                query += " WHERE " + " AND ".join(conditions)
+            
+            cursor.execute(query, params)
+            rows = cursor.fetchall()
+            return self._rows_to_dicts(rows)
+
+    def add_prompt(self, type_, num, title, prompt):
+        with connect_db() as conn:
+            cursor = conn.cursor()
+            cursor.execute("INSERT INTO prompts (type, num, title, prompt) VALUES (?, ?, ?);", (type_, num, title, prompt))
+
+    def update_prompt(self, prompt_id, type_=None, num=None, title=None, prompt=None):
+        with connect_db() as conn:
+            cursor = conn.cursor()
+            query = "UPDATE prompts SET "
+            fields = []
+            params = []
+            if type_:
+                fields.append("type = ?")
+                params.append(type_)
+            if title:
+                fields.append("title = ?")
+                params.append(title)
+            if prompt:
+                fields.append("prompt = ?")
+                params.append(prompt)
+            query += ", ".join(fields) + " WHERE num = ?;"
+            params.append(prompt_id)
+            cursor.execute(query, params)

pages/chapter_params.py

@@ -1,22 +1,29 @@
 import streamlit as st
+from db.db import DatabaseHandler
+
+# Instanciation de la base de données
+db = DatabaseHandler()
 
 def area_change(key, ):
     new_value = st.session_state[key]
-    
-    key = key[5:]
-    chapter_key = f"chapter_{key}"
+    key = key[5:] # remove 'area_' prefix
 
-    if chapter_key in st.session_state:
-        st.session_state[chapter_key]["prompt_system"] = new_value
+    db.update_prompt(key, prompt=new_value)
 
 
 def page():
     st.subheader("Définissez vos paramètres")
     
-    chapters = []
-    for chapter in st.session_state["chapters"]:
-        chapters.append(st.session_state[f"chapter_{chapter['num']}"])
-        
+    type = "Installation"
+    # type = st.selectbox("Type de diagnostique ?", ["Installation", "Difficulté"], key="mode")
+    if(type == "Installation"):
+        diag_type = "installation"
+    else:
+        diag_type = "difficult"
+
+    # Chargement de l'enregistrement correspondant aux filtres
+    chapters = db.get_prompt_by_filters(type_=diag_type)
+
     parts_sorted = sorted(chapters, key=lambda part: part.get('num', float('inf')))
     
     # Création de tabs pour chaque 'part' trié

pages/chatbot.py

@@ -1,11 +1,14 @@
 import streamlit as st
 import uuid
-import os
+from db.db import DatabaseHandler
 
 from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
 from model import selector
 from st_copy_to_clipboard import st_copy_to_clipboard
 
+# Instanciation de la base de données
+db = DatabaseHandler()
+
 chapter_num = 0
 chapter_session_key = f"chapter_{chapter_num}"
 
@@ -85,6 +88,14 @@ def page():
         return
 
 
+    # Chargement de l'enregistrement correspondant aux filtres
+    chapterDB = db.get_prompt_by_filters(num=chapter_num)
+    if len(chapterDB) == 0:
+        st.text("Chapitre non trouvé")
+        return
+    
+    chapterDB = chapterDB[0]
+
     #################
     # Some controls #
     #################
@@ -96,7 +107,7 @@ def page():
         st.session_state[chapter_session_key]["messages"] = [ ]
 
     if len(st.session_state[chapter_session_key]["messages"]) < 2 :
-        st.session_state[chapter_session_key]["messages"] = [ SystemMessage(content=chapter["prompt_system"]) ]
+        st.session_state[chapter_session_key]["messages"] = [ SystemMessage(content=chapterDB['prompt_system']) ]
 
 
     ############