Update app.py

app.py

@@ -1,8 +1,6 @@
-from flask import Flask, render_template, request, redirect, url_for
+import gradio as gr
 import sqlite3
 
-app = Flask(__name__)
-
 # Database setup
 def init_db():
     conn = sqlite3.connect('inventory.db')
@@ -18,49 +16,67 @@ def init_db():
 
 init_db()
 
-@app.route('/')
-def index():
+# Functions to handle inventory
+def view_inventory():
     conn = sqlite3.connect('inventory.db')
     cursor = conn.cursor()
     cursor.execute("SELECT * FROM inventory")
     items = cursor.fetchall()
     conn.close()
-    return render_template('index.html', items=items)
+    return items
 
-@app.route('/add', methods=['POST'])
-def add_item():
-    name = request.form['name']
-    quantity = request.form['quantity']
-    price = request.form['price']
-    
+def add_item(name, quantity, price):
     conn = sqlite3.connect('inventory.db')
     cursor = conn.cursor()
     cursor.execute("INSERT INTO inventory (name, quantity, price) VALUES (?, ?, ?)", (name, quantity, price))
     conn.commit()
     conn.close()
-    return redirect(url_for('index'))
+    return f"Added {name} to inventory."
 
-@app.route('/delete/<int:item_id>')
 def delete_item(item_id):
     conn = sqlite3.connect('inventory.db')
     cursor = conn.cursor()
     cursor.execute("DELETE FROM inventory WHERE id = ?", (item_id,))
     conn.commit()
     conn.close()
-    return redirect(url_for('index'))
+    return f"Deleted item with ID {item_id}."
 
-@app.route('/update/<int:item_id>', methods=['POST'])
-def update_item(item_id):
-    name = request.form['name']
-    quantity = request.form['quantity']
-    price = request.form['price']
-    
+def update_item(item_id, name, quantity, price):
     conn = sqlite3.connect('inventory.db')
     cursor = conn.cursor()
     cursor.execute("UPDATE inventory SET name = ?, quantity = ?, price = ? WHERE id = ?", (name, quantity, price, item_id))
     conn.commit()
     conn.close()
-    return redirect(url_for('index'))
+    return f"Updated item with ID {item_id}."
+
+# Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("# Inventory Management System")
+    with gr.Row():
+        with gr.Column():
+            name = gr.Textbox(label="Item Name")
+            quantity = gr.Number(label="Quantity")
+            price = gr.Number(label="Price")
+            add_btn = gr.Button("Add Item")
+            add_output = gr.Textbox()
+        with gr.Column():
+            item_id_del = gr.Number(label="Item ID to Delete")
+            delete_btn = gr.Button("Delete Item")
+            delete_output = gr.Textbox()
+        with gr.Column():
+            item_id_upd = gr.Number(label="Item ID to Update")
+            name_upd = gr.Textbox(label="New Name")
+            quantity_upd = gr.Number(label="New Quantity")
+            price_upd = gr.Number(label="New Price")
+            update_btn = gr.Button("Update Item")
+            update_output = gr.Textbox()
+    
+    inventory_list = gr.Dataframe(headers=["ID", "Name", "Quantity", "Price"])
+    refresh_btn = gr.Button("Refresh Inventory")
+    
+    add_btn.click(add_item, inputs=[name, quantity, price], outputs=add_output)
+    delete_btn.click(delete_item, inputs=item_id_del, outputs=delete_output)
+    update_btn.click(update_item, inputs=[item_id_upd, name_upd, quantity_upd, price_upd], outputs=update_output)
+    refresh_btn.click(view_inventory, outputs=inventory_list)
 
-if __name__ == '__main__':
-    app.run(debug=True)
+demo.launch()