Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
-from io import BytesIO
+import tempfile
 
 # Initialize the content generation agent with DuckDuckGo tool
 content_agent = CodeAgent(
@@ -22,22 +22,20 @@ def enhance_content(content):
 
 # Function to save content as a downloadable text file
 def generate_text_download(content, filename="generated_blog.txt"):
-    b = BytesIO()
-    b.write(content.encode())
-    b.seek(0)
-    return b, filename
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".txt") as tmp_file:
+        tmp_file.write(content.encode())
+        return tmp_file.name, filename
 
 # Function to save content as a downloadable PDF
 def generate_pdf_download(content, filename="generated_blog.pdf"):
     from fpdf import FPDF
-    pdf = FPDF()
-    pdf.add_page()
-    pdf.set_font("Arial", size=12)
-    pdf.multi_cell(0, 10, content)
-    b = BytesIO()
-    pdf.output(b)
-    b.seek(0)
-    return b, filename
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
+        pdf = FPDF()
+        pdf.add_page()
+        pdf.set_font("Arial", size=12)
+        pdf.multi_cell(0, 10, content)
+        pdf.output(tmp_file.name)
+        return tmp_file.name, filename
 
 # Streamlit app title
 st.title("SmolAgent Content Writer")
@@ -103,12 +101,13 @@ with tabs[2]:
         content = st.session_state["last_content"]
 
         # Text file download
-        text_file, text_filename = generate_text_download(content)
-        st.download_button(label="Download as Text File", data=text_file, file_name=text_filename, mime="text/plain")
+        text_file_path, text_filename = generate_text_download(content)
+        with open(text_file_path, "rb") as file:
+            st.download_button(label="Download as Text File", data=file, file_name=text_filename, mime="text/plain")
 
         # PDF file download
-        pdf_file, pdf_filename = generate_pdf_download(content)
-        st.download_button(label="Download as PDF", data=pdf_file, file_name=pdf_filename, mime="application/pdf")
+        pdf_file_path, pdf_filename = generate_pdf_download(content)
+        with open(pdf_file_path, "rb") as file:
+            st.download_button(label="Download as PDF", data=file, file_name=pdf_filename, mime="application/pdf")
     else:
-        st.write("No content available for download. Generate content first.")
-
+        st.write("No content available for download. Generate content first.")