... workday start now

core/DetectLanguage.py

@@ -5,13 +5,13 @@
 #import re
 
 from os import getenv
-#from typing import Any
+from typing import Any
 #from typing import Dict
 #from typing import IO
 #from typing import List
-#from typing import Optional
+from typing import Optional
 #from typing import Tuple
-#from typing import Union
+from typing import Union
 #from io import BytesIO
 
 # Third-party libraries
@@ -59,7 +59,11 @@ class PRESET_MODELS:
         else:
             return {"temperature": 0.2, "frequency_penalty": 0.2, "presence_penalty": 0.2}
 
-def detect_language(input_text: str, temperature: float = 0.01, context_window: int = 128) -> str:
+def detect_language(input_text: str, 
+                    temperature: Optional[float] = 0.2, 
+                    context_window: Optional[int] = 128,
+                    model: Optional[str] = "gpt-4o-mini" 
+                    ) -> str:
     """
     Détecte la langue d'un texte donné.
 
@@ -89,7 +93,7 @@ def detect_language(input_text: str, temperature: float = 0.01, context_window:
         #################
 
         response = client.chat.completions.create(
-            model="gpt-4o-mini",
+            model=f"{model}",
             temperature=temperature,
             messages=[
                 {

pages/main.py

@@ -498,7 +498,9 @@ def recorder_released():
                     # Transcrire les 10 premiers minutes audio en texte
                     st.session_state.language_detected = detect_language( 
                         input_text = transcribe_audio(tmp_extract),
-                        temperature = 0.01
+                        temperature = 0.2,
+                        context_window=512,
+                        model="gpt-4o-mini" 
                     )
                     first_ten_minutes_audio = AudioSegment.empty()
 
@@ -600,7 +602,6 @@ def recorder_released():
             #    if 'tmp_file' in locals():
             #        os.unlink(tmp_file.name)
         ##st.session_state.rec_widget.close()
-
     
     #if "rec_widget" in st.session_state:
     #    list_dir_obj = dir(st.session_state.rec_widget)
@@ -724,7 +725,10 @@ def main_page():
             # 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
+                    input_text = st.session_state.user_input, 
+                    temperature = 0.01,
+                    context_window = 512,
+                    model="gpt-4o"
                 )
             
             st.session_state.audio_list = []