Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import firebase_admin
 from firebase_admin import credentials, firestore
-from datetime import datetime
 
 # Firebase Admin SDK initialization
 cred = credentials.Certificate("key.json")  # Replace with your Firebase service account key file
@@ -10,62 +9,63 @@ db = firestore.client()
 
 # Function to get orders from Firestore and calculate totals
 def fetch_orders():
-    # Fetch orders from Firestore
     orders = db.collection("orders").get()
-    
-    # Initialize variables to hold display data and grand total
     orders_display = []
     grand_total = 0
 
-    # Iterate through each order to extract relevant information
     for order in orders:
         order_data = order.to_dict()
+        order_id = order.id  # Capture document ID for delete operations
         name = order_data.get("name")
         phone = order_data.get("phone", "N/A")
-        hostel = order_data.get("hostel_name", "N/A")  # Fetch hostel information
+        hostel = order_data.get("hostel_name", "N/A")
         items = order_data.get("items", [])
-        
-        # Ensure items is a list of dictionaries
-        if not isinstance(items, list):
-            items = []  # Default to an empty list if items is not in the correct format
-        
-        # Calculate total for each individual order
         total_price = order_data.get("total_price", 0)
         grand_total += total_price
 
-        # Append each item as a separate row
         for item in items:
             if isinstance(item, dict):
                 category = item.get("category", "")
                 item_name = item.get("item", "")
                 quantity = item.get("quantity", 0)
                 price = item.get("price", 0)
-                orders_display.append([name, phone, hostel, category, item_name, quantity, price])
-                # Clear name, phone, and hostel for subsequent items for a cleaner look
+                orders_display.append([order_id, name, phone, hostel, category, item_name, quantity, price])
                 name, phone, hostel = "", "", ""
 
-        # Append a subtotal row after all items for the current person
         orders_display.append(["", "", "", "", "Subtotal", "", total_price])
 
-    # Append a grand total row at the end
     orders_display.append(["", "", "", "", "Grand Total", "", grand_total])
-
     return orders_display
 
+# Function to delete an order
+def delete_order(order_id):
+    try:
+        db.collection("orders").document(order_id).delete()
+        return f"Order with ID {order_id} deleted successfully!"
+    except Exception as e:
+        return f"Failed to delete order: {e}"
+
 # Gradio Interface
 def gradio_display_interface():
     with gr.Blocks() as demo:
-        # Display section for individual order totals with headers
         gr.Markdown("### ADMIN PANEL")
-        orders_display = gr.Dataframe(value=[], headers=["Name", "Phone", "Hostel", "Category", "Item", "Quantity", "Price"], interactive=False, wrap=True)
 
-        # Fetch and display the orders when the button is clicked
+        # Display section for individual order totals with headers
+        orders_display = gr.Dataframe(value=[], headers=["ID", "Name", "Phone", "Hostel", "Category", "Item", "Quantity", "Price"], interactive=False, wrap=True)
+        
+        # Input field for delete operation
+        order_id = gr.Textbox(label="Order ID (for Delete)", placeholder="Enter Order ID here")
+        
+        # Textbox for displaying the result message
+        result_message = gr.Textbox(label="Result", interactive=False)
+
+        # Buttons for operations
         fetch_button = gr.Button("Fetch Orders")
-        fetch_button.click(
-            fn=fetch_orders,
-            inputs=[],
-            outputs=[orders_display]
-        )
+        delete_button = gr.Button("Delete Order")
+
+        # Button functionalities
+        fetch_button.click(fetch_orders, inputs=[], outputs=[orders_display])
+        delete_button.click(delete_order, inputs=[order_id], outputs=[result_message])
 
     return demo