Update functions.py

functions.py

@@ -8,6 +8,10 @@ import wikipedia
 import json
 import streamlit as st
 import requests
+from docx import Document
+from docx.shared import Pt
+from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
+import io
 
 client = OpenAI()
 
@@ -21,6 +25,39 @@ client = OpenAI()
 
 #     return transcript
 
+def export_to_word(podcast_info):
+    # Create a new Word document
+    doc = Document()
+    doc.add_heading(podcast_info['title'], 0)
+
+    # Adding podcast summary
+    p = doc.add_paragraph()
+    run = p.add_run("Podcast Summary:\n")
+    run.bold = True
+    run.font.size = Pt(12)
+    p.add_run(podcast_info['podcast_summary'])
+
+    # Adding podcast guest details
+    p = doc.add_paragraph()
+    run = p.add_run("\nPodcast Guest:\n")
+    run.bold = True
+    run.font.size = Pt(12)
+    p.add_run(podcast_info['podcast_guest'])
+
+    # Adding key moments
+    p = doc.add_paragraph()
+    run = p.add_run("\nKey Moments:\n")
+    run.bold = True
+    run.font.size = Pt(12)
+    p.add_run(podcast_info['podcast_highlights'])
+
+    # Save the document to a byte stream
+    byte_io = io.BytesIO()
+    doc.save(byte_io)
+    byte_io.seek(0)
+
+    return byte_io
+    
 @st.cache_data
 def load_whisper_api(audio):