Update app.py

app.py

@@ -1,33 +1,57 @@
 from flask import Flask, jsonify, request
-from datetime import datetime, timedelta
+from flask_cors import CORS
+import datetime
 
 app = Flask(__name__)
+CORS(app)  # Enable CORS for Hugging Face compatibility
 
-# Sample data for tables
-tables = {
-    1: {"status": "Available", "available_time": None},
-    2: {"status": "Reserved", "available_time": datetime.now() + timedelta(minutes=10)},
-    3: {"status": "Reserved", "available_time": datetime.now() + timedelta(minutes=20)},
-}
-
-@app.route('/get_table_status/<int:table_id>', methods=['GET'])
-def get_table_status(table_id):
-    if table_id in tables:
-        table = tables[table_id]
-        if table['status'] == 'Reserved' and table['available_time']:
-            available_in = (table['available_time'] - datetime.now()).seconds // 60
-            return jsonify({
-                "status": "Reserved",
-                "available_in": available_in,
-                "message": f"Table {table_id} will be available in {available_in} minutes."
-            })
-        else:
-            return jsonify({
-                "status": "Available",
-                "message": f"Table {table_id} is available."
-            })
-    else:
-        return jsonify({"error": "Table not found"}), 404
-
-if __name__ == '__main__':
-    app.run(debug=True)
+# In-memory storage for table bookings
+bookings = {}  # {table_id: {time_slot: booked}}
+
+# Available time slots (12 PM to 10 PM)
+time_slots = [f"{hour}:00" for hour in range(12, 23)]
+
+# Initialize bookings for 5 tables
+for table_id in range(1, 6):
+    bookings[table_id] = {}
+
+@app.route("/api/availability/<int:table_id>", methods=["GET"])
+def get_availability(table_id):
+    if table_id not in bookings:
+        return jsonify({"error": "Invalid table ID"}), 404
+
+    # Get current hour (for live time comparison)
+    current_hour = datetime.datetime.now().hour
+    booked_slots = bookings[table_id].keys()
+    available_slots = [slot for slot in time_slots if slot not in booked_slots and int(slot.split(":")[0]) >= current_hour]
+
+    # Find next available slot
+    next_slot = available_slots[0] if available_slots else None
+
+    return jsonify({
+        "table_id": table_id,
+        "available_slots": available_slots,
+        "next_slot": next_slot
+    })
+
+@app.route("/api/book", methods=["POST"])
+def book_table():
+    data = request.get_json()
+    table_id = data.get("table_id")
+    time_slot = data.get("time_slot")
+
+    if not table_id or not time_slot:
+        return jsonify({"success": False, "message": "Missing table_id or time_slot"}), 400
+
+    if table_id not in bookings:
+        return jsonify({"success": False, "message": "Invalid table ID"}), 404
+
+    if time_slot in bookings[table_id]:
+        return jsonify({"success": False, "message": "Time slot already booked"}), 400
+
+    # Book the slot
+    bookings[table_id][time_slot] = True
+    return jsonify({"success": True, "message": "Table booked successfully"})
+
+if __name__ == "__main__":
+    app.run(debug=True, host="0.0.0.0", port=5000)