Spaces:
Running
Running
File size: 3,676 Bytes
8f69d14 277d8e8 8f69d14 277d8e8 8f69d14 6ea53e0 8f69d14 6ea53e0 8f69d14 277d8e8 8f69d14 6ea53e0 277d8e8 8f69d14 6ea53e0 8f69d14 6ea53e0 8f69d14 6ea53e0 277d8e8 6ea53e0 277d8e8 6ea53e0 277d8e8 6ea53e0 277d8e8 8f69d14 6ea53e0 8f69d14 6ea53e0 277d8e8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 |
# app.py
import streamlit as st
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
from huggingface_hub import login
import warnings
warnings.filterwarnings("ignore")
# Set page configuration
st.set_page_config(
page_title="DuckDuckGo Search Tool",
page_icon="🔍",
layout="wide",
)
# Function to initialize the API with user-provided token
def initialize_api(token=None):
if token:
login(token)
try:
# Initialize the model
return HfApiModel(model_id="meta-llama/Llama-3.3-70B-Instruct")
except Exception as e:
st.error(f"Error initializing model: {str(e)}")
return None
else:
st.warning("Please enter your Hugging Face API token to use this app.")
return None
# Initialize tools
search_tool = DuckDuckGoSearchTool()
# Streamlit App
st.title("DuckDuckGo Search Tool")
st.markdown("Search the web using DuckDuckGo and get AI-powered responses")
# Initialize session state for token
if "hf_token" not in st.session_state:
st.session_state.hf_token = ""
st.session_state.model = None
# Sidebar for token input
with st.sidebar:
st.title("Configuration")
# Token input
token_input = st.text_input(
"Enter your Hugging Face API Token:",
value=st.session_state.hf_token,
type="password",
help="Get your token from huggingface.co/settings/tokens"
)
if token_input != st.session_state.hf_token:
st.session_state.hf_token = token_input
# Reset model when token changes
st.session_state.model = None
# Button to initialize/test the token
if st.button("Initialize API"):
with st.spinner("Testing your token..."):
model = initialize_api(st.session_state.hf_token)
if model:
st.session_state.model = model
st.success("✅ API initialized successfully!")
else:
st.error("❌ Failed to initialize the API with the provided token.")
# Check if the model is initialized
if not st.session_state.model and st.session_state.hf_token:
with st.spinner("Initializing model..."):
st.session_state.model = initialize_api(st.session_state.hf_token)
if st.session_state.model:
st.success("Model initialized successfully!")
# Main content area - only show if token is provided
if st.session_state.hf_token and st.session_state.model:
st.header("AI Web Search")
st.markdown("Ask any question and the AI will search the web for answers")
query = st.text_input("Enter your search query:", placeholder="What are the latest advancements in renewable energy?")
if st.button("Search"):
if query:
with st.spinner("Searching and processing..."):
agent = CodeAgent(tools=[search_tool], model=st.session_state.model)
response = agent.run(query)
st.success("Search complete!")
st.markdown("### Results")
st.markdown(response)
else:
st.warning("Please enter a search query")
else:
# Token not provided
st.info("Please enter your Hugging Face API token in the sidebar to get started.")
# Show more information to help users understand what they need
st.markdown("""
### How to Get a Hugging Face Token
1. Go to [huggingface.co](https://huggingface.co/) and create an account if you don't have one
2. Visit your [settings page](https://huggingface.co/settings/tokens)
3. Create a new token with read access
4. Copy the token and paste it in the sidebar
""") |