Update interm.py

interm.py

@@ -1,30 +1,17 @@
 import streamlit as st
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
+from fpdf import FPDF
 import tempfile
 
-# Initialize the content generation agent with DuckDuckGo tool
-content_agent = CodeAgent(
-    tools=[DuckDuckGoSearchTool()], 
-    model=HfApiModel()
-)
-
-# Function to display agent activity
-def display_agent_activity(prompt, result):
-    st.write("### Agent Activity:")
-    st.write("**Prompt Sent to Agent:**")
-    st.code(prompt, language="text")
-    st.write("**Agent Output:**")
-    st.code(result, language="text")
-
-# Function to save content as a downloadable text file
-def generate_text_download(content, filename="generated_blog.txt"):
+# Initialize content generation agent
+content_agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=HfApiModel())
+
+def save_as_text(content, 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
+def save_as_pdf(content, filename):
     with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
         pdf = FPDF()
         pdf.add_page()
@@ -33,83 +20,89 @@ def generate_pdf_download(content, filename="generated_blog.pdf"):
         pdf.output(tmp_file.name)
         return tmp_file.name, filename
 
-# Streamlit app title
-st.title("SmolAgents Content Writer")
-
-# Tabs for the interface
-tabs = st.tabs(["Generate Content", "Agent Activity", "Download Options"])
+st.title("SmolAgents Content Generator")
+tabs = st.tabs(["Generated Content", "Generate Summary", "Agent Activity"])
 
-# Tab 1: Generate Content
+# Tab 1: Generated Content
 with tabs[0]:
-    st.header("Generate Content")
-    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 AI in Healthcare")
+    st.header("Generate and View Content")
 
-    # Input field for choosing content tone
-    tone = st.selectbox("Choose the tone of the content:", ["Professional", "Casual", "Persuasive", "Informative"])
+    prompt = st.text_area("Enter your topic or prompt:")
+    tone = st.selectbox("Choose the tone:", ["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..."):
+    if st.button("Generate Content"):
+        if prompt:
+            with st.spinner("Generating content..."):
                 try:
-                    # Generate blog content using the agent
                     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}"
+                        f"Write a detailed blog post on the topic: {prompt}.\n"
+                        f"Structure: Introduction, Key Points, Applications, Ethical Considerations, Conclusion.\n"
+                        f"Tone: {tone}"
                     )
-                    blog_result = content_agent.run(agent_prompt)
-
-                    # 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"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.markdown(summary_result)
-
-                    # Store result for download
-                    st.session_state["last_content"] = blog_result
+                    result = content_agent.run(agent_prompt)
+                    st.session_state["generated_content"] = result
+                    st.subheader("Generated Content")
+                    st.write(result)
                 except Exception as e:
-                    st.error("An error occurred while generating content. Please try again.")
+                    st.error("Error generating content.")
         else:
-            st.warning("Please enter a valid blog topic or prompt.")
+            st.warning("Please enter a valid prompt.")
+
+    if "generated_content" in st.session_state:
+        content = st.session_state["generated_content"]
+        text_file_path, text_filename = save_as_text(content, "content.txt")
+        pdf_file_path, pdf_filename = save_as_pdf(content, "content.pdf")
 
-# Tab 2: Agent Activity
+        with open(text_file_path, "rb") as file:
+            st.download_button("Download as TXT", data=file, file_name=text_filename, mime="text/plain")
+
+        with open(pdf_file_path, "rb") as file:
+            st.download_button("Download as PDF", data=file, file_name=pdf_filename, mime="application/pdf")
+
+# Tab 2: Generate Summary
 with tabs[1]:
-    st.header("Agent Activity")
-    if "last_content" in st.session_state:
-        display_agent_activity(blog_prompt, st.session_state["last_content"])
+    st.header("Generate and View Summary")
+
+    if "generated_content" in st.session_state:
+        content = st.session_state["generated_content"]
+
+        if st.button("Generate Summary"):
+            with st.spinner("Generating summary..."):
+                try:
+                    summary_prompt = f"Summarize the following content:\n{content}"
+                    summary = content_agent.run(summary_prompt)
+                    st.session_state["content_summary"] = summary
+                    st.subheader("Summary")
+                    st.markdown(summary)
+                except Exception as e:
+                    st.error("Error generating summary.")
+
+        if "content_summary" in st.session_state:
+            summary = st.session_state["content_summary"]
+            text_file_path, text_filename = save_as_text(summary, "summary.txt")
+            pdf_file_path, pdf_filename = save_as_pdf(summary, "summary.pdf")
+
+            with open(text_file_path, "rb") as file:
+                st.download_button("Download Summary as TXT", data=file, file_name=text_filename, mime="text/plain")
+
+            with open(pdf_file_path, "rb") as file:
+                st.download_button("Download Summary as PDF", data=file, file_name=pdf_filename, mime="application/pdf")
+
     else:
-        st.write("No recent activity to display.")
+        st.write("No content generated yet. Go to the first tab to create content.")
 
-# Tab 3: Download Options
+# Tab 3: Agent Activity
 with tabs[2]:
-    st.header("Download Options")
-    if "last_content" in st.session_state:
-        content = st.session_state["last_content"]
+    st.header("Agent Activity")
 
-        # Text file download
-        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")
+    if "generated_content" in st.session_state:
+        st.write("**Prompt:**")
+        st.code(prompt, language="text")
+        st.write("**Generated Content:**")
+        st.code(st.session_state["generated_content"], language="text")
 
-        # PDF file download
-        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")
+        if "content_summary" in st.session_state:
+            st.write("**Generated Summary:**")
+            st.code(st.session_state["content_summary"], language="text")
     else:
-        st.write("No content available for download. Generate content first.")
+        st.write("No activity to display.")