update app

app.py

@@ -1,26 +1,42 @@
-from smolagents import launch_gradio_demo
-from tool import LinkupSearchTool
 import gradio as gr
+from tool import LinkupSearchTool
+
+tool = LinkupSearchTool()  # we will pass API key at runtime only
+
+def run_search(query, depth, api_key):
+    if not query or not depth or not api_key:
+        return "❌ All fields are required."
+
+    try:
+        return tool.forward(query=query, depth=depth, api_key=api_key)
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
+
+with gr.Blocks() as demo:
+    gr.Markdown("## 🔍 Linkup Web Search Tool\nEnter your query and API key below.")
+
+    with gr.Row():
+        query_input = gr.Textbox(label="Search Query", placeholder="e.g. AI trends in 2024")
+        depth_input = gr.Dropdown(label="Search Depth", choices=["standard", "deep"])
+        api_key_input = gr.Textbox(label="Linkup API Key", type="password")
 
-tool = LinkupSearchTool()
+    output = gr.Markdown()
+    btn = gr.Button("Search")
 
-# Launch original demo to extract layout
-demo = launch_gradio_demo(tool, return_interface=True)
+    btn.click(
+        fn=run_search,
+        inputs=[query_input, depth_input, api_key_input],
+        outputs=output
+    )
 
-# Find the submit button and inputs
-submit_btn = demo.children[-1]  # last element is usually the submit button
-input_components = demo.children[0].children  # first child is the input layout
+    # Disable button unless all fields are filled
+    def enable_submit(q, d, k):
+        return bool(q and d and k)
 
-# Hook: disable submit if any input is empty
-def check_required_fields(*args):
-    return all(args)
+    query_input.change(enable_submit, [query_input, depth_input, api_key_input], [btn])
+    depth_input.change(enable_submit, [query_input, depth_input, api_key_input], [btn])
+    api_key_input.change(enable_submit, [query_input, depth_input, api_key_input], [btn])
 
-demo.load(
-    check_required_fields,
-    inputs=input_components,
-    outputs=[submit_btn],
-    every=0.1,
-    queue=False,
-)
+    btn.interactive = False  # disabled by default
 
 demo.launch()