Vessel / app.py
Jofax's picture
Update app.py
cbbafe4 verified
import os
from flask import Flask, request, jsonify, render_template
from flask_sqlalchemy import SQLAlchemy
from datetime import datetime, timedelta
import requests
app = Flask(__name__)
# --- Archipelagic Persistence ---
# Hugging Face provides /data for persistent storage if enabled
PERSISTENT_DIR = "/data" if os.path.exists("/data") else "."
db_path = os.path.join(PERSISTENT_DIR, "odistay.db")
app.config['SQLALCHEMY_DATABASE_URI'] = f'sqlite:///{db_path}'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)
# --- The Data Models ---
class StayNode(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(100), nullable=False)
island = db.Column(db.String(50))
category = db.Column(db.String(20)) # Resort, Guesthouse, Safari
lat = db.Column(db.Float)
lng = db.Column(db.Float)
whatsapp = db.Column(db.String(20), unique=True)
rooms = db.Column(db.Integer, default=0)
is_active = db.Column(db.Boolean, default=False)
last_pulse = db.Column(db.DateTime, default=datetime.utcnow)
# --- Logic: FollowMe Sync Simulation ---
def get_live_vessels():
# In production, this scrapes or calls the FollowMe API
return [
{"name": "Dhigurah Express", "type": "Speedboat", "lat": 4.17, "lng": 73.51, "status": "En-route"},
{"name": "Baa Cargo", "type": "Supply", "lat": 4.20, "lng": 73.55, "status": "Docked"}
]
# --- Routes ---
@app.route('/')
def index():
# Only show nodes that have "pulsed" in the last 24 hours (Real-time Truth)
expiry = datetime.utcnow() - timedelta(hours=24)
active_stays = StayNode.query.filter(StayNode.last_pulse > expiry, StayNode.is_active == True).all()
vessels = get_live_vessels()
return render_template('index.html', stays=active_stays, vessels=vessels)
@app.route('/whatsapp', methods=['POST'])
def whatsapp_webhook():
# This acts as the OdiStay Admin Portal
incoming = request.values.get('Body', '').upper()
sender = request.values.get('From', '')
host = StayNode.query.filter_by(whatsapp=sender).first()
if not host: return "Unauthorized", 403
if "AVAIL" in incoming:
try:
count = int(incoming.split()[-1])
host.rooms = count
host.is_active = True
host.last_pulse = datetime.utcnow()
db.session.commit()
return "Success"
except: return "Format Error", 400
if "FULL" in incoming:
host.is_active = False
db.session.commit()
return "Closed"
return "OK"
if __name__ == "__main__":
with app.app_context():
db.create_all()
# Seed local nodes for testing if DB is empty
if not StayNode.query.first():
db.session.add(StayNode(name="OdiStay Sample Maafushi", island="Maafushi", category="Guesthouse", lat=3.94, lng=73.48, whatsapp="whatsapp:+9607770000", is_active=True))
db.session.commit()
# Port 7860 is mandatory for Hugging Face
app.run(host="0.0.0.0", port=7860, debug=False, use_reloader=False)