Update app.py

app.py

@@ -47,6 +47,17 @@ writer = Agent(
     allow_delegation=False
 )
 
+quiz_creator = Agent(
+    role='Créateur de Quiz',
+    goal=dedent("""Créer des questions de quiz à choix multiples basées sur les informations 
+    fournies par le Chercheur."""),
+    backstory=dedent("""Un expert en évaluation pédagogique spécialisé dans la création 
+    de questions de quiz engageantes et éducatives avec différents niveaux de difficulté."""),
+    llm=llm,
+    verbose=True,
+    allow_delegation=False
+)
+
 def extract_json_from_result(result_text: str) -> list:
     """Extrait le JSON des résultats de la crew."""
     try:
@@ -72,31 +83,81 @@ def flashcard_creation_task(research_task: Task) -> Task:
     return Task(
         description=dedent("""Transformer les informations fournies par le Chercheur
         en une série de flashcards au format JSON. Je veux une vingtaine de flashcard robuste et complet avec des questions et des réponses détailler. Chaque flashcard doit avoir une question
-        d'un côté et une réponse concise de l'autre."""),
+        d'un côté et une réponse concise de l'autre.
+        
+        Format JSON attendu:
+        [
+            {
+                "question": "Question 1?",
+                "answer": "Réponse 1"
+            },
+            {
+                "question": "Question 2?",
+                "answer": "Réponse 2"
+            }
+        ]
+        """),
         expected_output="Une liste de flashcards au format JSON.",
         agent=writer,
         context=[research_task]
     )
 
+def quiz_creation_task(research_task: Task) -> Task:
+    return Task(
+        description=dedent("""Transformer les informations fournies par le Chercheur
+        en une série de questions de quiz à choix multiples au format JSON.
+        Créer 10 questions de quiz avec quatre options de réponse pour chaque question,
+        dont une seule est correcte. Inclure une explication pour chaque réponse correcte.
+        
+        Format JSON attendu:
+        [
+            {
+                "question": "Question 1?",
+                "options": ["Option A", "Option B", "Option C", "Option D"],
+                "correctAnswer": "Option A",
+                "explanation": "Explication de pourquoi cette réponse est correcte"
+            },
+            {
+                "question": "Question 2?",
+                "options": ["Option A", "Option B", "Option C", "Option D"],
+                "correctAnswer": "Option C",
+                "explanation": "Explication de pourquoi cette réponse est correcte"
+            }
+        ]
+        """),
+        expected_output="Une liste de questions de quiz au format JSON.",
+        agent=quiz_creator,
+        context=[research_task]
+    )
+
 @app.route('/')
 def index():
     return render_template('index.html')
 
 @app.route('/generate', methods=['POST'])
-def generate_flashcards():
+def generate_content():
     topic = request.json.get('topic')
+    content_type = request.json.get('type', 'flashcards')  # 'flashcards' ou 'quiz'
+    
     if not topic:
         return jsonify({'error': 'Veuillez entrer un sujet.'}), 400
 
     try:
-        # Création des tâches
+        # Création de la tâche de recherche
         research = research_task(topic)
-        flashcard_creation = flashcard_creation_task(research)
-
+        
+        # Sélection de la tâche en fonction du type demandé
+        if content_type == 'quiz':
+            creation_task = quiz_creation_task(research)
+            agents = [researcher, quiz_creator]
+        else:  # Par défaut: flashcards
+            creation_task = flashcard_creation_task(research)
+            agents = [researcher, writer]
+        
         # Création de la crew
         crew = Crew(
-            agents=[researcher, writer],
-            tasks=[research, flashcard_creation],
+            agents=agents,
+            tasks=[research, creation_task],
             process=Process.sequential,
             verbose=True
         )
@@ -107,13 +168,13 @@ def generate_flashcards():
         # Extraction du JSON depuis le dernier résultat de tâche
         if result.tasks_output and len(result.tasks_output) > 0:
             last_task_output = result.tasks_output[-1].raw
-            flashcards = extract_json_from_result(last_task_output)
-            return jsonify({'success': True, 'flashcards': flashcards})
+            content_data = extract_json_from_result(last_task_output)
+            return jsonify({'success': True, content_type: content_data})
         else:
             return jsonify({'error': 'Aucun résultat généré par la crew.'}), 500
 
     except Exception as e:
-        return jsonify({'error': 'error'}), 500
+        return jsonify({'error': str(e)}), 500
 
 if __name__ == '__main__':
     app.run(debug=True)