Update app.py

app.py

@@ -1,10 +1,11 @@
 import streamlit as st
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
-import yfinance as yf
 
-# Define agents for both functionalities
-search_agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=HfApiModel())
-blog_agent = CodeAgent(tools=[], model=HfApiModel())  # Replace with blog-specific tools/models if needed
+# Define the blog-writing agent with DuckDuckGo tool
+blog_agent = CodeAgent(
+    tools=[DuckDuckGoSearchTool()], 
+    model=HfApiModel()
+)
 
 # Function to log agent actions
 def log_agent_action(prompt, result, agent_name):
@@ -14,78 +15,33 @@ def log_agent_action(prompt, result, agent_name):
     st.write("**Agent Output:**")
     st.code(result, language="text")
 
-# Function to retrieve stock data using yfinance
-def get_stock_info(symbol):
-    try:
-        stock = yf.Ticker(symbol)
-        info = stock.info
-        return {
-            "Symbol": info.get("symbol", "N/A"),
-            "Name": info.get("shortName", "N/A"),
-            "Current Price": info.get("regularMarketPrice", "N/A"),
-            "Currency": info.get("currency", "N/A"),
-            "Previous Close": info.get("regularMarketPreviousClose", "N/A"),
-            "Market Cap": info.get("marketCap", "N/A"),
-        }
-    except Exception as e:
-        return {"Error": f"Could not retrieve stock information: {e}"}
-
 # Streamlit app title
-st.title("AI Agent Hub: Blog Writing & Stock Data Retrieval")
-
-# Main navigation options
-st.sidebar.header("Select a Feature")
-selected_feature = st.sidebar.radio(
-    "What would you like to do?",
-    ("Blog Writing Agent", "Stock Data Helper")
-)
-
-# Prompt input and execution logic
-if selected_feature == "Blog Writing Agent":
-    st.header("Blog Writing Agent")
-    blog_prompt = st.text_area("Enter your blog topic or prompt:")
-    
-    if st.button("Generate Blog Content"):
-        if blog_prompt:
-            with st.spinner("Generating blog content..."):
-                try:
-                    # Run the blog agent
-                    blog_result = blog_agent.run(blog_prompt)
-                    # Display result and log backend activity
-                    st.write("Generated Blog Content:")
-                    st.write(blog_result)
-                    log_agent_action(blog_prompt, blog_result, "Blog Writing Agent")
-                except Exception as e:
-                    st.error(f"An error occurred: {e}")
-        else:
-            st.warning("Please enter a blog prompt to continue.")
-
-elif selected_feature == "Stock Data Helper":
-    st.header("Stock Data Helper")
-    stock_prompt = st.text_area("Enter your query (e.g., company name, stock symbol):")
-    use_yfinance = st.checkbox("Fetch additional data using yfinance (e.g., AAPL for Apple)")
-
-    if st.button("Retrieve Stock Data"):
-        if stock_prompt:
-            with st.spinner("Retrieving stock data..."):
-                try:
-                    # Run the search agent for stock data
-                    stock_result = search_agent.run(stock_prompt)
-                    st.write("Stock Data Result from Agent:")
-                    st.write(stock_result)
-                    log_agent_action(stock_prompt, stock_result, "Stock Data Helper")
-
-                    # Fetch additional data using yfinance
-                    if use_yfinance:
-                        stock_info = get_stock_info(stock_prompt)
-                        st.write("Additional Stock Information (yfinance):")
-                        st.json(stock_info)
-
-                except Exception as e:
-                    st.error(f"An error occurred: {e}")
-        else:
-            st.warning("Please enter a stock-related query to continue.")
+st.title("AI Blog Writing Agent with Real-Time Insights")
+
+# App description
+st.write("Generate creative blogs enriched with real-time insights using the AI Blog Writing Agent powered by SmolAgents and DuckDuckGo.")
+
+# Input blog topic or prompt
+blog_prompt = st.text_area("Enter your blog topic or prompt:", placeholder="E.g., The Future of AI in Healthcare")
+
+# Button to generate blog content
+if st.button("Generate Blog Content"):
+    if blog_prompt:
+        with st.spinner("Generating blog content with real-time insights..."):
+            try:
+                # Run the blog agent with the given prompt
+                blog_result = blog_agent.run(blog_prompt)
+                # Display the generated blog content
+                st.subheader("Generated Blog Content:")
+                st.write(blog_result)
+
+                # Log backend activity
+                log_agent_action(blog_prompt, blog_result, "Blog Writing Agent with DuckDuckGo")
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
+    else:
+        st.warning("Please enter a blog topic or prompt to proceed.")
 
 # Footer
 st.markdown("---")
-st.caption("Powered by SmolAgents, Streamlit, and yfinance")
+st.caption("Powered by SmolAgents, DuckDuckGo, and Streamlit")