Update app.py

app.py

@@ -5,6 +5,8 @@ from swarm import Swarm, Agent
 from bs4 import BeautifulSoup
 import requests
 import os
+import io
+from reportlab.pdfgen import canvas
 
 # Function to fetch OpenAI API key
 def fetch_openai_api_key():
@@ -97,6 +99,29 @@ def orchestrate_workflow(client, url):
     final_summary = writer_result.messages[-1]["content"]
     return final_summary
 
+# Helper functions to create text and PDF files
+def create_text_file(content):
+    """Create a downloadable text file."""
+    return io.StringIO(content)
+
+def create_pdf_file(content):
+    """Create a downloadable PDF file."""
+    buffer = io.BytesIO()
+    c = canvas.Canvas(buffer)
+    c.drawString(100, 750, "Generated Report")
+    c.drawString(100, 730, "--------------------")
+    lines = content.split("\n")
+    y = 700
+    for line in lines:
+        if y < 50:  # Create a new page if the content overflows
+            c.showPage()
+            y = 750
+        c.drawString(100, y, line)
+        y -= 20
+    c.save()
+    buffer.seek(0)
+    return buffer
+
 # Streamlit App UI
 st.markdown(
     """
@@ -110,7 +135,6 @@ st.markdown(
     unsafe_allow_html=True,
 )
 
-
 # 1. Add the title at the top
 st.markdown('<div class="title">Swarm-based Web Content Analyzer 🧐</div>', unsafe_allow_html=True)
 
@@ -133,13 +157,12 @@ st.markdown(
 # 5. Add one line-spacing after the acknowledgment
 st.markdown('<div style="margin-bottom: 20px;"></div>', unsafe_allow_html=True)
 
-
 fetch_openai_api_key()
 
 if 'OPENAI_API_KEY' in os.environ and os.environ['OPENAI_API_KEY']:
     client = initialize_swarm_client()
 
-    # 4. Add interface for URL input
+    # Add interface for URL input
     st.subheader("Enter the Website URL 🔗")
     url = st.text_input("Enter the URL of the website you want to scrape", placeholder="https://huggingface.co/models")
 
@@ -154,9 +177,25 @@ if 'OPENAI_API_KEY' in os.environ and os.environ['OPENAI_API_KEY']:
             st.success("✅ Workflow complete!")
             st.write("### 📜 Final Report:")
             st.write(final_report)
+
+            # Add download buttons for the report
+            text_file = create_text_file(final_report)
+            pdf_file = create_pdf_file(final_report)
+
+            st.download_button(
+                label="Download Report as Text",
+                data=text_file,
+                file_name="report.txt",
+                mime="text/plain",
+            )
+
+            st.download_button(
+                label="Download Report as PDF",
+                data=pdf_file,
+                file_name="report.pdf",
+                mime="application/pdf",
+            )
         else:
             st.error("❌ Please enter a valid URL.")
 else:
     st.sidebar.warning("⚠️ OpenAI API Key not set. Please check your Hugging Face secrets configuration.")
-
-