Update app.py

app.py

@@ -1,27 +1,23 @@
 import streamlit as st
-
-# Used for Streamlit deployment
 from swarm import Swarm, Agent
-
-
-
 from bs4 import BeautifulSoup
 import requests
-from dotenv import load_dotenv
+from huggingface_hub import HfApi
 import os
 
-# Load environment variables from .env file if available
-load_dotenv()
-
-# Function to set OpenAI API key dynamically in the session state
-def set_openai_api_key():
-    api_key_input = st.text_input("Enter your OpenAI API Key", type="password")
-    if api_key_input:
-        os.environ['OPENAI_API_KEY'] = api_key_input
-        st.success("OpenAI API Key set successfully!")
-    else:
-        st.warning("Please enter your OpenAI API Key to continue.")
-
+# Function to fetch OpenAI API key from Hugging Face secrets
+def fetch_openai_api_key():
+    try:
+        # Replace 'OPENAI_API_KEY' with the exact key name you use in Hugging Face secrets
+        api = HfApi()
+        secret_key = api.get_secret("OPENAI_API_KEY")
+        if secret_key:
+            os.environ['OPENAI_API_KEY'] = secret_key
+            st.success("OpenAI API Key retrieved and set successfully!")
+        else:
+            st.error("Could not retrieve the OpenAI API Key. Please check your Hugging Face secrets.")
+    except Exception as e:
+        st.error(f"Error retrieving OpenAI API Key: {str(e)}")
 
 # Initialize the Swarm client
 def initialize_swarm_client():
@@ -42,7 +38,7 @@ def scrape_website(url):
 scraper_agent = Agent(
     name="Scraper Agent",
     instructions="You are an agent that scrapes content from websites.",
-    functions=[scrape_website]  # The agent can use the scrape_website function
+    functions=[scrape_website]
 )
 
 # Define the analysis function
@@ -55,7 +51,7 @@ def analyze_content(content):
 research_agent = Agent(
     name="Research Agent",
     instructions="You are an agent that analyzes content and extracts key insights.",
-    functions=[analyze_content]  # The agent can use the analyze_content function
+    functions=[analyze_content]
 )
 
 # Define the writing function
@@ -69,7 +65,7 @@ def write_summary(context_variables):
 writer_agent = Agent(
     name="Writer Agent",
     instructions="You are an agent that writes summaries of research.",
-    functions=[write_summary]  # The agent can use the write_summary function
+    functions=[write_summary]
 )
 
 # Orchestrate the workflow
@@ -96,41 +92,51 @@ def orchestrate_workflow(client, url):
     writer_result = client.run(
         agent=writer_agent,
         messages=[{"role": "user", "content": f"Write a summary based on this analysis: {analysis_summary}"}],
-        context_variables={"analysis": analysis_summary}  # Pass the analysis to the writer agent
+        context_variables={"analysis": analysis_summary}
     )
 
     final_summary = writer_result.messages[-1]["content"]
     return final_summary
 
 # Streamlit App UI
-st.title("🔍 OpenAI SWARM Web Scraping and Content Analysis with Multi-Agent System")
-st.caption("This app scrapes a website, analyzes the content, and generates a summary using a multi-agent system built on OpenAI's Swarm framework.")
-
-# Input for OpenAI API Key
-st.subheader("OpenAI API Key Setup")
-set_openai_api_key()
+st.title("🌐 Swarm-based Multi-Agent Web Scrapper and Content Analyzer")
+st.caption("""
+**Effortlessly extract, analyze, and summarize information from any website!**  
+This app leverages a **multi-agent system** built on OpenAI's Swarm framework to:
+- **Scrape content** from websites.
+- **Analyze and extract key insights** from the scraped data.
+- **Generate concise summaries** tailored to your needs.  
+Simply provide a URL, and let the agents do the rest!
+""")
+
+# Fetch OpenAI API Key from Hugging Face secrets
+st.subheader("🔑 OpenAI API Key Setup")
+fetch_openai_api_key()
 
 # Initialize Swarm client only after API key is set
 if 'OPENAI_API_KEY' in os.environ and os.environ['OPENAI_API_KEY']:
-    # Initialize the Swarm client after API key is entered
+    # Initialize the Swarm client after API key is set
     client = initialize_swarm_client()
 
     # Input field for the website URL
+    st.subheader("🌍 Enter the Website URL")
     url = st.text_input("Enter the URL of the website you want to scrape", placeholder="https://example.com")
 
     # Run Workflow button
-    if st.button("Run Workflow"):
+    if st.button("🚀 Run Workflow"):
         if url:
-            with st.spinner("Running the multi-agent workflow..."):
+            with st.spinner("Running the multi-agent workflow... This may take a moment."):
                 final_report = orchestrate_workflow(client, url)
-            st.success("Workflow complete!")
-            st.write("### Final Report:")
+            st.success("✅ Workflow complete!")
+            st.write("### 📜 Final Report:")
             st.write(final_report)
         else:
-            st.error("Please enter a valid URL.")
+            st.error("❌ Please enter a valid URL.")
 else:
-    st.warning("Please set your OpenAI API Key to proceed.")
+    st.warning("⚠️ OpenAI API Key not set. Please ensure it's properly configured in Hugging Face secrets.")
 
 # Footer with credits
 st.write("---")
-st.write("**Acknowledgement:**")
+st.markdown("""
+### Acknowledgement: 
+""")