Update app.py

app.py

@@ -54,41 +54,48 @@ editor = Agent(
     verbose=True,
 )
 
+def parse_sections(crew_output) -> List[str]:
+    """Parse the sections from CrewOutput object."""
+    try:
+        # Extraire le raw output
+        raw_output = crew_output.raw
+        # Nettoyer la string (enlever les guillemets externes s'ils existent)
+        cleaned_output = raw_output.strip('"\'')
+        # Évaluer la string comme une liste Python
+        sections = eval(cleaned_output)
+        return sections
+    except Exception as e:
+        st.error(f"Erreur lors du parsing des sections: {e}")
+        st.error(f"Raw output: {crew_output.raw}")
+        return []
+
 def assign_tasks(topic: str) -> List[Task]:
     # Tâche d'analyse qui retourne une liste de sections
     analyze_topic_task = Task(
         description=f"""Analyser le thème '{topic}' et générer un plan détaillé pour l'exposé.
-        IMPORTANT: Ta réponse doit être UNIQUEMENT une liste Python sous forme de string.
-        Exemple de réponse: "['Introduction', 'Historique', 'État actuel', 'Perspectives futures', 'Conclusion']"
+        IMPORTANT: Ta réponse doit être UNIQUEMENT une liste Python.
+        Exemple de réponse: ['Introduction', 'Historique', 'État actuel', 'Perspectives futures', 'Conclusion']
         Ne pas ajouter d'autres explications ou texte.""",
-        expected_output="Une liste Python sous forme de string contenant les sections de l'exposé",
+        expected_output="Une liste Python contenant les sections de l'exposé",
         agent=topic_analyst
     )
     
     tasks = [analyze_topic_task]
     
     # Création dynamique des tâches d'écriture pour chaque section
-    def create_writing_tasks(sections_result: str) -> List[Task]:
-        try:
-            # Conversion de la string en liste Python
-            sections = ast.literal_eval(sections_result)
-            writing_tasks = []
-            
-            for section in sections:
-                write_task = Task(
-                    description=f"""Rédiger la section '{section}' de l'exposé sur le thème '{topic}'.
-                    Concentre-toi uniquement sur cette section spécifique.
-                    Utilise un style académique et professionnel.""",
-                    expected_output=f"Contenu détaillé de la section '{section}' au format markdown",
-                    agent=content_writer,
-                    context=[analyze_topic_task]
-                )
-                writing_tasks.append(write_task)
-            
-            return writing_tasks
-        except (ValueError, SyntaxError) as e:
-            st.error(f"Erreur lors de la conversion du plan en liste: {e}")
-            return []
+    def create_writing_tasks(sections: List[str]) -> List[Task]:
+        writing_tasks = []
+        for section in sections:
+            write_task = Task(
+                description=f"""Rédiger la section '{section}' de l'exposé sur le thème '{topic}'.
+                Concentre-toi uniquement sur cette section spécifique.
+                Utilise un style académique et professionnel.""",
+                expected_output=f"Contenu détaillé de la section '{section}' au format markdown",
+                agent=content_writer,
+                context=[analyze_topic_task]
+            )
+            writing_tasks.append(write_task)
+        return writing_tasks
     
     # Tâche d'édition finale
     def create_editing_task(writing_tasks: List[Task]) -> Task:
@@ -122,28 +129,35 @@ if st.button("Générer l'exposé"):
             manager_llm=llm,
             verbose=True,
         )
-        sections_result = crew_analysis.kickoff()
-        
-        # Création des tâches d'écriture basées sur le résultat de l'analyse
-        writing_tasks = task_creators['create_writing_tasks'](sections_result)
+        analysis_result = crew_analysis.kickoff()
         
-        # Création de la tâche d'édition
-        editing_task = task_creators['create_editing_task'](writing_tasks)
+        # Parsing des sections depuis le résultat
+        sections = parse_sections(analysis_result)
+        st.write("Sections identifiées:", sections)
         
-        # Toutes les tâches dans l'ordre
-        all_tasks = [task_creators['analyze_task']] + writing_tasks + [editing_task]
-        
-        # Création et exécution du crew final
-        crew = Crew(
-            agents=[project_manager, topic_analyst, content_writer, editor],
-            tasks=all_tasks,
-            process=Process.hierarchical,
-            manager_llm=llm,
-            verbose=True,
-        )
+        if sections:
+            # Création des tâches d'écriture basées sur les sections
+            writing_tasks = task_creators['create_writing_tasks'](sections)
+            
+            # Création de la tâche d'édition
+            editing_task = task_creators['create_editing_task'](writing_tasks)
+            
+            # Toutes les tâches dans l'ordre
+            all_tasks = [task_creators['analyze_task']] + writing_tasks + [editing_task]
+            
+            # Création et exécution du crew final
+            crew = Crew(
+                agents=[project_manager, topic_analyst, content_writer, editor],
+                tasks=all_tasks,
+                process=Process.hierarchical,
+                manager_llm=llm,
+                verbose=True,
+            )
 
-    with st.spinner("Génération de l'exposé..."):
-        result = crew.kickoff()
-        
-        st.success("Exposé généré avec succès!")
-        st.markdown(result)
+            with st.spinner("Génération de l'exposé..."):
+                result = crew.kickoff()
+                
+                st.success("Exposé généré avec succès!")
+                st.markdown(result)
+        else:
+            st.error("Impossible de continuer sans une liste valide de sections.")