Update app.py

app.py

@@ -5,80 +5,56 @@ 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:
-            return HfApiModel(model_id="Qwen/Qwen2.5-32B-Instruct")
-        except Exception as e:
-            st.error(f"Error initializing model: {str(e)}")
-            return None
-    return None
-
-# Initialize tools
+# Initialize tools first (outside session state)
 search_tool = DuckDuckGoSearchTool()
 
 def main():
-    st.title("DuckDuckGo Search Tool")
-    st.markdown("Search the web using DuckDuckGo and get AI-powered responses")
-
-    # Initialize session state
-    if "hf_token" not in st.session_state:
-        st.session_state.update({
-            "hf_token": "",
-            "model": None
-        })
-
-    # Sidebar configuration
+    # Streamlit config
+    st.set_page_config(
+        page_title="DuckDuckGo Search Tool",
+        page_icon="🔍",
+        layout="wide",
+    )
+    
+    # Session state initialization
+    if "model" not in st.session_state:
+        st.session_state.model = None
+        st.session_state.hf_token = ""
+
+    # Sidebar logic
     with st.sidebar:
         st.title("Configuration")
-        token_input = st.text_input(
-            "Hugging Face API Token:", 
-            value=st.session_state.hf_token,
+        new_token = st.text_input(
+            "Hugging Face Token",
             type="password",
-            help="Get token from huggingface.co/settings/tokens"
+            value=st.session_state.hf_token
         )
         
-        if token_input != st.session_state.hf_token:
-            st.session_state.hf_token = token_input
-            st.session_state.model = None
-        
+        if new_token != st.session_state.hf_token:
+            st.session_state.hf_token = new_token
+            st.session_state.model = None  # Reset model on token change
+
         if st.button("Initialize API"):
-            with st.spinner("Testing token..."):
-                model = initialize_api(st.session_state.hf_token)
-                if model:
-                    st.session_state.model = model
-                    st.success("✅ API initialized!")
-                else:
-                    st.error("❌ Initialization failed")
+            with st.spinner("Initializing..."):
+                try:
+                    login(st.session_state.hf_token)
+                    st.session_state.model = HfApiModel(model_id="meta-llama/Llama-3.3-70B-Instruct")
+                    st.success("API Initialized!")
+                except Exception as e:
+                    st.error(f"Error: {str(e)}")
 
     # Main app logic
     if st.session_state.model:
-        st.header("AI Web Search")
-        query = st.text_input("Search query:", placeholder="Latest renewable energy advancements?")
+        st.title("AI Web Search")
+        query = st.text_input("Enter your search query")
         
-        if st.button("Search") and query:
+        if query and st.button("Search"):
             with st.spinner("Searching..."):
                 agent = CodeAgent(tools=[search_tool], model=st.session_state.model)
                 response = agent.run(query)
-                st.success("Search complete!")
-                st.markdown("### Results")
-                st.markdown(response)
+                st.markdown(f"## Results\n{response}")
     else:
-        st.info("Please enter your Hugging Face API token in the sidebar")
-        # ... rest of your info message ...
+        st.info("Please enter your Hugging Face token in the sidebar")
 
-# Hugging Face Spaces requires this specific structure
 if __name__ == "__main__":
-    main()
-    # Keep the server running
-    while True:
-        pass
+    main()  # No infinite loop needed when using proper streamlit run