Update app.py

app.py

@@ -1,85 +1,69 @@
-import sqlite3
 import gradio as gr
-import os
 
-DATABASE = 'restaurant.db'
-
-
-# Initialize database
-def init_db():
-    with sqlite3.connect(DATABASE) as conn:
-        cursor = conn.cursor()
-        cursor.execute('''
-            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)
-            )
-        ''')
-        # 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 "
-                       "(1, 'Pancakes', 'Fluffy pancakes with syrup.', 5.99, 1, 'pancakes.png')")
-        cursor.execute("INSERT OR IGNORE INTO Items (id, name, description, price, category_id, image) VALUES "
-                       "(2, 'Classic Ramen', 'Delicious ramen with pork and egg.', 12.99, 2, 'ramen.png')")
-        conn.commit()
-
-
-if not os.path.exists(DATABASE):
-    init_db()
-
-
-# Gradio Functions
-def get_categories():
-    with sqlite3.connect(DATABASE) as conn:
-        cursor = conn.cursor()
-        cursor.execute("SELECT * FROM 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()
-        cursor.execute("SELECT * FROM Items WHERE category_id = ?", (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 display_categories():
-    return get_categories()
-
-
-def display_items(category_id):
-    return get_items(category_id)
-
-
-# Gradio App
+# Food item data
+food_data = [
+    {
+        "name": "Apple",
+        "image": "apple.jpg",
+        "nutrition": {
+            "Calories": "52 kcal",
+            "Proteins": "0.3 g",
+            "Carbohydrates": "14 g"
+        }
+    },
+    {
+        "name": "Banana",
+        "image": "banana.jpg",
+        "nutrition": {
+            "Calories": "96 kcal",
+            "Proteins": "1.3 g",
+            "Carbohydrates": "27 g"
+        }
+    },
+    {
+        "name": "Chicken Breast",
+        "image": "chicken_breast.jpg",
+        "nutrition": {
+            "Calories": "165 kcal",
+            "Proteins": "31 g",
+            "Carbohydrates": "0 g"
+        }
+    },
+    {
+        "name": "Rice",
+        "image": "rice.jpg",
+        "nutrition": {
+            "Calories": "130 kcal",
+            "Proteins": "2.7 g",
+            "Carbohydrates": "28 g"
+        }
+    }
+]
+
+# Function to show food details
+def food_details(food_name):
+    # Find the food item by name
+    food_item = next((food for food in food_data if food["name"] == food_name), None)
+    if food_item:
+        # Format nutrition details
+        nutrition_details = "\n".join(f"{key}: {value}" for key, value in food_item["nutrition"].items())
+        return f"### {food_item['name']}\n\n**Nutrition Details:**\n\n{nutrition_details}"
+    return "Food item not found."
+
+# Gradio Interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Restaurant App")
+    gr.Markdown("# Food Nutrition App")
     
-    with gr.Tab("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)
-
-demo.launch(share=True)
-
+    # Food items gallery
+    with gr.Row():
+        for food in food_data:
+            with gr.Column():
+                gr.Markdown(f"### {food['name']}")
+                food_button = gr.Button(image=food["image"])
+                food_button.click(
+                    food_details,
+                    inputs=gr.Textbox(value=food["name"], visible=False),
+                    outputs=gr.Markdown()
+                )
+
+demo.launch()