handle big files

audit_page/audit.py

@@ -46,12 +46,15 @@ def display_content_doc(content:dict,col:st):
             images = page["images"]
         col1,col2,col3 = col.columns(3)
         for i, (img_bytes, img_width, img_height) in enumerate(images):
-            if i%3 == 0:
-                col1.image(Image.open(BytesIO(img_bytes)), caption=f'', width=img_width)
-            elif i%3 == 1:
-                col2.image(Image.open(BytesIO(img_bytes)), caption=f'', width=img_width)
-            else:
-                col3.image(Image.open(BytesIO(img_bytes)), caption=f'', width=img_width)
+            try:
+                if i%3 == 0:
+                    col1.image(Image.open(BytesIO(img_bytes)), caption=f'', width=img_width)
+                elif i%3 == 1:
+                    col2.image(Image.open(BytesIO(img_bytes)), caption=f'', width=img_width)
+                else:
+                    col3.image(Image.open(BytesIO(img_bytes)), caption=f'', width=img_width)
+            except:
+                pass
 
     elif option == "texte":
         if number == 0:

audit_page/compte_rendu.py

@@ -32,10 +32,11 @@ def cr_main():
 
     À partir du document ci-dessous, générez un compte rendu détaillé contenant les sections suivantes :
 
-    2. **Résumé** : Fournissez un résumé concis du document, en mettant en avant les points principaux, les relations essentielles, les concepts , les dates et les lieux, les conclusions et les détails importants.
+    2. **Résumé** : Fournissez une synthèse du document, en mettant en avant les points principaux, les relations essentielles, les concepts , les dates et les lieux, les conclusions et les détails importants.
 
     3. **Notes** :
     - Présentez les points clés sous forme de liste à puces avec des émojis pertinents pour souligner la nature de chaque point.
+    - N'oubliez pas de relever tout les entités et les relations.
     - Incluez des sous-points (sans émojis) sous les points principaux pour offrir des détails ou explications supplémentaires.
 
     4. **Actions** : Identifiez et listez les actions spécifiques, tâches ou étapes recommandées ou nécessaires selon le contenu du document.
@@ -48,7 +49,7 @@ def cr_main():
 
 
     ### Résumé :
-    [Fournissez un résumé concis du document ici.]
+    [Fournissez un résumé concis du document ici;n'oubliez pas de relever tout les entités et les relations.]
 
     ### Notes :
     - 📌 **Point Principal 1**

audit_page/dialogue_doc.py

@@ -107,51 +107,53 @@ def doc_dialog_main():
     elif audit["type de fichier"] == "audio":
         text = get_text_from_content_for_audio(content)
 
-    
+    prompt_cr = dedent(f'''
 
-    if choice == "compte_rendu":
-        if "cr" not in st.session_state or st.session_state.cr == "":
-            with st.spinner("Génération du compte rendu..."):
-                prompt_cr = dedent(f'''
-                    À partir du document ci-dessous, générez un compte rendu détaillé contenant les sections suivantes :
+                À partir du document ci-dessous, générez un compte rendu détaillé contenant les sections suivantes :
+
+                2. **Résumé** : Fournissez une synthèse complète du document, en mettant en avant les points principaux, les relations essentielles, les concepts , les dates et les lieux, les conclusions et les détails importants.
+
+                3. **Notes** :
+                - Présentez les points clés sous forme de liste à puces avec des émojis pertinents pour souligner la nature de chaque point.
+                - N'oubliez pas de relever tout les entités et les relations.
+                - Incluez des sous-points (sans émojis) sous les points principaux pour offrir des détails ou explications supplémentaires.
 
-                    2. **Résumé** : Fournissez un résumé concis du document, en mettant en avant les points principaux, les relations essentielles, les concepts , les dates et les lieux, les conclusions et les détails importants.
+                4. **Actions** : Identifiez et listez les actions spécifiques, tâches ou étapes recommandées ou nécessaires selon le contenu du document.
 
-                    3. **Notes** :
-                    - Présentez les points clés sous forme de liste à puces avec des émojis pertinents pour souligner la nature de chaque point.
-                    - Incluez des sous-points (sans émojis) sous les points principaux pour offrir des détails ou explications supplémentaires.
+                **Document :**
+                                
+                {text}
 
-                    4. **Actions** : Identifiez et listez les actions spécifiques, tâches ou étapes recommandées ou nécessaires selon le contenu du document.
+                **Format de sortie :**
 
-                    **Document :**
-                                    
-                    {text}
 
-                    **Format de sortie :**
+                ### Résumé :
+                [Fournissez un résumé concis du document ici;n'oubliez pas de relever tout les entités et les relations.]
 
+                ### Notes :
+                - 📌 **Point Principal 1**
+                - Sous-point A
+                - Sous-point B
+                - 📈 **Point Principal 2**
+                - Sous-point C
+                - Sous-point D
+                - 📝 **Point Principal 3**
+                - Sous-point E
+                - Sous-point F
 
-                    ### Résumé :
-                    [Fournissez un résumé concis du document ici.]
+                ### Actions :
+                1. [Action 1]
+                2. [Action 2]
+                3. [Action 3]
+                4. ...
 
-                    ### Notes :
-                    - 📌 **Point Principal 1**
-                    - Sous-point A
-                    - Sous-point B
-                    - 📈 **Point Principal 2**
-                    - Sous-point C
-                    - Sous-point D
-                    - 📝 **Point Principal 3**
-                    - Sous-point E
-                    - Sous-point F
+                ---
+                ''')
 
-                    ### Actions :
-                    1. [Action 1]
-                    2. [Action 2]
-                    3. [Action 3]
-                    4. ...
 
-                    ---
-                    ''')
+    if choice == "compte_rendu":
+        if "cr" not in st.session_state or st.session_state.cr == "":
+            with st.spinner("Génération du compte rendu..."):
                 cr = generate_response_via_langchain(prompt_cr,stream=False,model="gpt-4o")
                 st.session_state.cr =  cr
                 st.session_state.cr_chat_history = []
@@ -211,6 +213,13 @@ def doc_dialog_main():
             with st.spinner("Génération du graphe..."):
                 keywords_list = [keyword.strip() for keyword in audit["Mots clés"].strip().split(",")]
                 allowed_nodes_types =keywords_list+ ["Person","Organization","Location","Event","Date","Time","Ressource","Concept"]
+
+                number_tokens = audit["Nombre de tokens"]
+                if number_tokens > 10000:
+                    if st.session_state.cr == "":
+                        st.session_state.cr = generate_response_via_langchain(prompt_cr,stream=False,model="gpt-4o")
+                    text = st.session_state.cr
+                
                 graph = get_graph(text,allowed_nodes=allowed_nodes_types)
                 st.session_state.graph = graph
                 st.session_state.filter_views = {}