Update app.py

app.py

@@ -1,11 +1,6 @@
 # app.py
-import os
-import random
-import requests
 import streamlit as st
-from io import BytesIO
-from PIL import Image
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, Tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
 from huggingface_hub import login
 import warnings
 
@@ -13,10 +8,9 @@ warnings.filterwarnings("ignore")
 
 # Set page configuration
 st.set_page_config(
-    page_title="SmolaAgents Tools",
-    page_icon="🤖",
+    page_title="DuckDuckGo Search Tool",
+    page_icon="🔍",
     layout="wide",
-    initial_sidebar_state="expanded"
 )
 
 # Function to initialize the API with user-provided token
@@ -33,91 +27,22 @@ def initialize_api(token=None):
         st.warning("Please enter your Hugging Face API token to use this app.")
         return None
 
-# Define all your tools
-class RandomComicFetcher(Tool):
-    name = "fetch_random_comic_image"
-    description = "Fetches the image of a random XKCD comic"
-    inputs = {}
-    output_type = "string"
-
-    def __init__(self):
-        super().__init__()
-        self.xkcd_base_url = "https://xkcd.com/"
-        self.xkcd_latest_url = f"{self.xkcd_base_url}info.0.json"
-
-    def forward(self):
-        try:
-            # Fetch the latest comic info to get max comic number
-            latest_resp = requests.get(self.xkcd_latest_url)
-            latest_resp.raise_for_status()
-            latest_data = latest_resp.json()
-            max_comic_id = latest_data["num"]
-
-            # Pick a random comic ID between 1 and latest
-            selected_id = random.randint(1, max_comic_id)
-            comic_info_url = f"{self.xkcd_base_url}{selected_id}/info.0.json"
-
-            # Fetch comic metadata
-            comic_resp = requests.get(comic_info_url)
-            comic_resp.raise_for_status()
-            comic_data = comic_resp.json()
-            comic_img_url = comic_data["img"]
-            comic_title = comic_data["title"]
-            comic_alt = comic_data["alt"]
-
-            # Download and return the comic image
-            image_resp = requests.get(comic_img_url)
-            image_resp.raise_for_status()
-            img = Image.open(BytesIO(image_resp.content))
-            
-            return img, comic_title, comic_alt, comic_img_url
-        except requests.exceptions.RequestException as err:
-            st.error(f"Could not fetch comic: {err}")
-            return None, "Error fetching comic", "Error details", ""
-
-class PrimeCheckTool(Tool):
-    name = "prime_check"
-    description = "Checks if a given number is a prime number."
-    inputs = {
-        "number": {
-            "type": "integer",
-            "description": "The number to check for primality.",
-        }
-    }
-    output_type = "boolean"
-
-    def forward(self, number: int) -> bool:
-        if number < 2:
-            return False
-        for i in range(2, int(number**0.5) + 1):
-            if number % i == 0:
-                return False
-        return True
-
 # Initialize tools
 search_tool = DuckDuckGoSearchTool()
-random_comic = RandomComicFetcher()
-prime_check_tool = PrimeCheckTool()
 
 # Streamlit App
-st.title("SmolaAgents Tools")
-st.markdown("Select a tool from the sidebar and interact with AI-powered assistants")
+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 tool selection and token input
+# Sidebar for token input
 with st.sidebar:
     st.title("Configuration")
     
