Update app.py

app.py

@@ -87,7 +87,7 @@ def image_generation(query):
 def handle_input(user_prompt, image=None, audio=None, websearch=False):
     if audio:
         if isinstance(audio, str):
-            audio = open(audio, "rb") 
+            audio = open(audio, "rb")
         transcription = client.audio.transcriptions.create(
             file=(audio.name, audio.read()),
             model="whisper-large-v3"
@@ -96,9 +96,12 @@ def handle_input(user_prompt, image=None, audio=None, websearch=False):
 
     tools = [
         FunctionTool.from_defaults(fn=numpy_code_calculator, name="Numpy Code Calculator"),
-        FunctionTool.from_defaults(fn=web_search, name="Web Search"),
         FunctionTool.from_defaults(fn=image_generation, name="Image Generation"),
     ]
+    
+    # Add the web search tool only if websearch mode is enabled
+    if websearch:
+        tools.append(FunctionTool.from_defaults(fn=web_search, name="Web Search"))
 
     llm = Groq(model=MODEL, api_key=os.environ.get("GROQ_API_KEY"))
     agent = ReActAgent.from_tools(tools, llm=llm, verbose=True)
@@ -116,6 +119,7 @@ def handle_input(user_prompt, image=None, audio=None, websearch=False):
 
     return response
 
+
 # Gradio UI Setup
 def create_ui():
     with gr.Blocks() as demo: