Update app.py

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
-from langchain.agents import load_tools, initialize_agent
+from langchain_community.agent_toolkits.load_tools import load_tools  # Updated import
+from langchain.agents import initialize_agent
 from langchain.agents import AgentType
-from langchain_community.chat_models import ChatOpenAI  # Updated import
+from langchain_openai import ChatOpenAI  # Updated import
 import os
 
 # Set your OpenAI API key (ensure to store it securely in Hugging Face Spaces environment variables)
@@ -14,7 +15,7 @@ _ = load_dotenv()
 
 # Define the LLM model
 llm_model = "gpt-3.5-turbo"
-llm = ChatOpenAI(temperature=0, model=llm_model)
+llm = ChatOpenAI(temperature=0, model=llm_model, openai_api_key=os.getenv('OPENAI_API_KEY'))  # Ensure to pass the API key
 
 # Load tools
 tools = load_tools(["llm-math", "wikipedia"], llm=llm)