Update app.py

app.py

@@ -162,45 +162,96 @@ def cancel_order(order_id, customer_id):
         "customer_id": customer_id,
         "action": "Order Canceled"
     }
-    log_action("cancel_order", details)
+    try:
+ # Connect to the database
+        connection = sqlite3.connect("2911.db")  # Replace with your .db file path
+        cursor = connection.cursor()
 
+        # Update the order status
+        query_order = """
+        UPDATE Orders
+        SET status = 'Canceled'
+        WHERE order_id = ? AND customer_id = ?;
+        """
+        cursor.execute(query_order, (order_id, customer_id))
+
+        if cursor.rowcount == 0:
+            return f"No matching order found for Order ID {order_id} and Customer ID {customer_id}."
+
+        # Update the shipping status
+        query_shipping = """
+        UPDATE Shipping
+        SET shipping_status = 'Canceled'
+        WHERE order_id = ? AND customer_id = ?;
+        """
+        cursor.execute(query_shipping, (order_id, customer_id))
+
+        cursor.execute("SELECT cancel_fee FROM Orders WHERE order_id = ?", (order_id,))
+        row = cursor.fetchone()
+        if not row:
+            cancel_fee = 0.0
+        
+        cancel_fee = float(row[0])
+
+        # Insert refund details
+        query_refund = """
+        INSERT INTO Refund (order_id, cancellation_date, cancel_fee, refund_amount, refund_status)
+        VALUES (?, datetime('now'), ?, ?, 'Under Process');
+        """
+        cursor.execute(query_refund, (order_id, cancel_fee, refund_amount))
+
+        connection.commit()
+        log_action("cancel_order", details)  # Log the cancellation
+        return f"Order ID {order_id} for Customer ID {customer_id} has been canceled successfully. Shipping status updated."
+
+    except sqlite3.Error as e:
+        return f"An error occurred: {e}"
+
+    finally:
+        if connection:
+            connection.close()
 
-@tool
-def process_refund(order_id, refund_amount, reason):
-    """
-    Logs the action of processing a refund for an order.
-    Args:
-        order_id (int): The unique ID of the order for which the refund is processed.
-        refund_amount (float): The amount to be refunded.
-        reason (str): The reason for initiating the refund.
-    Returns:
-        None: The function logs the action and does not return any value.
-    """
-    details = {
-        "order_id": order_id,
-        "refund_amount": refund_amount,
-        "reason": reason,
-        "action": "Refund Processed"
-    }
-    log_action("process_refund", details)
 
 
-@tool
 def change_address(order_id, new_address):
     """
-    Logs the action of changing the shipping address for an order.
+    Updates the shipping address for a specific order in the database.
     Args:
         order_id (int): The unique ID of the order for which the address is being changed.
         new_address (str): The new shipping address to be updated.
     Returns:
-        None: The function logs the action and does not return any value.
+        str: A confirmation message indicating the update status.
     """
     details = {
         "order_id": order_id,
         "new_address": new_address,
         "action": "Address Changed"
     }
-    log_action("change_address", details)
+
+    try:
+        # Connect to the SQLite database
+        connection = sqlite3.connect("2911_updates.db")  # Replace with your .db file path
+        cursor = connection.cursor()
+
+        # SQL query to update the address
+        query = """
+        UPDATE shipping
+        SET shipping_address = ?
+        WHERE order_id = ?;
+        """
+        
+        # Execute the query with placeholders
+        cursor.execute(query, (new_address, st.session_state.email, order_id))
+        connection.commit()  # Commit the changes
+        
+        return f"Update successful for Order ID {order_id} with new address: {new_address}"
+    
+    except sqlite3.Error as e:
+        return f"An error occurred: {e}"
+    
+    finally:
+        if connection:
+            connection.close()
 
 @tool
 def register_feedback(intent, customer_id, feedback, rating):
@@ -444,7 +495,7 @@ def chatbot_interface():
         prompt = build_prompt(details)
     
 
-    tools = [sql_tool,defer_to_human, process_refund, cancel_order, change_address, register_feedback, days_since]
+    tools = [sql_tool,defer_to_human, cancel_order, change_address, register_feedback, days_since]
     
     chatbot = AzureChatOpenAI(
         model_name=model_name,