Update app.py

app.py

@@ -1,50 +1,38 @@
-from fastapi import FastAPI, Query
+from fastapi import FastAPI, HTTPException, Query
+from typing import List, Dict
 import json
-from typing import Optional
+from fuzzywuzzy import process  # Install fuzzywuzzy: pip install fuzzywuzzy
+
 
 app = FastAPI()
 
-DATA_FILE = "data.json"
+# Load data from the JSON file.  (Crucial: Handle potential file errors)
+try:
+    with open("converted_data.json", "r") as f:
+        data = json.load(f)
+except FileNotFoundError:
+    raise HTTPException(status_code=500, detail="File 'converted_data.json' not found.")
+except json.JSONDecodeError as e:
+    raise HTTPException(status_code=500, detail=f"Error decoding JSON: {e}")
 
-def load_data():
-    """Loads data from the JSON file."""
-    try:
-        with open(DATA_FILE, "r") as f:
-            data = json.load(f)
-        return data
-    except FileNotFoundError:
-        return {}
 
-@app.get("/search")
-async def search_links(
-    query: str = Query(..., description="Search term to find relevant links."),
-    quality: Optional[str] = Query(None, description="Optional: Filter links by quality (e.g., 1080P, 720P).")
-):
+@app.get("/search_links/")
+async def search_links(query: str = Query(..., description="Search query")):
     """
-    Searches for links based on the provided query and optional quality.
-
-    Args:
-        query: The search term.
-        quality: Optional quality filter (e.g., "1080P", "720P").
-
-    Returns:
-        A list of matching links with their titles and source URLs.
+    Searches the movie titles and returns links matching the query using fuzzy search.
     """
-    data = load_data()
-    results = []
-    query_lower = query.lower()
-
-    for url, links in data.items():
-        if query_lower in url.lower():
-            # Add the main URL itself as a potential match, regardless of quality filter
-            results.append({"source_url": url, "link": url, "title": url})
-        for link_data in links:
-            if query_lower in link_data["title"].lower():
-                if quality is None or quality.lower() in link_data["title"].lower():
-                    results.append({"source_url": url, "link": link_data["link"], "title": link_data["title"]})
-
-    return results
-
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    results = {}
+    for title, links in data.items():
+        ratio = process.extractOne(query, title)
+        if ratio[1] > 80:  # Adjust threshold as needed
+            results[title] = links
+        else:
+             # Use extract on links in each entry
+             for entry in links:
+                if entry:
+                    ratio_entry = process.extractOne(query, entry["title"])
+                    if ratio_entry and ratio_entry[1]>80:
+                         results[title] = [entry]
+
+    
+    return results