Promotingai commited on
Commit
92282e9
·
verified ·
1 Parent(s): 208ff39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -39,7 +39,15 @@ if prompt:
39
  )
40
 
41
  # Récupérer le contenu de la réponse
42
- msg = response.content
 
 
 
 
 
 
 
 
43
 
44
  # Ajouter et afficher la réponse de l'assistant
45
  st.session_state.messages.append({"role": "assistant", "content": msg})
 
39
  )
40
 
41
  # Récupérer le contenu de la réponse
42
+ if isinstance(response.content, list):
43
+ # Supposons que tous les éléments de la liste sont des objets TextBlock
44
+ msg = " ".join(block.text for block in response.content)
45
+ elif isinstance(response.content, dict) and 'text' in response.content:
46
+ # Supposons que la réponse soit un dictionnaire avec une clé 'text'
47
+ msg = response.content['text']
48
+ else:
49
+ # Si ce n'est ni une liste ni un dictionnaire avec 'text'
50
+ msg = str(response.content)
51
 
52
  # Ajouter et afficher la réponse de l'assistant
53
  st.session_state.messages.append({"role": "assistant", "content": msg})