Update app.py

app.py

@@ -296,19 +296,58 @@ def api_francais():
     def generate_stream():
         try:
             logger.info("Démarrage du streaming de génération...")
+            content_generated = False
+            chunk_count = 0
             
             for chunk in client.models.generate_content_stream(
                 model=model_name,
                 contents=[user_prompt],
                 config=config
             ):
+                chunk_count += 1
+                
+                # Vérifier si la réponse a été bloquée
+                if hasattr(chunk, 'prompt_feedback') and chunk.prompt_feedback:
+                    logger.warning(f"Prompt feedback: {chunk.prompt_feedback}")
+                    if hasattr(chunk.prompt_feedback, 'block_reason'):
+                        yield "data: " + json.dumps({
+                            'type': 'error',
+                            'content': f'Contenu bloqué par les filtres de sécurité: {chunk.prompt_feedback.block_reason}'
+                        }) + "\n\n"
+                        return
+                
                 if not chunk.candidates:
+                    logger.warning(f"Chunk {chunk_count} sans candidats")
+                    continue
+                
+                candidate = chunk.candidates[0]
+                
+                # Vérifier le statut du candidat
+                if hasattr(candidate, 'finish_reason') and candidate.finish_reason:
+                    finish_reason = str(candidate.finish_reason)
+                    if finish_reason != 'STOP' and finish_reason != '0':
+                        logger.warning(f"Génération terminée avec raison: {finish_reason}")
+                        if 'SAFETY' in finish_reason:
+                            yield "data: " + json.dumps({
+                                'type': 'error',
+                                'content': 'Contenu bloqué par les filtres de sécurité.'
+                            }) + "\n\n"
+                            return
+                
+                if not hasattr(candidate, 'content') or not candidate.content:
+                    logger.warning(f"Chunk {chunk_count} sans contenu")
+                    continue
+                
+                if not hasattr(candidate.content, 'parts') or not candidate.content.parts:
+                    logger.warning(f"Chunk {chunk_count} sans parts")
                     continue
                     
-                for part in chunk.candidates[0].content.parts:
-                    if not part.text:
+                for part in candidate.content.parts:
+                    if not hasattr(part, 'text') or not part.text:
                         continue
-                        
+                    
+                    content_generated = True
+                    
                     if hasattr(part, 'thought') and part.thought:
                         # Pensées du modèle (mode deepthink)
                         yield "data: " + json.dumps({
@@ -322,8 +361,16 @@ def api_francais():
                             'content': part.text
                         }) + "\n\n"
 
-            logger.info("Fin du streaming de génération.")
-            yield "data: " + json.dumps({'type': 'done'}) + "\n\n"
+            logger.info(f"Fin du streaming. Chunks traités: {chunk_count}, Contenu généré: {content_generated}")
+            
+            if not content_generated:
+                logger.error("Aucun contenu n'a été généré!")
+                yield "data: " + json.dumps({
+                    'type': 'error',
+                    'content': 'Aucun contenu généré. Le modèle n\'a pas produit de réponse.'
+                }) + "\n\n"
+            else:
+                yield "data: " + json.dumps({'type': 'done'}) + "\n\n"
 
         except Exception as e:
             logger.exception("Erreur pendant la génération de contenu.")