|
|
from flask import Flask, request, jsonify, make_response |
|
|
import os, time, html, requests, json |
|
|
from itsdangerous import BadSignature, URLSafeSerializer |
|
|
|
|
|
SECRET_KEY = os.getenv("SECRET_KEY", "change-me") |
|
|
TOKEN_TTL = int(os.getenv("TOKEN_TTL", "3600")) |
|
|
CHAT_MODEL_URL = os.getenv("CHAT_MODEL_URL", "") |
|
|
|
|
|
app = Flask(__name__) |
|
|
signer = URLSafeSerializer(SECRET_KEY, salt="jarvis-facegate") |
|
|
|
|
|
def verify_token(token: str): |
|
|
try: |
|
|
data = signer.loads(token) |
|
|
if time.time() - float(data.get("ts", 0)) > TOKEN_TTL: |
|
|
return None |
|
|
return data |
|
|
except BadSignature: |
|
|
return None |
|
|
|
|
|
def shell(name=""): |
|
|
safe = html.escape(name or "Agent") |
|
|
return f"""<!doctype html><html><head><meta charset=utf-8> |
|
|
<meta name=viewport content="width=device-width,initial-scale=1"/> |
|
|
<title>Jarvis Chat</title> |
|
|
<style> |
|
|
body{{margin:0;background:#0a0f18;color:#e8f3ff;font-family:ui-sans-serif,system-ui}} |
|
|
.wrap{{max-width:900px;margin:5vh auto;padding:24px}} |
|
|
.hud{{border:1px solid #3ee7ff4f;border-radius:16px;padding:16px;background: |
|
|
linear-gradient(180deg,rgba(10,16,24,.85),rgba(10,16,24,.65));box-shadow:0 20px 60px rgba(0,0,0,.5)}} |
|
|
h1{{margin:0 0 8px}} .muted{{color:#a9c2d0}} |
|
|
.row{{display:flex;gap:8px;margin-top:12px}} |
|
|
input,button{{padding:10px;border-radius:10px;border:1px solid #3ee7ff30;background:#0f1724;color:#e8f3ff}} |
|
|
button{{background:linear-gradient(90deg,#3ee7ff,#7bf5c8);color:#0a0f18;border:none;font-weight:800}} |
|
|
#log{{white-space:pre-wrap;min-height:180px;margin-top:12px;background:#0f1724;border:1px solid #3ee7ff21;border-radius:12px;padding:12px}} |
|
|
</style></head> |
|
|
<body><div class=wrap> |
|
|
<div class=hud> |
|
|
<h1>Jarvis Chat</h1> |
|
|
<div class=muted>Welcome, {safe}. You are now connected.</div> |
|
|
<div id=log></div> |
|
|
<div class=row> |
|
|
<input id=msg placeholder="Type your message..." style="flex:1"/> |
|
|
<button onclick="send()">Send</button> |
|
|
</div> |
|
|
</div> |
|
|
</div> |
|
|
<script> |
|
|
async function send(){{ |
|
|
const el = document.getElementById('msg'); |
|
|
const txt = el.value.trim(); |
|
|
if(!txt) return; |
|
|
el.value=''; |
|
|
const res = await fetch('/api/chat', {{method:'POST', headers:{{'Content-Type':'application/json'}}, body: JSON.stringify({{q:txt}})}}); |
|
|
const data = await res.json(); |
|
|
const log = document.getElementById('log'); |
|
|
log.textContent += "\\nYou: " + txt + "\\nJarvis: " + (data.a || data.error || '...') + "\\n"; |
|
|
}} |
|
|
</script> |
|
|
</body></html>""" |
|
|
|
|
|
@app.get("/") |
|
|
def home(): |
|
|
token = request.args.get("token","") |
|
|
info = verify_token(token) if token else None |
|
|
if not info: |
|
|
return make_response("<h3>Access denied. Missing/invalid token.</h3>", 401) |
|
|
name = info.get("name","User") |
|
|
return make_response(shell(name), 200, {"Content-Type":"text/html; charset=utf-8"}) |
|
|
|
|
|
@app.post("/api/chat") |
|
|
def api_chat(): |
|
|
|
|
|
q = (request.json or {}).get("q","").strip() |
|
|
if not q: |
|
|
return jsonify({"error":"empty message"}), 400 |
|
|
|
|
|
if CHAT_MODEL_URL: |
|
|
try: |
|
|
r = requests.post(CHAT_MODEL_URL, json={"inputs": q}, timeout=60) |
|
|
if r.ok: |
|
|
|
|
|
data = r.json() |
|
|
a = data.get("generated_text") or data.get("answer") or str(data) |
|
|
return jsonify({"a": a}) |
|
|
return jsonify({"error": f"model status {r.status_code}"}), 502 |
|
|
except Exception as e: |
|
|
return jsonify({"error": f"upstream error: {e}"}), 502 |
|
|
else: |
|
|
|
|
|
return jsonify({"a": f"(stub) You said: {q}"}), 200 |
|
|
|
|
|
if __name__ == "__main__": |
|
|
app.run(host="0.0.0.0", port=7860) |
|
|
|
