smol_agent / app.py
Akshayram1's picture
Update app.py
c9f0eeb verified
raw
history blame
1.75 kB
import streamlit as st
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
# 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):
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")
# Streamlit app title
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, DuckDuckGo, and Streamlit")