ImPolymath commited on
Commit
80b8a6c
·
1 Parent(s): eaf3f5c

minors updates

Browse files
Files changed (1) hide show
  1. pages/main.py +31 -0
pages/main.py CHANGED
@@ -57,6 +57,29 @@ from core.core import init_process_mode
57
  from core.moderation import api_moderation_openai_text
58
  from core.audio_isolation import isolate_audio
59
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
60
  def user_message_moderation_check(user_message: Optional[str] = st.session_state.user_input if isinstance(st.session_state.user_input, str) else None):
61
  # Appeler la fonction de modération
62
  moderation_result = api_moderation_openai_text(user_message)
@@ -471,8 +494,16 @@ def main_page():
471
  # Réinitialiser l'état précédent
472
  st.session_state.full_response = ""
473
 
 
 
 
 
 
 
474
  with st.chat_message("user", avatar="👤"):
 
475
  st.markdown(st.session_state.user_input)
 
476
 
477
  main_page()
478
 
 
57
  from core.moderation import api_moderation_openai_text
58
  from core.audio_isolation import isolate_audio
59
 
60
+
61
+ def detection_langue_du_message_utilisateur(
62
+ user_input: str,
63
+ detect_lang_from_text: Optional[bool] = True,
64
+ ):
65
+ # Traitement du message texte de l'utilisateur
66
+ if (detect_lang_from_text):
67
+ language_detected = detect_language(
68
+ input_text = user_input,
69
+ temperature = 0.01,
70
+ context_window = 512,
71
+ model="gpt-4o-mini"
72
+ )
73
+ return {
74
+ "text":f"{user_input}",
75
+ "language": language_detected.strip()
76
+ }
77
+ else:
78
+ return {
79
+ "text":f"{user_input}",
80
+ "language": "default"
81
+ }
82
+
83
  def user_message_moderation_check(user_message: Optional[str] = st.session_state.user_input if isinstance(st.session_state.user_input, str) else None):
84
  # Appeler la fonction de modération
85
  moderation_result = api_moderation_openai_text(user_message)
 
494
  # Réinitialiser l'état précédent
495
  st.session_state.full_response = ""
496
 
497
+ result_detection_lang_1 = detection_langue_du_message_utilisateur(
498
+ st.session_state.user_input
499
+ )
500
+ user_text_lang = result_detection_lang_1["language"] if (result_detection_lang_1["language"] != "default") else "Unknown"
501
+
502
+
503
  with st.chat_message("user", avatar="👤"):
504
+ st.subheader(f"🌐 {user_text_lang}")
505
  st.markdown(st.session_state.user_input)
506
+
507
 
508
  main_page()
509