minors updates

pages/main.py

@@ -3,6 +3,7 @@ import base64
 import io
 import json
 import os
+import uuid
 import re
 import tempfile
 import time
@@ -17,9 +18,12 @@ from typing import Union
 from io import BytesIO
 from copy import deepcopy
 
+
 # Third-party libraries
 import requests
 import streamlit as st
+import streamlit.components.v1 as components
+
 #from audiorecorder import audiorecorder
 from openai import OpenAI
 from pydub import AudioSegment
@@ -53,6 +57,23 @@ from core.moderation import api_moderation_openai_text
 
 
 
+def save_attachment(attachment):
+    """Sauvegarde la pièce jointe et retourne le chemin."""
+    # Créer un dossier pour les pièces jointes s'il n'existe pas
+    attachments_dir = 'attachments'
+    os.makedirs(attachments_dir, exist_ok=True)
+    
+    # Générer un nom de fichier unique
+    file_extension = os.path.splitext(attachment.name)[1]
+    filename = f"{uuid.uuid4()}{file_extension}"
+    file_path = os.path.join(attachments_dir, filename)
+    
+    # Sauvegarder le fichier
+    with open(file_path, 'wb') as f:
+        f.write(attachment.getbuffer())
+    
+    return file_path
+
 # Au début du fichier, après les imports
 st.set_page_config(
     page_title=f"DEMORRHA - (v{__version__})",
@@ -309,126 +330,130 @@ def main_page():
                 format_func=lambda lang: f"{LANGUAGES_EMOJI.get(lang, '')} {lang}"
             )
 
-    with st.container(border=True):
-        # Interface utilisateur pour le chat textuel
-        st.session_state.user_input = st.chat_input(
-            get_translation("entrez_message")
-        )
-        
-        st.experimental_audio_input("Record a voice message",on_change=recorder_released, key="rec_widget")
-
-        if st.session_state.user_input:
-            # Appeler la fonction de modération
-            moderation_result = api_moderation_openai_text(st.session_state.user_input)
-            if moderation_result.get("flagged"):
-                st.error("Votre message a été jugé inapproprié et ne peut pas être traité.")
-                return  # Arrêter le traitement si le message est inapproprié
-            elif "error" in moderation_result:
-                st.error(moderation_result["error"])
-                return  # Gérer les erreurs de modération
-
-            # Réinitialiser l'état précédent
-            st.session_state.full_response = ""
-
-            with st.chat_message("user", avatar="👤"):
-                st.markdown(st.session_state.user_input)
-
-            # Traitement du message texte de l'utilisateur
-            if st.session_state.language_detected is None:
-                st.session_state.language_detected = detect_language(
-                    input_text = st.session_state.user_input, 
-                    temperature = 0.01,
-                    context_window = 512,
-                    model="gpt-4o"
-                )
+    if st.session_state.get('show_report_form', False):
+        # show_report_form()
+        pass
+    else:
+        with st.container(border=True):
+            # Interface utilisateur pour le chat textuel
+            st.session_state.user_input = st.chat_input(
+                get_translation("entrez_message")
+            )
             
-            st.session_state.audio_list = []
-
-            for cursor_selected_lang in st.session_state.selected_languages:
-                st.session_state.target_language = cursor_selected_lang["iso-639-1"]
-                target_language_name = cursor_selected_lang["language"]
-
-                # Réinitialiser les messages avant de traiter une nouvelle entrée
-                st.session_state.messages = []
-                st.session_state.full_response = "" 
-                
-                # Initialisation du mode de traitement pour la langue cible actuelle
-                st.session_state.system_prompt, st.session_state.operation_prompt = init_process_mode(from_lang=
-                    (
-                        st.session_state.language_detected if "language_detected" in st.session_state.language_detected else convert_language_name_to_iso6391( 
-                            st.session_state.interface_language 
-                        )
-                    ), 
-                    to_lang=st.session_state.target_language 
-                )
-                # display error with st.error ; if (st.session_state.system_prompt, st.session_state.operation_prompt) is "", "" or None, None, and raise error 
-                if (not st.session_state.system_prompt) or (not st.session_state.operation_prompt):
-                    st.error("Erreur : Les prompts système ou d'opération sont vides.")
-                    raise ValueError("Les prompts système ou d'opération ne peuvent pas être vides.")
+            st.experimental_audio_input("Record a voice message",on_change=recorder_released, key="rec_widget")
+
+            if st.session_state.user_input:
+                # Appeler la fonction de modération
+                moderation_result = api_moderation_openai_text(st.session_state.user_input)
+                if moderation_result.get("flagged"):
+                    st.error("Votre message a été jugé inapproprié et ne peut pas être traité.")
+                    return  # Arrêter le traitement si le message est inapproprié
+                elif "error" in moderation_result:
+                    st.error(moderation_result["error"])
+                    return  # Gérer les erreurs de modération
+
+                # Réinitialiser l'état précédent
+                st.session_state.full_response = ""
+
+                with st.chat_message("user", avatar="👤"):
+                    st.markdown(st.session_state.user_input)
+
+                # Traitement du message texte de l'utilisateur
+                if st.session_state.language_detected is None:
+                    st.session_state.language_detected = detect_language(
+                        input_text = st.session_state.user_input, 
+                        temperature = 0.01,
+                        context_window = 512,
+                        model="gpt-4o"
+                    )
                 
