Update app.py

app.py

@@ -1,20 +1,21 @@
+from flask import Flask, render_template, request, jsonify
 import os
 import json
 from textwrap import dedent
-
-import streamlit as st
 from crewai import Agent, Crew, Process, Task
 from crewai_tools import SerperDevTool
 from crewai import LLM
 from typing import List
 
+app = Flask(__name__)
+
 # Configuration des clés API
 os.environ["GEMINI_API_KEY"] = os.environ.get("GEMINI_API_KEY")
 os.environ["SERPER_API_KEY"] = os.environ.get("SERPER_API_KEY")
 
 llm = LLM(
     model="gemini/gemini-1.5-flash",
-    temperature=1,
+    temperature=0.7,
     timeout=120,
     max_tokens=8000,
 )
@@ -49,7 +50,6 @@ writer = Agent(
 def extract_json_from_result(result_text: str) -> list:
     """Extrait le JSON des résultats de la crew."""
     try:
-        # Trouve le début et la fin du JSON dans le texte
         json_start = result_text.find('[')
         json_end = result_text.rfind(']') + 1
         if json_start == -1 or json_end == 0:
@@ -60,7 +60,6 @@ def extract_json_from_result(result_text: str) -> list:
     except (json.JSONDecodeError, ValueError) as e:
         raise ValueError(f"Erreur lors de l'extraction du JSON : {str(e)}")
 
-# Définition des tâches
 def research_task(topic: str) -> Task:
     return Task(
         description=dedent(f"""Effectuer une recherche approfondie sur le sujet '{topic}'.
@@ -79,55 +78,42 @@ def flashcard_creation_task(research_task: Task) -> Task:
         context=[research_task]
     )
 
-# Interface Streamlit
-st.title("🤖 Générateur de Flashcards avec CrewAI")
-
-topic = st.text_input("Entrez le sujet des flashcards:", "Intelligence Artificielle")
+@app.route('/')
+def index():
+    return render_template('index.html')
 
-if st.button("Générer les Flashcards"):
+@app.route('/generate', methods=['POST'])
+def generate_flashcards():
+    topic = request.json.get('topic')
     if not topic:
-        st.error("Veuillez entrer un sujet.")
-    else:
-        with st.spinner('Création des flashcards en cours...'):
-            try:
-                # Création des tâches
-                research = research_task(topic)
-                flashcard_creation = flashcard_creation_task(research)
-
-                # Création de la crew
-                crew = Crew(
-                    agents=[researcher, writer],
-                    tasks=[research, flashcard_creation],
-                    process=Process.sequential,
-                    verbose=True
-                )
-
-                # Exécution de la crew
-                result = crew.kickoff()
-                
-                # 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)
-                    
-                    # Affichage des flashcards
-                    st.success("Flashcards générées avec succès !")
-                    
-                    # Création d'onglets pour différents modes d'affichage
-                    tab1, tab2 = st.tabs(["Mode Étude", "Mode JSON"])
-                    
-                    with tab1:
-                        for i, card in enumerate(flashcards, 1):
-                            with st.expander(f"Flashcard {i}: {card['question']}", expanded=False):
-                                st.write("**Réponse:**", card['answer'])
-                    
-                    with tab2:
-                        st.json(flashcards)
-                
-                else:
-                    st.error("Aucun résultat généré par la crew.")
-                    
-            except Exception as e:
-                st.error(f"Une erreur est survenue : {str(e)}")
-                st.write("Résultat brut pour débogage :")
-                st.write(result)
+        return jsonify({'error': 'Veuillez entrer un sujet.'}), 400
+
+    try:
+        # Création des tâches
+        research = research_task(topic)
+        flashcard_creation = flashcard_creation_task(research)
+
+        # Création de la crew
+        crew = Crew(
+            agents=[researcher, writer],
+            tasks=[research, flashcard_creation],
+            process=Process.sequential,
+            verbose=True
+        )
+
+        # Exécution de la crew
+        result = crew.kickoff()
+        
+        # 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})
+        else:
+            return jsonify({'error': 'Aucun résultat généré par la crew.'}), 500
+
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+if __name__ == '__main__':
+    app.run(debug=True)