Update app.py

app.py

@@ -1,143 +1,90 @@
-# presentation_generator.py
-
 import os
-from crewai import Crew, LLM, Agent
-from crewai_tools import Tool, SerperDevTool
-import streamlit as st
-import base64
-# import pypdfium2 as pdfium # Removed
-# from fpdf import FPDF # Removed
-
 
-# Replace with your actual API keys
-GEMINI_API_KEY = "AIzaSyD6yZxfVOnh63GXBJjakAupk9aP4CZrgrQ"
-SERPER_API_KEY = "9b90a274d9e704ff5b21c0367f9ae1161779b573"
-
-llm = LLM(
-    model="gemini/gemini-1.5-flash",
-    temperature=0.7,
-    timeout=120,
-    max_tokens=8000,
-    api_key=GEMINI_API_KEY
+import streamlit as st
+from crewai import Agent, Crew, Process, Task
+from crewai.tools import SerperDevTool
+from langchain_google_genai import ChatGoogleGenerativeAI
+
+# Assurez-vous que vos clés API sont définies dans les variables d'environnement
+os.environ["OPENAI_API_KEY"] = st.secrets["OPENAI_API_KEY"]
+os.environ["SERPER_API_KEY"] = st.secrets["SERPER_API_KEY"]
+
+# Définition des agents
+project_manager = Agent(
+    role="Chef de Projet",
+    goal="Coordonner les autres agents pour assurer la cohérence et la qualité de l'exposé.",
+    backstory="Un chef de projet expérimenté avec une expertise dans la gestion d'équipes et la coordination de projets complexes.",
+    llm=ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=os.environ["GOOGLE_API_KEY"]),
+    verbose=True,
+    allow_delegation=True,
 )
 
-serper_tool = SerperDevTool(api_key=SERPER_API_KEY)
-
-# Plan Generator Agent
-class PlanGeneratorAgent(Agent):
-    def __init__(self):
-        super().__init__(
-            name="PlanGenerator",
-            role="Presentation Planner",
-            goal="Create detailed and well-structured presentation outlines",
-            backstory="""You are an experienced presentation strategist who excels at 
-            creating logical and impactful presentation structures. You understand how 
-            to organize information for maximum audience engagement.""",
-            llm=llm,
-            tools=[serper_tool]
-        )
-
-    def run(self, theme):
-        research = serper_tool.search(query=f"Best practices for {theme} presentations")
-        plan = "Introduction\nMain Points\nConclusion"
-        return plan
-
-# Content Generator Agent
-class ContentGeneratorAgent(Agent):
-    def __init__(self):
-        super().__init__(
-            name="ContentGenerator",
-            role="Content Creator",
-            goal="Generate engaging and informative presentation content",
-            backstory="""You are a creative content specialist with expertise in 
-            transforming ideas into compelling presentation material. You know how 
-            to balance information with engagement.""",
-            llm=llm
-        )
-
-    def run(self, section, theme):
-        prompt = f"Generate {section} for a presentation on {theme}."
-        content = self.llm.generate(prompt)
-        return content
-
-# PDF Compiler Agent # Removed
-# class PDFCompilerAgent(Agent):
-#     def __init__(self):
-#         super().__init__(
-#             name="PDFCompiler",
-#             role="Document Formatter",
-#             goal="Create professional and well-formatted PDF presentations",
-#             backstory="""You are a document formatting expert who knows how to 
-#             transform content into visually appealing and professional PDFs. You 
-#             understand typography, layout, and document structure.""",
-#             llm=llm
-#         )
-
-#     def run(self, content_dict):
-#         pdf = FPDF()
-#         pdf.add_page()
-#         pdf.set_font("Arial", size=12)
+topic_analyst = Agent(
+    role="Analyste de Thème",
+    goal="Analyser le thème de l'exposé et générer un plan détaillé.",
+    backstory="Un expert en recherche et en structuration de contenu, capable de décomposer des sujets complexes en plans clairs et concis.",
+    llm=ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=os.environ["GOOGLE_API_KEY"]),
+    verbose=True,
+    tools=[SerperDevTool()],
+)
 
