Update app.py

app.py

@@ -1,8 +1,9 @@
 import streamlit as st
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
+from io import BytesIO
 
-# Initialize the blog-writing agent with DuckDuckGo tool
-blog_agent = CodeAgent(
+# Initialize the content generation agent with DuckDuckGo tool
+content_agent = CodeAgent(
     tools=[DuckDuckGoSearchTool()], 
     model=HfApiModel()
 )
@@ -15,61 +16,99 @@ def display_agent_activity(prompt, result):
     st.write("**Agent Output:**")
     st.code(result, language="text")
 
-# Function to save generated content to a file
-def save_to_file(content, filename="generated_blog.txt"):
-    with open(filename, "w") as file:
-        file.write(content)
-    return filename
-
 # Function to enhance generated content
 def enhance_content(content):
     return f"{content}\n\n---\nGenerated with insights and enhanced formatting."
 
+# 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
+
+# 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
+
 # Streamlit app title
-st.title("Smolagnets-based Blog Writer")
-
-# App description
-st.write("Generate high-quality blog content enriched with real-time insights.")
-
-# Input field for blog topic or prompt
-blog_prompt = st.text_area("Enter your blog topic or prompt:", placeholder="E.g., The Future of AI in Healthcare")
-
-# Input field for choosing content tone
-tone = st.selectbox("Choose the tone of the content:", ["Professional", "Casual", "Persuasive", "Informative"])
-
-# Option to include a summary
-include_summary = st.checkbox("Include a summary of the generated content")
-
-# Button to trigger blog content generation
-if st.button("Generate Blog Content"):
-    if blog_prompt:
-        with st.spinner("Generating blog content, please wait..."):
-            try:
-                # Generate blog content using the agent
-                blog_result = blog_agent.run(f"{blog_prompt}\nTone: {tone}")
-                
-                # Optionally enhance the content
-                blog_result = enhance_content(blog_result)
-
-                # Display the generated blog content
-                st.subheader("Generated Blog Content")
-                st.write(blog_result)
-
-                # Display a summary if requested
-                if include_summary:
-                    summary_prompt = f"Summarize this content:\n{blog_result}"
-                    summary_result = blog_agent.run(summary_prompt)
-                    st.subheader("Content Summary")
-                    st.write(summary_result)
-
-                # Log and display agent activity
-                display_agent_activity(blog_prompt, blog_result)
-
-                # Save the content to a file
-                filename = save_to_file(blog_result)
-                st.success(f"Content saved to {filename}")
-            except Exception as e:
-                st.error("An error occurred while generating content. Please try again.")
+st.title("SmolAgent Content Writer")
+
+# Tabs for the interface
+tabs = st.tabs(["Generate Content", "Agent Activity", "Download Options"])
+
+# Tab 1: Generate Content
+with tabs[0]:
+    st.header("Generate Content")
+    st.write("Generate high-quality blog content enriched with real-time insights using SmolAgent.")
+
+    # Input field for blog topic or prompt
+    blog_prompt = st.text_area("Enter your blog topic or prompt:", placeholder="E.g., The Future of Technology in Healthcare")
+
+    # Input field for choosing content tone
+    tone = st.selectbox("Choose the tone of the content:", ["Professional", "Casual", "Persuasive", "Informative"])
+
+    # Option to include a summary
+    include_summary = st.checkbox("Include a summary of the generated content")
+
+    # Button to trigger blog content generation
+    if st.button("Generate Blog Content"):
+        if blog_prompt:
+            with st.spinner("Generating blog content, please wait..."):
+                try:
+                    # Generate blog content using the agent
+                    blog_result = content_agent.run(f"{blog_prompt}\nTone: {tone}")
+
+                    # Optionally enhance the content
+                    blog_result = enhance_content(blog_result)
+
+                    # Display the generated blog content
+                    st.subheader("Generated Blog Content")
+                    st.write(blog_result)
+
+                    # Display a summary if requested
+                    if include_summary:
+                        summary_prompt = f"Summarize this content:\n{blog_result}"
+                        summary_result = content_agent.run(summary_prompt)
+                        st.subheader("Content Summary")
+                        st.write(summary_result)
+
+                    # Store result for download
+                    st.session_state["last_content"] = blog_result
+                except Exception as e:
+                    st.error("An error occurred while generating content. Please try again.")
+        else:
+            st.warning("Please enter a valid blog topic or prompt.")
+
+# Tab 2: Agent Activity
+with tabs[1]:
+    st.header("Agent Activity")
+    if "last_content" in st.session_state:
+        display_agent_activity(blog_prompt, st.session_state["last_content"])
+    else:
+        st.write("No recent activity to display.")
+
+# Tab 3: Download Options
+with tabs[2]:
+    st.header("Download Options")
+    if "last_content" in st.session_state:
+        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")
+
+        # 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")
     else:
-        st.warning("Please enter a valid blog topic or prompt.")
+        st.write("No content available for download. Generate content first.")