Update main.py

add new function to search by variable

main.py

@@ -1,8 +1,8 @@
-from fastapi import FastAPI, HTTPException, Request, Form
+from fastapi import FastAPI, HTTPException, Request, Form, Query
 from fastapi.responses import HTMLResponse, RedirectResponse
 from fastapi.templating import Jinja2Templates
 from pydantic import BaseModel
-from typing import List
+from typing import List, Optional
 import json
 import os
 
@@ -53,15 +53,33 @@ def create_book_api(book: Book):
     Create a new book record via JSON API.
     """
     books = read_books_file()
-    
+
     # Check if a book with the same id already exists
     if any(existing_book["id"] == book.id for existing_book in books):
         raise HTTPException(status_code=400, detail="Book with this id already exists.")
-    
+
     books.append(book.dict())
     write_books_file(books)
     return book
 
+@app.get("/api/books/search", response_model=List[Book])
+def search_books_api(
+    book_name: Optional[str] = Query(None),
+    year: Optional[int] = Query(None),
+    the_author: Optional[str] = Query(None)
+):
+    """
+    Search for books by book name, year, or author.
+    """
+    books = read_books_file()
+    filtered_books = [
+        book for book in books
+        if (book_name and book_name.lower() in book["book_name"].lower()) or
+           (year and book["year"] == year) or
+           (the_author and the_author.lower() in book["the_author"].lower())
+    ]
+    return filtered_books
+
 # ---------------------------
 # HTML Endpoints using Jinja2 & Bootstrap
 # ---------------------------
@@ -92,17 +110,17 @@ def add_book(
     Process the form data and add a new book.
     """
     books = read_books_file()
-    
+
     # Check if a book with the same id already exists
     if any(existing_book["id"] == id for existing_book in books):
         return templates.TemplateResponse(
             "add_book.html",
             {"request": request, "error": "Book with this id already exists."}
         )
-    
+
     new_book = {"id": id, "book_name": book_name, "year": year, "the_author": the_author}
     books.append(new_book)
     write_books_file(books)
-    
+
     # Redirect back to the homepage after a successful submission
     return RedirectResponse("/", status_code=303)