-    # Check for token in environment first
-    env_token = os.environ.get("HF_TOKEN")
-    if env_token and not st.session_state.hf_token:
-        st.session_state.hf_token = env_token
-        st.success("HF_TOKEN found in environment variables")
-    
     # Token input 
     token_input = st.text_input(
         "Enter your Hugging Face API Token:", 
@@ -140,23 +65,6 @@ with st.sidebar:
                 st.success("✅ API initialized successfully!")
             else:
                 st.error("❌ Failed to initialize the API with the provided token.")
-    
-    st.divider()
-    
-    st.title("Tool Selection")
-    tool_choice = st.radio(
-        "Choose a tool:",
-        ["Search Tool", "XKCD Comic Fetcher", "Prime Number Checker"]
-    )
-    
-    st.divider()
-    st.markdown("### About")
-    st.markdown("""
-    This app demonstrates the capabilities of SmolaAgents, allowing you to:
-    - Search the web with an AI assistant
-    - Fetch random XKCD comics with AI commentary
-    - Check if numbers are prime with creative explanations
-    """)
 
 # Check if the model is initialized
 if not st.session_state.model and st.session_state.hf_token:
@@ -167,105 +75,24 @@ if not st.session_state.model and st.session_state.hf_token:
 
 # Main content area - only show if token is provided
 if st.session_state.hf_token and st.session_state.model:
-    # Main content area
-    if tool_choice == "Search Tool":
-        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")
-
-    elif tool_choice == "XKCD Comic Fetcher":
-        st.header("XKCD Comic Explorer")
-        st.markdown("Fetch a random XKCD comic and get AI commentary")
-        
-        if "comic_fetched" not in st.session_state:
-            st.session_state.comic_fetched = False
-            st.session_state.comic_img = None
-            st.session_state.comic_title = ""
-            st.session_state.comic_alt = ""
-            st.session_state.comic_url = ""
-        
-        col1, col2 = st.columns([1, 2])
-        
-        with col1:
-            if st.button("Fetch Random Comic"):
-                with st.spinner("Fetching a random comic..."):
-                    img, title, alt, url = random_comic.forward()
-                    if img:
-                        st.session_state.comic_fetched = True
-                        st.session_state.comic_img = img
-                        st.session_state.comic_title = title
-                        st.session_state.comic_alt = alt
-                        st.session_state.comic_url = url
-                        st.experimental_rerun()
-        
-        if st.session_state.comic_fetched:
-            with col1:
-                st.image(st.session_state.comic_img, caption=st.session_state.comic_title)
-                st.caption(f"Alt text: {st.session_state.comic_alt}")
-                st.markdown(f"[View on XKCD]({st.session_state.comic_url})")
-            
-            with col2:
-                st.subheader("Ask AI about this comic")
-                query = st.text_input("What would you like to know about this comic?", 
-                                    placeholder="Explain this comic in a funny way")
-                
-                if st.button("Ask AI"):
-                    with st.spinner("Generating response..."):
-                        agent = CodeAgent(tools=[random_comic], model=st.session_state.model)
-                        response = agent.run(query if query else "Tell me about this XKCD comic.")
-                        st.markdown("### AI Response")
-                        st.markdown(response)
-
-    elif tool_choice == "Prime Number Checker":
-        st.header("Prime Number Checker")
-        st.markdown("Check if a number is prime and get a creative explanation")
-        
-        number = st.number_input("Enter a number to check:", min_value=1, step=1, value=23)
-        explanation_style = st.selectbox(
-            "Choose explanation style:",
-            ["Poetic", "Nursery Rhyme", "Scientific", "Humorous", "Historical"]
-        )
-        
-        if st.button("Check Prime"):
-            with st.spinner("Checking and generating response..."):
-                is_prime = prime_check_tool.forward(int(number))
-                
-                # Format query based on selected style
-                query = f"Check if the number {number} is a prime number and explain it in a {explanation_style.lower()} style."
-                
-                agent = CodeAgent(tools=[prime_check_tool], model=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)
-                
-                # Display result with some styling
-                prime_status = "✅ PRIME" if is_prime else "❌ NOT PRIME"
-                st.markdown(f"### Result: {prime_status}")
-                
-                st.markdown("### AI Explanation")
+                st.success("Search complete!")
+                st.markdown("### Results")
                 st.markdown(response)
-                
-                # Optional: Show a mathematical representation
-                if st.checkbox("Show mathematical details"):
-                    if is_prime:
-                        st.markdown(f"{number} has no factors other than 1 and itself.")
-                    else:
-                        # Find the factors
-                        factors = [i for i in range(1, number + 1) if number % i == 0]
-                        st.markdown(f"Factors of {number}: {', '.join(map(str, factors))}")
+        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.")
+    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("""
@@ -275,15 +102,4 @@ else:
     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
-    
-    Your token will be used to access the language models needed for this application.
-    
-    ### Why is a token required?
-    
-    This app uses Hugging Face's API to access powerful language models like Llama 3. Your API token
-    grants access to these models, which do the heavy lifting of understanding and responding to your queries.
-    """)
-
-# Add footer
-st.sidebar.divider()
-st.sidebar.markdown("Made with ❤️ using SmolaAgents")
+    """)