Added validation to frontend

fe/gradio_app.py

@@ -36,6 +36,22 @@ def search_prompt(query: str, n: int) -> list:
         return [["Error", str(e)]]
 
 
+def validate_and_search(query: str, n: int) -> list:
+    """
+    Validate inputs and perform the search if inputs are valid.
+    Args:
+        query: The search query.
+        n: The number of results to return.
+    Returns:
+        list: A list of results or an error message.
+    """
+    if not query.strip():
+        return [["Error", "Your input is empty, please enter a query."]]
+    if n <= 0 or not isinstance(n, int):
+        return [["Error", "Invalid number of top results. Please enter a positive number."]]
+    # Call the backend search function
+    return search_prompt(query, n)
+
 with gr.Blocks() as demo:
     gr.Markdown(f"# {APP_NAME}")
 
@@ -49,7 +65,7 @@ with gr.Blocks() as demo:
     )
 
     search_button = gr.Button("Search")
-    search_button.click(search_prompt, inputs=[query, n], outputs=[output])
+    search_button.click(validate_and_search, inputs=[query, n], outputs=[output])
 
 
 demo.launch(server_name="0.0.0.0", server_port=FRONTEND_PORT)