Update playground.py

tabs/playground.py

@@ -32,23 +32,25 @@ def display_proba_next_token(prompt,temperature,max_token):
         probabilities = {key: np.exp(value) for key, value in logprobs_dict.items()}
 
         # Afficher le texte de la réponse
-        st.write("Token suivant : ->"+list(probabilities.keys())[0]+"<-")  # Afficher le texte de la réponse
+        st.write("**Token suivant :** ->**:red["+list(probabilities.keys())[0]+"]**<-")  # Afficher le texte de la réponse
         # st.write("Texte de la réponse : ->"+response['choices'][0]['text'][0]+"<-")  # Afficher le texte de la réponse
         st.write("")
 
         # Afficher les probabilités pour le prochain token
-        for token, proba in probabilities.items():
-            st.write(f"Token: ->{token}<- ; proba: {proba:.2f}")
+        st.write("**"+tr("Probabilité d'apparition des token:")+"**")
+        for i,(token, proba) in enumerate(probabilities.items()):
+            st.write(f"&nbsp;&nbsp;&nbsp;&nbsp;{i+1} - Token: ->**:red[{token}]**<- ; proba: {proba:.0%}")
 
         # Afficher le texte de la réponse complete
         st.write("")
         try:
             text = response['choices'][0]['text']
-            st.write("Texte de la réponse complète: ->" + str(text) + "<-")
+            st.write("**"+tr("Texte de la réponse complète")+":**")
+            st.write("->**:red[" + str(text) + "]**<-")
         except KeyError:
-            st.error("La réponse ne contient pas de texte supplémentaire.")
+            st.error(tr("La réponse ne contient pas de texte supplémentaire."))
     else:
-        st.error("La réponse ne contient pas de texte supplémentaire.")
+        st.error(tr("La réponse ne contient pas de texte supplémentaire."))
     # st.write("Texte de la réponse complete: ->"+str(response['choices'][0]['text'])+"<-")  # Afficher le texte de la réponse
     # st.write("")
     # st.write(response)
@@ -64,12 +66,14 @@ def run():
     st.markdown('''
                 ---
                 ''')
+
     st.header("**"+tr("Prédiction du token suivant")+"**")
     st.markdown(tr(
         """
-        Cet espace a pour objectif d'observer la génération de token par ChatGPT'
+        Cet espace a pour objectif d'observer la génération de token par ChatGPT  
         """)
     , unsafe_allow_html=True)
+    st.write("")
     if 'OPENAI_API_KEY' in st.session_state:
         try:
             openai.api_key = st.session_state['OPENAI_API_KEY']