Update app.py

app.py

@@ -1,6 +1,9 @@
+from pathlib import Path
+
 import gradio as gr
+
 from tool import LinkupSearchTool
-from pathlib import Path
+
 
 def run_search(query, depth, api_key):
     if not query.strip() or not depth or not api_key.strip():
@@ -12,21 +15,28 @@ def run_search(query, depth, api_key):
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
+
 def enable_submit(q, d, k):
     return gr.Button.update(interactive=bool(q.strip() and d and k.strip()))
 
+
 def load_readme():
     path = Path(__file__).parent / "README.md"
-    if path.exists():
-        return path.read_text()
+    if not path.exists():
+        return "README.md not found."
 
     lines = path.read_text().splitlines()
-    if lines[0].strip() == "---":
-        end_index = next((i for i, line in enumerate(lines[1:], 1) if line.strip() == "---"), -1)
+
+    # Strip frontmatter if present
+    if lines and lines[0].strip() == "---":
+        end_index = next(
+            (i for i, line in enumerate(lines[1:], 1) if line.strip() == "---"), -1
+        )
         if end_index != -1:
-            lines = lines[end_index + 1:]  # Skip frontmatter
+            lines = lines[end_index + 1 :]  # skip frontmatter block
     return "\n".join(lines)
 
+
 with gr.Blocks(title="Linkup Web Search Tool") as demo:
     gr.Markdown(
         """
@@ -55,7 +65,9 @@ with gr.Blocks(title="Linkup Web Search Tool") as demo:
         )
 
     search_btn = gr.Button("🔍 Search", interactive=True)
-    output = gr.Markdown("💬 *Enter a query and click Search to begin.*", elem_id="output")
+    output = gr.Markdown(
+        "💬 *Enter a query and click Search to begin.*", elem_id="output"
+    )
 
     search_btn.click(
         fn=run_search,
@@ -64,9 +76,15 @@ with gr.Blocks(title="Linkup Web Search Tool") as demo:
         show_progress=True,
     )
 
-    query_input.input(enable_submit, [query_input, depth_input, api_key_input], [search_btn])
-    depth_input.input(enable_submit, [query_input, depth_input, api_key_input], [search_btn])
-    api_key_input.input(enable_submit, [query_input, depth_input, api_key_input], [search_btn])
+    query_input.input(
+        enable_submit, [query_input, depth_input, api_key_input], [search_btn]
+    )
+    depth_input.input(
+        enable_submit, [query_input, depth_input, api_key_input], [search_btn]
+    )
+    api_key_input.input(
+        enable_submit, [query_input, depth_input, api_key_input], [search_btn]
+    )
 
     gr.Markdown("---")
     gr.Markdown("## 📄 Tool Documentation")