-                with st.status(f'({target_language_name}) - {get_translation("traduction_en_cours")}', expanded=True) as response_status:
-                    with st.chat_message("assistant", avatar="👻"):
-                        message_placeholder = st.empty()
-                        st.session_state.response_generator = process_message(
-                            st.session_state.user_input, 
-                            st.session_state.operation_prompt, 
-                            st.session_state.system_prompt
-                        )
+                st.session_state.audio_list = []
 
-                        response_status.update(label=f'({target_language_name}) - {get_translation("traduction_en_cours")}', state="running", expanded=True)
-                        for response_chunk in st.session_state.response_generator:
-                            message_placeholder.markdown(response_chunk)
+                for cursor_selected_lang in st.session_state.selected_languages:
+                    st.session_state.target_language = cursor_selected_lang["iso-639-1"]
+                    target_language_name = cursor_selected_lang["language"]
 
-                        st.session_state.end_response = st.session_state.response_generator.close()  # Obtenir la réponse complète à la fin
-                        if st.session_state.full_response != "":
-                            message_placeholder.markdown(st.session_state.full_response)
+                    # Réinitialiser les messages avant de traiter une nouvelle entrée
+                    st.session_state.messages = []
+                    st.session_state.full_response = "" 
+                    
+                    # Initialisation du mode de traitement pour la langue cible actuelle
+                    st.session_state.system_prompt, st.session_state.operation_prompt = init_process_mode(from_lang=
+                        (
+                            st.session_state.language_detected if "language_detected" in st.session_state.language_detected else convert_language_name_to_iso6391( 
+                                st.session_state.interface_language 
+                            )
+                        ), 
+                        to_lang=st.session_state.target_language 
+                    )
+                    # display error with st.error ; if (st.session_state.system_prompt, st.session_state.operation_prompt) is "", "" or None, None, and raise error 
+                    if (not st.session_state.system_prompt) or (not st.session_state.operation_prompt):
+                        st.error("Erreur : Les prompts système ou d'opération sont vides.")
+                        raise ValueError("Les prompts système ou d'opération ne peuvent pas être vides.")
+                    
+                    with st.status(f'({target_language_name}) - {get_translation("traduction_en_cours")}', expanded=True) as response_status:
+                        with st.chat_message("assistant", avatar="👻"):
+                            message_placeholder = st.empty()
+                            st.session_state.response_generator = process_message(
+                                st.session_state.user_input, 
+                                st.session_state.operation_prompt, 
+                                st.session_state.system_prompt
+                            )
 
-                            if st.session_state.enable_tts_for_input_from_text_field:
-                                response_status.update(label=f'({target_language_name}) - {get_translation("traduction_terminee")} ; {get_translation("synthese_vocale_en_cours")}', state="running", expanded=False)
-                                st.session_state.tts_audio, st.session_state.tts_duration = process_tts_message(st.session_state.full_response)
-                                del st.session_state.full_response
-                                if st.session_state.tts_audio:
-                                    st.audio(base64.b64decode(st.session_state.tts_audio.encode()), format="audio/mp3", autoplay=False)
-                                    st.session_state.audio_list.append((base64.b64decode(st.session_state.tts_audio.encode()), st.session_state.tts_duration))
-                                    response_status.update(label=f'({target_language_name}) - {get_translation("traduction_terminee")} ; {get_translation("synthese_vocale_terminee")}', state="complete", expanded=False)
-                                else:
-                                    response_status.update(label=f'({target_language_name}) - {get_translation("erreur_synthese_vocale")}', state="error", expanded=False)
+                            response_status.update(label=f'({target_language_name}) - {get_translation("traduction_en_cours")}', state="running", expanded=True)
+                            for response_chunk in st.session_state.response_generator:
+                                message_placeholder.markdown(response_chunk)
+
+                            st.session_state.end_response = st.session_state.response_generator.close()  # Obtenir la réponse complète à la fin
+                            if st.session_state.full_response != "":
+                                message_placeholder.markdown(st.session_state.full_response)
+
+                                if st.session_state.enable_tts_for_input_from_text_field:
+                                    response_status.update(label=f'({target_language_name}) - {get_translation("traduction_terminee")} ; {get_translation("synthese_vocale_en_cours")}', state="running", expanded=False)
+                                    st.session_state.tts_audio, st.session_state.tts_duration = process_tts_message(st.session_state.full_response)
+                                    del st.session_state.full_response
+                                    if st.session_state.tts_audio:
+                                        st.audio(base64.b64decode(st.session_state.tts_audio.encode()), format="audio/mp3", autoplay=False)
+                                        st.session_state.audio_list.append((base64.b64decode(st.session_state.tts_audio.encode()), st.session_state.tts_duration))
+                                        response_status.update(label=f'({target_language_name}) - {get_translation("traduction_terminee")} ; {get_translation("synthese_vocale_terminee")}', state="complete", expanded=False)
+                                    else:
+                                        response_status.update(label=f'({target_language_name}) - {get_translation("erreur_synthese_vocale")}', state="error", expanded=False)
 
