Update app.py

app.py

@@ -5,6 +5,7 @@ import io
 from pydub import AudioSegment
 from youtube_transcript_api import YouTubeTranscriptApi
 import PyPDF2
+from pptx import Presentation
 
 # Set OpenAI API key
 openai.api_key = os.getenv('OPENAI_API_KEY')
@@ -87,6 +88,37 @@ def generate_section(title, text):
     )
     return response.choices[0].message.content.strip()
 
+# Function to create PowerPoint presentation
+def create_presentation(summary, key_concepts, key_takeaways, case_studies, glossary, faqs):
+    prs = Presentation()
+
+    # Add title slide
+    slide = prs.slides.add_slide(prs.slide_layouts[0])
+    title = slide.shapes.title
+    subtitle = slide.placeholders[1]
+    title.text = "Lecture Notes"
+    subtitle.text = "Generated by AI Notes Generation Bot"
+
+    # Add slides for each section
+    def add_slide(title, content):
+        slide = prs.slides.add_slide(prs.slide_layouts[1])
+        slide.shapes.title.text = title
+        textbox = slide.placeholders[1]
+        textbox.text = content
+
+    add_slide("Summary", summary)
+    add_slide("Key Concepts", key_concepts)
+    add_slide("Key Takeaways", key_takeaways)
+    add_slide("Case Studies/Examples", case_studies)
+    add_slide("Glossary", glossary)
+    add_slide("FAQs", faqs)
+
+    # Save the presentation to a BytesIO object
+    ppt_buffer = io.BytesIO()
+    prs.save(ppt_buffer)
+    ppt_buffer.seek(0)
+    return ppt_buffer
+
 # Main app
 def main():
     st.set_page_config(layout="wide")
@@ -154,23 +186,31 @@ def main():
                     f"<p>{key_takeaways}</p>" +
                     "</div>", unsafe_allow_html=True)
 
-        st.markdown("### Case Studies/Examples:")
-        case_studies = generate_section("Case Studies/Examples", summary)
-        st.markdown(f"<div style='background-color: black; color: white; padding: 10px; border-radius: 5px;'>" +
-                    f"<p>{case_studies}</p>" +
+                case_studies = generate_section("Case Studies/Examples", summary)
+        st.markdown(f"<div style='background-color: black; color: white; padding: 10px; border-radius: 5px;'>"
+                    f"<p>{case_studies}</p>"
                     "</div>", unsafe_allow_html=True)
 
         st.markdown("### Glossary:")
         glossary = generate_section("Glossary", summary)
-        st.markdown(f"<div style='background-color: black; color: white; padding: 10px; border-radius: 5px;'>" +
-                    f"<p>{glossary}</p>" +
+        st.markdown(f"<div style='background-color: black; color: white; padding: 10px; border-radius: 5px;'>"
+                    f"<p>{glossary}</p>"
                     "</div>", unsafe_allow_html=True)
 
         st.markdown("### FAQs:")
         faqs = generate_section("FAQs", summary)
-        st.markdown(f"<div style='background-color: black; color: white; padding: 10px; border-radius: 5px;'>" +
-                    f"<p>{faqs}</p>" +
+        st.markdown(f"<div style='background-color: black; color: white; padding: 10px; border-radius: 5px;'>"
+                    f"<p>{faqs}</p>"
                     "</div>", unsafe_allow_html=True)
 
+        # Option to download the PowerPoint presentation
+        ppt_buffer = create_presentation(summary, key_concepts, key_takeaways, case_studies, glossary, faqs)
+        st.download_button(
+            label="Download Presentation",
+            data=ppt_buffer,
+            file_name="Lecture_Notes_Presentation.pptx",
+            mime="application/vnd.openxmlformats-officedocument.presentationml.presentation"
+        )
+
 if __name__ == "__main__":
     main()