Update app.py

app.py

@@ -5,8 +5,9 @@ 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
+# import pypdfium2 as pdfium # Removed
+# from fpdf import FPDF # Removed
+
 
 # Replace with your actual API keys
 GEMINI_API_KEY = "AIzaSyD6yZxfVOnh63GXBJjakAupk9aP4CZrgrQ"
@@ -59,30 +60,30 @@ class ContentGeneratorAgent(Agent):
         content = self.llm.generate(prompt)
         return content
 
-# PDF Compiler Agent
-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)
-
-        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"
+# 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)
+
+#         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
 class PresentationCrew(Crew):
@@ -91,32 +92,34 @@ class PresentationCrew(Crew):
             name="PresentationGenerator",
             description="Generates a high-quality presentation based on a given theme.",
             llm=llm,
-            agents=[PlanGeneratorAgent(), ContentGeneratorAgent(), PDFCompilerAgent()]
+            agents=[PlanGeneratorAgent(), ContentGeneratorAgent()] # Removed PDFCompilerAgent
         )
 
     def run(self, theme):
         try:
-            plan = self.delegate("Generate a detailed plan for the presentation on {}".format(theme))
+            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 = self.delegate("Generate content for the presentation section: {}".format(section), theme=theme)
+                    content = content_agent.run(section, theme)
                     content_dict[section] = content
 
-            pdf_path = self.delegate("Compile the following content into a PDF: {}".format(content_dict))
-            return pdf_path
+            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):
-    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 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")
@@ -125,10 +128,13 @@ def main():
 
     if st.button("Generate Presentation"):
         crew = PresentationCrew()
-        pdf_path = crew.run(theme)
-        if pdf_path:
+        presentation_content = crew.run(theme)
+        if presentation_content:
             st.success("Presentation generated successfully!")
-            st.markdown(get_pdf_download_link(pdf_path), unsafe_allow_html=True)
+            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.")