Update app.py

app.py

@@ -1,14 +1,17 @@
 # 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
+from fpdf import FPDF
 
 # 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,
@@ -17,7 +20,7 @@ llm = LLM(
     api_key=GEMINI_API_KEY
 )
 
-serper_tool = SerperDevTool(api_key=SERPER_API_KEY)
+serper_tool = SerperDevTool(api_key=SERPERDEV_API_KEY)
 
 # Plan Generator Agent
 class PlanGeneratorAgent(Agent):
@@ -45,10 +48,10 @@ class ContentGeneratorAgent(Agent):
             llm=llm
         )
 
-    def run(self, section):
+    def run(self, section, theme):
         # Generate content logic
-        content = f"Content for {section}"
-        print(content)
+        prompt = f"Generate {section} for a presentation on {theme}."
+        content = self.llm.generate(prompt)
         return content
 
 # PDF Compiler Agent
@@ -60,13 +63,17 @@ class PDFCompilerAgent(Agent):
             llm=llm
         )
 
-    def run(self, content):
+    def run(self, content_dict):
         # Compile PDF logic
-        doc = pdfium.PdfDocument.new()
-        page = doc.new_page(width=612, height=792)
-        canvas = page.get_canvas()
-        # Add content to the page
-        doc.save("presentation.pdf")
+        pdf = FPDF()
+        pdf.add_page()
+        pdf.set_font("Arial", size=12)
+
+        for section, content in content_dict.items():
+            pdf.cell(0, 10, f"{section}", ln=True)
+            pdf.multi_cell(0, 10, f"{content}", ln=True)
+
+        pdf.output("presentation.pdf")
         return "presentation.pdf"
 
 # Main Crew
@@ -80,14 +87,24 @@ class PresentationCrew(Crew):
         )
 
     def run(self, theme):
-        plan = self.delegate("Generate a detailed plan for the presentation on {}".format(theme))
-        sections = plan.split("\n")
-        content = []
-        for section in sections:
-            if section:
-                content.append(self.delegate("Generate content for the presentation section: {}".format(section)))
-        pdf = self.delegate("Compile the following content into a PDF: {}".format("\n".join(content)))
-        return pdf
+        try:
+            # Generate plan
+            plan = self.delegate("Generate a detailed plan for the presentation on {}".format(theme))
+            sections = plan.split("\n")
+            content_dict = {}
+
+            # Generate content for each section
+            for section in sections:
+                if section:
+                    content = self.delegate("Generate content for the presentation section: {}".format(section), theme=theme)
+                    content_dict[section] = content
+
+            # Compile PDF
+            pdf_path = self.delegate("Compile the following content into a PDF: {}".format(content_dict))
+            return pdf_path
+        except Exception as e:
+            print(f"An error occurred: {e}")
+            return None
 
 # Streamlit App
 def get_pdf_download_link(pdf_path):
@@ -104,8 +121,11 @@ def main():
     if st.button("Generate Presentation"):
         crew = PresentationCrew()
         pdf_path = crew.run(theme)
-        st.success("Presentation generated successfully!")
-        st.markdown(get_pdf_download_link(pdf_path), unsafe_allow_html=True)
+        if pdf_path:
+            st.success("Presentation generated successfully!")
+            st.markdown(get_pdf_download_link(pdf_path), unsafe_allow_html=True)
+        else:
+            st.error("Failed to generate presentation.")
 
 if __name__ == "__main__":
     main()