+                                else:
+                                    response_status.update(label=f'({target_language_name}) - {get_translation("traduction_terminee")}', state="complete", expanded=False)
                             else:
-                                response_status.update(label=f'({target_language_name}) - {get_translation("traduction_terminee")}', state="complete", expanded=False)
-                        else:
-                            response_status.update(label=f'({target_language_name}) - {get_translation("erreur_traduction")}', state="error", expanded=False)
-
-            if st.session_state.audio_list:
-                with st.status(f"{get_translation('concatenation_audio_en_cours')}", expanded=False) as audio_status:
-                    audio_status.update(label=f"{get_translation('concatenation_audio_en_cours')}", state="running", expanded=False)
-                    try:
-                        st.session_state.final_audio = concatenate_audio_files(st.session_state.audio_list)
-                        with st.container(border=True):
-
-                            # Générer un nom de fichier unique
-                            st.session_state.timestamp = time.strftime("%Y%m%d-%H%M%S")
-                            st.session_state.langues = "_".join([lang["iso-639-1"] for lang in st.session_state.selected_languages])
-                            st.session_state.nom_fichier = f"reponse_audio_{st.session_state.langues}_{st.session_state.timestamp}.mp3"
-
-                            st.audio(st.session_state.final_audio, format="audio/mp3", autoplay=st.session_state.autoplay_tts)
-
-                            st.download_button(
-                                label=f"📥 {get_translation('telecharger_audio')}",
-                                data=st.session_state.final_audio,
-                                file_name=st.session_state.nom_fichier,
-                                mime="audio/mp3",
-                                use_container_width=True,
-                                type="primary",
-                                key=f"download_button_{st.session_state.langues}_{st.session_state.timestamp}",
-                            )
+                                response_status.update(label=f'({target_language_name}) - {get_translation("erreur_traduction")}', state="error", expanded=False)
 
-                            audio_status.update(label=f"{get_translation('concatenation_audio_terminee')}", state="complete", expanded=True)
-                    except Exception as e:
-                        st.error(f"{get_translation('erreur_concatenation_audio')} : {str(e)}")
+                if st.session_state.audio_list:
+                    with st.status(f"{get_translation('concatenation_audio_en_cours')}", expanded=False) as audio_status:
+                        audio_status.update(label=f"{get_translation('concatenation_audio_en_cours')}", state="running", expanded=False)
+                        try:
+                            st.session_state.final_audio = concatenate_audio_files(st.session_state.audio_list)
+                            with st.container(border=True):
+
+                                # Générer un nom de fichier unique
+                                st.session_state.timestamp = time.strftime("%Y%m%d-%H%M%S")
+                                st.session_state.langues = "_".join([lang["iso-639-1"] for lang in st.session_state.selected_languages])
+                                st.session_state.nom_fichier = f"reponse_audio_{st.session_state.langues}_{st.session_state.timestamp}.mp3"
+
+                                st.audio(st.session_state.final_audio, format="audio/mp3", autoplay=st.session_state.autoplay_tts)
+
+                                st.download_button(
+                                    label=f"📥 {get_translation('telecharger_audio')}",
+                                    data=st.session_state.final_audio,
+                                    file_name=st.session_state.nom_fichier,
+                                    mime="audio/mp3",
+                                    use_container_width=True,
+                                    type="primary",
+                                    key=f"download_button_{st.session_state.langues}_{st.session_state.timestamp}",
+                                )
+
+                                audio_status.update(label=f"{get_translation('concatenation_audio_terminee')}", state="complete", expanded=True)
+                        except Exception as e:
+                            st.error(f"{get_translation('erreur_concatenation_audio')} : {str(e)}")
+                            
+                            audio_status.update(label=f"{get_translation('erreur_concatenation_audio')} : {str(e)}", state="error", expanded=True)
                         
-                        audio_status.update(label=f"{get_translation('erreur_concatenation_audio')} : {str(e)}", state="error", expanded=True)
-                    
 
 
 def clear_inputs_garbages(sessions_state_list: Optional[list] = 
@@ -446,11 +471,10 @@ def clear_inputs_garbages(sessions_state_list: Optional[list] =
         delete_session_state_var(it_var_name)
 
 
-#clear_inputs_garbages()
 
+main_page()
 
 
 
-main_page()