Update app.py

app.py

@@ -12,9 +12,6 @@ def home():
 
 @app.get("/search")
 def search_anime(keyword: str):
-    """
-    Example call: /search?keyword=naruto
-    """
     try:
         url = f"https://www.aniwave.se/filter?keyword={keyword}"
         headers = {"User-Agent": "Mozilla/5.0"}
@@ -23,7 +20,7 @@ def search_anime(keyword: str):
             raise HTTPException(status_code=500, detail="Failed to fetch search page")
 
         soup = BeautifulSoup(resp.text, "html.parser")
-        items = soup.select("div.item")  # each anime container
+        items = soup.select("div.item")
 
         results = []
         for item in items:
@@ -34,7 +31,6 @@ def search_anime(keyword: str):
             thumb_elem = item.select_one("div.ani.poster img")
             thumbnail = thumb_elem.get("src") if thumb_elem else None
 
-            # Episodes
             sub_elem = item.select_one("span.ep-status.sub span")
             dub_elem = item.select_one("span.ep-status.dub span")
             total_elem = item.select_one("span.ep-status.total span")
@@ -45,7 +41,6 @@ def search_anime(keyword: str):
                 "total": int(total_elem.text.strip()) if total_elem else None
             }
 
-            # Type (Tv, Special, Movie)
             type_elem = item.select_one("div.meta .inner .right")
             anime_type = type_elem.text.strip() if type_elem else None
 
@@ -58,6 +53,19 @@ def search_anime(keyword: str):
                     "type": anime_type
                 })
 
+        # Custom ranking
+        keyword_lower = keyword.lower()
+        def rank(item):
+            t = item["title"].lower()
+            if t == keyword_lower:
+                return 0  # exact match first
+            elif t.startswith(keyword_lower):
+                return 1  # starts with keyword second
+            else:
+                return 2  # the rest
+
+        results = sorted(results, key=lambda x: (rank(x), x["title"].lower()))
+
         return {"keyword": keyword, "results": results}
 
     except Exception as e: