Update app.py

app.py

@@ -19,19 +19,19 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# Initialize HF API key from environment variable
-@st.cache_resource
-def initialize_api():
-    hf_token = os.environ.get("HF_TOKEN")
-    if hf_token:
-        login(hf_token)
+# 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("Warning: HF_API_TOKEN not found in environment variables")
-    
-    # Initialize the model
-    return HfApiModel(model_id="meta-llama/Llama-3.3-70B-Instruct")
-
-model = initialize_api()
+        st.warning("Please enter your Hugging Face API token to use this app.")
+        return None
 
 # Define all your tools
 class RandomComicFetcher(Tool):
@@ -103,8 +103,46 @@ prime_check_tool = PrimeCheckTool()
 st.title("SmolaAgents Tools")
 st.markdown("Select a tool from the sidebar and interact with AI-powered assistants")
 
-# Sidebar for tool selection
+# 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
 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:", 
+        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.")
+    
+    st.divider()
+    
     st.title("Tool Selection")
     tool_choice = st.radio(
         "Choose a tool:",
@@ -120,102 +158,131 @@ with st.sidebar:
     - Check if numbers are prime with creative explanations
     """)
 
-# 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=model)
-                response = agent.run(query)
-                st.success("Search complete!")
-                st.markdown("### Results")
-                st.markdown(response)
-        else:
-            st.warning("Please enter a search query")
+# 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!")
 
-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:
+# 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:
-            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})")
+            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()
         
-        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.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})")
             
-            if st.button("Ask AI"):
-                with st.spinner("Generating response..."):
-                    agent = CodeAgent(tools=[random_comic], model=model)
-                    response = agent.run(query if query else "Tell me about this XKCD comic.")
-                    st.markdown("### AI Response")
-                    st.markdown(response)
+            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")
+    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)
+                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.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:
+    # Token not provided
+    st.info("👈 Please enter your Hugging Face API token in the sidebar to get started.")
     
-    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"]
-    )
+    # Show more information to help users understand what they need
+    st.markdown("""
+    ### How to Get a Hugging Face Token
     
-    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=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.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))}")
+    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
+    
+    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()