enhacnemnet

rag.py

@@ -90,9 +90,11 @@ def split_texts(text : str) -> list[str]:
 #########################################################################
 ### PAR ICI , CHOISIR UNE SCENE SPECIFIQUE DANS L'ARGUMENT DE LA FONCTION
 
-def get_random_chunk(scene_specific = 4) : # scene_specific = None signifie qu'on considère tout le récit
+def get_random_chunk(scene_specific = [4]) : # scene_specific = None signifie qu'on considère tout le récit
     if scene_specific:
-        chunks_scene = split_texts(scenes[scene_specific-1])
+        scene_specific_content = [scenes[i-1] for i in scene_specific]
+        scene_specific_content = " ".join(scene_specific_content)
+        chunks_scene = split_texts(scene_specific_content)
         print(f"Scene {scene_specific} has {len(chunks_scene)} chunks")
         print([chunk[0:50] for chunk in chunks_scene])
         print('---')