Update app.py

app.py

@@ -37,21 +37,21 @@ def generate_pdf_download(content, filename="generated_blog.pdf"):
 st.title("SmolAgents Content Writer")
 
 # Tabs for the interface
-tabs = st.tabs(["Generated Content", "Generated Summary", "Agent Activity"])
+tabs = st.tabs(["Generate Content", "Agent Activity", "Download Options"])
 
-# Tab 1: Generated Content
+# Tab 1: Generate Content
 with tabs[0]:
-    #st.header("Generated Content")
+    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 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"])
 
     # Option to include a summary
-    #include_summary = st.checkbox("Include a summary of the generated content")
+    include_summary = st.checkbox("Include a summary of the generated content")
 
     # Button to trigger blog content generation
     if st.button("Generate Blog Content"):
@@ -68,57 +68,48 @@ with tabs[0]:
                     blog_result = content_agent.run(agent_prompt)
 
                     # Display the generated blog content
-                    #st.subheader("Generated Blog Content")
+                    st.subheader("Generated Blog Content")
                     st.write(blog_result)
 
-                    # Store content for download
-                    st.session_state["last_content"] = blog_result
-
-                    # Generate a summary if requested
+                    # 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.session_state["last_summary"] = summary_result
+                        st.subheader("Content Summary")
+                        st.markdown(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.")
 
-    # Download option for content
+# 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_path, text_filename = generate_text_download(content)
         with open(text_file_path, "rb") as file:
-            st.download_button(label="Download Content as Text File", data=file, file_name=text_filename, mime="text/plain")
+            st.download_button(label="Download as Text File", data=file, file_name=text_filename, mime="text/plain")
+
+        # 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 Content as PDF", data=file, file_name=pdf_filename, mime="application/pdf")
-
-# Tab 2: Generated Summary
-with tabs[1]:
-    #st.header("Generated Summary")
-    if "last_summary" in st.session_state:
-        st.subheader("Content Summary")
-        st.markdown(st.session_state["last_summary"])
-
-        # Download option for summary
-        summary = st.session_state["last_summary"]
-        summary_text_file_path, summary_text_filename = generate_text_download(summary, filename="generated_summary.txt")
-        with open(summary_text_file_path, "rb") as file:
-            st.download_button(label="Download Summary as Text File", data=file, file_name=summary_text_filename, mime="text/plain")
-        summary_pdf_file_path, summary_pdf_filename = generate_pdf_download(summary, filename="generated_summary.pdf")
-        with open(summary_pdf_file_path, "rb") as file:
-            st.download_button(label="Download Summary as PDF", data=file, file_name=summary_pdf_filename, mime="application/pdf")
-    else:
-        st.write("No summary available. Generate content and include a summary to view it here.")
-
-# Tab 3: Agent Activity
-with tabs[2]:
-    st.header("Agent Activity")
-    if "last_content" in st.session_state:
-        display_agent_activity(blog_prompt, st.session_state["last_content"])
+            st.download_button(label="Download as PDF", data=file, file_name=pdf_filename, mime="application/pdf")
     else:
-        st.write("No recent activity to display.")
+        st.write("No content available for download. Generate content first.")