Update app.py

app.py

@@ -1,33 +1,33 @@
 import sqlite3
-import os
 import gradio as gr
+import os
 
-# Database setup
 DATABASE = 'restaurant.db'
 
+
+# Initialize database
 def init_db():
     with sqlite3.connect(DATABASE) as conn:
         cursor = conn.cursor()
-        # Create tables
         cursor.execute('''
-        CREATE TABLE IF NOT EXISTS Categories (
-            id INTEGER PRIMARY KEY AUTOINCREMENT,
-            name TEXT NOT NULL,
-            image TEXT NOT NULL
-        )
+            CREATE TABLE IF NOT EXISTS Categories (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                name TEXT NOT NULL,
+                image TEXT NOT NULL
+            )
         ''')
         cursor.execute('''
-        CREATE TABLE IF NOT EXISTS Items (
-            id INTEGER PRIMARY KEY AUTOINCREMENT,
-            name TEXT NOT NULL,
-            description TEXT,
-            price REAL NOT NULL,
-            category_id INTEGER NOT NULL,
-            image TEXT NOT NULL,
-            FOREIGN KEY (category_id) REFERENCES Categories(id)
-        )
+            CREATE TABLE IF NOT EXISTS Items (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                name TEXT NOT NULL,
+                description TEXT,
+                price REAL NOT NULL,
+                category_id INTEGER NOT NULL,
+                image TEXT NOT NULL,
+                FOREIGN KEY (category_id) REFERENCES Categories(id)
+            )
         ''')
-        # Insert sample data
+        # Add sample data
         cursor.execute("INSERT OR IGNORE INTO Categories (id, name, image) VALUES (1, 'Breakfast', 'breakfast.png')")
         cursor.execute("INSERT OR IGNORE INTO Categories (id, name, image) VALUES (2, 'Ramen', 'ramen.png')")
         cursor.execute("INSERT OR IGNORE INTO Items (id, name, description, price, category_id, image) VALUES "
@@ -36,10 +36,11 @@ def init_db():
                        "(2, 'Classic Ramen', 'Delicious ramen with pork and egg.', 12.99, 2, 'ramen.png')")
         conn.commit()
 
-# Initialize database
+
 if not os.path.exists(DATABASE):
     init_db()
 
+
 # Gradio Functions
 def get_categories():
     with sqlite3.connect(DATABASE) as conn:
@@ -48,6 +49,7 @@ def get_categories():
         categories = [{"ID": row[0], "Name": row[1], "Image": row[2]} for row in cursor.fetchall()]
     return categories
 
+
 def get_items(category_id):
     with sqlite3.connect(DATABASE) as conn:
         cursor = conn.cursor()
@@ -55,69 +57,29 @@ def get_items(category_id):
         items = [{"ID": row[0], "Name": row[1], "Description": row[2], "Price": row[3], "Image": row[5]} for row in cursor.fetchall()]
     return items
 
-def update_item(item_id, name, description, price, category_id, image):
-    with sqlite3.connect(DATABASE) as conn:
-        cursor = conn.cursor()
-        cursor.execute("UPDATE Items SET name = ?, description = ?, price = ?, category_id = ?, image = ? WHERE id = ?",
-                       (name, description, price, category_id, image, item_id))
-        conn.commit()
-    return f"Item {item_id} updated successfully!"
 
-def get_deals():
-    with sqlite3.connect(DATABASE) as conn:
-        cursor = conn.cursor()
-        cursor.execute("SELECT * FROM Items LIMIT 3")
-        deals = [{"ID": row[0], "Name": row[1], "Description": row[2], "Price": row[3], "Image": row[5]} for row in cursor.fetchall()]
-    return deals
-
-# Gradio Interface
 def display_categories():
-    categories = get_categories()
-    return categories
+    return get_categories()
 
-def display_items(category_id):
-    items = get_items(category_id)
-    return items
 
-def update_item_interface(item_id, name, description, price, category_id, image):
-    return update_item(item_id, name, description, price, category_id, image)
+def display_items(category_id):
+    return get_items(category_id)
 
-def display_deals():
-    return get_deals()
 
 # Gradio App
 with gr.Blocks() as demo:
-    gr.Markdown("# Dynamic Restaurant App")
+    gr.Markdown("# Restaurant App")
     
     with gr.Tab("Categories"):
-        gr.Markdown("### View Categories")
+        gr.Markdown("### View All Categories")
         categories_output = gr.JSON()
         gr.Button("Show Categories").click(display_categories, outputs=categories_output)
-    
+
     with gr.Tab("Items"):
         gr.Markdown("### View Items by Category")
         category_id_input = gr.Number(label="Enter Category ID")
         items_output = gr.JSON()
         gr.Button("Show Items").click(display_items, inputs=category_id_input, outputs=items_output)
-    
-    with gr.Tab("Update Item"):
-        gr.Markdown("### Update an Item")
-        item_id_input = gr.Number(label="Item ID")
-        name_input = gr.Textbox(label="Name")
-        description_input = gr.Textbox(label="Description")
-        price_input = gr.Number(label="Price")
-        category_id_input_update = gr.Number(label="Category ID")
-        image_input = gr.Textbox(label="Image")
-        update_output = gr.Textbox(label="Update Status")
-        gr.Button("Update Item").click(
-            update_item_interface, 
-            inputs=[item_id_input, name_input, description_input, price_input, category_id_input_update, image_input], 
-            outputs=update_output
-        )
-    
-    with gr.Tab("Deals"):
-        gr.Markdown("### View Best Deals")
-        deals_output = gr.JSON()
-        gr.Button("Show Deals").click(display_deals, outputs=deals_output)
 
 demo.launch()
+