Update app.py

app.py

@@ -1,10 +1,35 @@
 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
 
+# Function to log agent actions
+def log_agent_action(prompt, result, agent_name):
+    st.write(f"### Agent Activity ({agent_name}):")
+    st.write("**Prompt Sent to Agent:**")
+    st.code(prompt, language="text")
+    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")
 
@@ -26,8 +51,10 @@ if selected_feature == "Blog Writing Agent":
                 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:
@@ -36,15 +63,24 @@ if selected_feature == "Blog Writing Agent":
 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:")
+                    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:
@@ -52,4 +88,4 @@ elif selected_feature == "Stock Data Helper":
 
 # Footer
 st.markdown("---")
-st.caption("Powered by SmolAgents and Streamlit")
+st.caption("Powered by SmolAgents, Streamlit, and yfinance")