-#         for section, content in content_dict.items():
-#             pdf.cell(0, 10, f"{section}", ln=True)
-#             pdf.multi_cell(0, 10, f"{content}", ln=True)
+content_writer = Agent(
+    role="Rédacteur de Contenu",
+    goal="Rédiger les sections de l'exposé en se basant sur le plan et les recherches.",
+    backstory="Un rédacteur spécialisé dans la création de contenu clair, concis et engageant.",
+    llm=ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=os.environ["GOOGLE_API_KEY"]),
+    verbose=True,
+    tools=[SerperDevTool()],
+)
 
-#         pdf.output("presentation.pdf")
-#         return "presentation.pdf"
+editor = Agent(
+    role="Éditeur/Réviseur",
+    goal="Réviser et peaufiner le contenu, assurer la cohérence du style et du ton, corriger les erreurs.",
+    backstory="Un éditeur expérimenté avec un souci du détail et une excellente maîtrise de la langue.",
+    llm=ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=os.environ["GOOGLE_API_KEY"]),
+    verbose=True,
+)
 
-# Main Crew
-class PresentationCrew(Crew):
-    def __init__(self):
-        super().__init__(
-            name="PresentationGenerator",
-            description="Generates a high-quality presentation based on a given theme.",
-            llm=llm,
-            agents=[PlanGeneratorAgent(), ContentGeneratorAgent()] # Removed PDFCompilerAgent
+# Définition des tâches
+def assign_tasks(topic):
+    analyze_topic_task = Task(
+        description=f"Analyser le thème '{topic}' et générer un plan détaillé pour l'exposé, divisé en sections claires.",
+        expected_output="Un plan détaillé de l'exposé au format markdown.",
+        agent=topic_analyst,
+    )
+
+    write_content_task = Task(
+        description=f"Rédiger les sections de l'exposé en se basant sur le plan généré par l'analyste de thème et sur des recherches approfondies.",
+        expected_output="Contenu des sections au format markdown.",
+        agent=content_writer,
+        context=[analyze_topic_task],
+    )
+
+    edit_content_task = Task(
+        description="Réviser et corriger l'ensemble du contenu de l'exposé, en s'assurant de la cohérence, de la clarté et de l'absence d'erreurs.",
+        expected_output="Contenu de l'exposé révisé et corrigé au format markdown.",
+        agent=editor,
+        context=[write_content_task],
+    )
+
+    return [analyze_topic_task, write_content_task, edit_content_task]
+
+# Configuration de l'interface Streamlit
+st.title("🤖 Générateur d'Exposés avec CrewAI")
+
+topic = st.text_input("Entrez le thème de l'exposé:", "L'Intelligence Artificielle en 2024")
+
+if st.button("Générer l'exposé"):
+    with st.spinner("Création de l'équipe d'agents..."):
+        crew = Crew(
+            agents=[project_manager, topic_analyst, content_writer, editor],
+            tasks=assign_tasks(topic),
+            process=Process.hierarchical,
+            verbose=True,
         )
 
-    def run(self, theme):
-        try:
-            plan_agent = self.agents[0]
-            content_agent = self.agents[1]
-            
-            plan = plan_agent.run(theme)
-            sections = plan.split("\n")
-            content_dict = {}
-
-            for section in sections:
-                if section:
-                    content = content_agent.run(section, theme)
-                    content_dict[section] = content
-
-            print(content_dict)
-            return content_dict
-        except Exception as e:
-            print(f"An error occurred: {e}")
-            return None
-
-# Streamlit App Functions
-# def get_pdf_download_link(pdf_path): # Removed
-#     with open(pdf_path, "rb") as f:
-#         pdf_bytes = f.read()
-#     encoded = base64.b64encode(pdf_bytes).decode()
-#     return f'<a href="data:application/pdf;base64,{encoded}" download="presentation.pdf">Download PDF</a>'
-
-def main():
-    st.title("Advanced Presentation Generator")
-
-    theme = st.text_input("Enter the presentation theme:")
-
-    if st.button("Generate Presentation"):
-        crew = PresentationCrew()
-        presentation_content = crew.run(theme)
-        if presentation_content:
-            st.success("Presentation generated successfully!")
-            for section, content in presentation_content.items():
-                st.subheader(section)
-                st.write(content)
-                st.markdown("---")  # Separator between sections
-        else:
-            st.error("Failed to generate presentation.")
+    with st.spinner("Génération de l'exposé..."):
+        result = crew.kickoff()
 
-if __name__ == "__main__":
-    main()
+        st.success("Exposé généré avec succès!")
+        st.markdown(result)