Update app.py

app.py

@@ -4,53 +4,34 @@ import gradio as gr
 db = tinydb.TinyDB("books.json")
 table = db.table("books")
 
-def add_book(url, users):
-    # Create a dictionary to store the book information
+def Insert(url, users):
     book = {"url": url, "users": users}
-    # Insert the book into the table
     table.insert(book)
-    # Return a confirmation message
     return f"Added {url} by {users} to the database."
 
-def get_books_by_url(url):
-    # Query the database to get books with the specified URL
+def Fetch(url):
     books = table.search(tinydb.where("url") == url)
-    # Format the books as a string for display
     books_str = "\n".join([f"{book['url']} by {book['users']}" for book in books])
     return books_str
 
-def update_users_by_url(url, new_users):
-    # Fetch the existing users for the book
+def Update(url, new_users):
     existing_users = table.search(tinydb.where("url") == url)[0].get("users")
-    
-    # Update the users of the book with the specified URL
-    table.update({"users": new_users}, tinydb.where("url") == url)
-    
+    table.update({"users": new_users}, tinydb.where("url") == url)  
     return f"Updated users for the book '{url}' from '{existing_users}' to '{new_users}'."
 
-def delete_book_by_url(url):
-    # Remove the book with the specified URL from the table
+def Delete(url):
     table.remove(tinydb.where("url") == url)
     return f"Deleted book with URL '{url}' from the database."
 
 def process_data(option, url, users):
     if option == "Insert":
-        return add_book(url, users)
+        return Insert(url, users)
     elif option == "Fetch":
-        if url:
-            return get_books_by_url(url)
-        else:
-            return get_all_books()
+        return Fetch(url)
     elif option == "Update":
-        if url and users:
-            return update_users_by_url(url, users)
-        else:
-            return "URL and users are required for update."
+        return Update(url, users)
     elif option == "Delete":
-        if url:
-            return delete_book_by_url(url)
-        else:
-            return "URL is required for delete."
+        return Delete(url)
     else:
         return "Invalid option"