Update app.py

app.py

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