Update interm.py

interm.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(
@@ -16,31 +16,25 @@ def display_agent_activity(prompt, result):
     st.write("**Agent Output:**")
     st.code(result, language="text")
 
-# 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
+    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")
+st.title("SmolAgents Content Writer")
 
 # Tabs for the interface
 tabs = st.tabs(["Generate Content", "Agent Activity", "Download Options"])
@@ -48,10 +42,10 @@ 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.")
+    st.write("Generate high-quality blog content enriched with real-time insights using SmolAgents.")
 
     # 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")
+    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"])
@@ -65,10 +59,13 @@ with tabs[0]:
             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)
+                    agent_prompt = (
+                        f"Write a well-structured and detailed blog post on the following topic:\n"
+                        f"{blog_prompt}\n"
+                        f"Structure it with: Introduction, Key Trends, Applications, Ethical Considerations, and Conclusion."
+                        f" Use numbered or bulleted points for key takeaways. Tone: {tone}"
+                    )
+                    blog_result = content_agent.run(agent_prompt)
 
                     # Display the generated blog content
                     st.subheader("Generated Blog Content")
@@ -76,10 +73,13 @@ with tabs[0]:
 
                     # Display a summary if requested
                     if include_summary:
-                        summary_prompt = f"Summarize this content:\n{blog_result}"
+                        summary_prompt = (
+                            f"Provide a concise and well-ordered summary of the following content with key points as bullet points:\n"
+                            f"{blog_result}"
+                        )
                         summary_result = content_agent.run(summary_prompt)
                         st.subheader("Content Summary")
-                        st.write(summary_result)
+                        st.markdown(summary_result)
 
                     # Store result for download
                     st.session_state["last_content"] = blog_result
@@ -103,11 +103,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.")