Rename utils/imdb to utils/imdb_tools.py

utils/imdb

@@ -1,18 +0,0 @@
-from imdb import IMDb
-
-ia = IMDb()
-
-def search_imdb(title):
-    results = ia.search_movie(title)
-    if not results:
-        return "No results found"
-    
-    top = results[0]
-    ia.update(top)
-    return {
-        "Title": top.get("title"),
-        "Year": top.get("year"),
-        "Rating": top.get("rating"),
-        "Genres": top.get("genres"),
-        "Plot": top.get("plot")[0] if top.get("plot") else "No plot available"
-    }

utils/imdb_tools.py

@@ -0,0 +1,23 @@
+from imdb import IMDb
+
+# Create a global IMDb instance
+ia = IMDb()
+
+def search_imdb(title: str):
+    """
+    Search for a movie by title and return the most relevant result.
+    """
+    results = ia.search_movie(title)
+    if not results:
+        return {"error": "No results found"}
+
+    top_result = results[0]
+    ia.update(top_result)  # fetch full details
+
+    return {
+        "Title": top_result.get("title"),
+        "Year": top_result.get("year"),
+        "Rating": top_result.get("rating"),
+        "Genres": top_result.get("genres"),
+        "Plot": top_result.get("plot", ["No plot available"])[0],
+    }