refactor app

app.py

@@ -16,8 +16,12 @@ README_URL_TEMPLATE = "https://huggingface.co/datasets/{}/raw/main/README.md"
 
 async def fetch_similar_datasets(dataset_id: str, limit: int = 10) -> List[Dict]:
     async with httpx.AsyncClient() as client:
-        response = await client.get(f"{API_URL}?dataset_id={dataset_id}&n={limit}")
-        return response.json()["results"][1:] if response.status_code == 200 else []
+        response = await client.get(f"{API_URL}?dataset_id={dataset_id}&n={limit + 1}")
+        if response.status_code == 200:
+            results = response.json()["results"]
+            # Remove the input dataset from the results
+            return [r for r in results if r["dataset_id"] != dataset_id][:limit]
+        return []
 
 
 async def fetch_dataset_card(dataset_id: str) -> str:
@@ -106,6 +110,9 @@ def format_results(
 async def search_similar_datasets(dataset_id: str, limit: int = 10):
     results = await fetch_similar_datasets(dataset_id, limit)
 
+    if not results:
+        return "No similar datasets found."
+
     # Fetch dataset cards and info concurrently
     dataset_cards = await asyncio.gather(
         *[fetch_dataset_card(result["dataset_id"]) for result in results]