Update

app.py

@@ -1,202 +1,142 @@
 # =============================================
-# app.py v8.0 — RobotAI Server (ESP32 + HF Cloud + Telegram)
-# Author: GPT-5 Assistant for Cường Phan
+# RobotAI v8.5 — All-in-One (UI + ESP32 + Telegram)
+# Author: GPT-5 for Cường Phan
 # =============================================
 
 import os
 import json
-import uuid
-import time
-import queue
 import requests
-import threading
+from flask import Flask, request, jsonify, render_template_string
+from gtts import gTTS
 from datetime import datetime
-from flask import Flask, request, jsonify, send_file
 
 app = Flask(__name__)
 
 # ======================
-# Configuration
+# Config
 # ======================
 HF_TOKEN = os.environ.get("HF_TOKEN", "")
-HF_MODEL = os.environ.get("HF_MODEL", "mistralai/Mistral-7B-Instruct-v0.3")
-HF_STT_MODEL = os.environ.get("HF_STT_MODEL", "openai/whisper-small")
+HF_MODEL = os.environ.get("HF_MODEL", "mistralai/Mistral-7B-Instruct-v0.3")  # model public
 HF_TTS_MODEL = os.environ.get("HF_TTS_MODEL", "espnet/kan-bayashi_ljspeech_vits")
 TELEGRAM_TOKEN = os.environ.get("TELEGRAM_TOKEN", "")
 TELEGRAM_CHAT_ID = os.environ.get("TELEGRAM_CHAT_ID", "")
 
-COMMAND_QUEUE = queue.Queue()
-CONFIG_FILE = "config.json"
 AUDIO_DIR = "audio_cache"
+os.makedirs(AUDIO_DIR, exist_ok=True)
 
-if not os.path.exists(AUDIO_DIR):
-    os.makedirs(AUDIO_DIR)
 
 # ======================
-# Utility Functions
+# Utils
 # ======================
 def log(msg):
-    print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] {msg}")
-
-def save_config(data):
-    with open(CONFIG_FILE, "w", encoding="utf-8") as f:
-        json.dump(data, f, ensure_ascii=False, indent=2)
-
-def load_config():
-    if os.path.exists(CONFIG_FILE):
-        with open(CONFIG_FILE, "r", encoding="utf-8") as f:
-            return json.load(f)
-    return {}
-
-def send_telegram_message(text):
-    if not TELEGRAM_TOKEN or not TELEGRAM_CHAT_ID:
-        return
-    url = f"https://api.telegram.org/bot{TELEGRAM_TOKEN}/sendMessage"
-    payload = {"chat_id": TELEGRAM_CHAT_ID, "text": text}
-    try:
-        requests.post(url, json=payload, timeout=5)
-    except Exception as e:
-        log(f"[Telegram Error] {e}")
+    print(f"[{datetime.now().strftime('%H:%M:%S')}] {msg}")
+
+def send_telegram(text):
+    if TELEGRAM_TOKEN and TELEGRAM_CHAT_ID:
+        url = f"https://api.telegram.org/bot{TELEGRAM_TOKEN}/sendMessage"
+        requests.post(url, json={"chat_id": TELEGRAM_CHAT_ID, "text": text})
 
-def hf_query(model_id, inputs, parameters=None, timeout=60):
-    url = f"https://api-inference.huggingface.co/models/{model_id}"
+def detect_lang(text):
+    vi_chars = "ăâđêôơưáàảãạấầẩẫậắằẳẵặéèẻẽẹếềểễệóòỏõọốồổỗộớờởỡợúùủũụứừửữựýỳỷỹỵ"
+    return "vi" if any(c in text for c in vi_chars) else "en"
+
+def query_huggingface(prompt, max_new_tokens=200):
+    url = f"https://api-inference.huggingface.co/models/{HF_MODEL}"
     headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
-    payload = {"inputs": inputs}
-    if parameters:
-        payload["parameters"] = parameters
+    payload = {"inputs": prompt, "parameters": {"max_new_tokens": max_new_tokens, "temperature": 0.7}}
     try:
-        r = requests.post(url, headers=headers, json=payload, timeout=timeout)
+        r = requests.post(url, headers=headers, json=payload, timeout=90)
+        if r.status_code == 200:
+            out = r.json()
+            if isinstance(out, list) and len(out) > 0:
+                return out[0].get("generated_text", out[0].get("text", ""))
+            elif isinstance(out, dict):
+                return out.get("generated_text", str(out))
+        elif r.status_code == 404:
+            return "⚠️ Model not found. Please check HF_MODEL name."
+        else:
+            return f"⚠️ Error: {r.status_code} - {r.text}"
     except Exception as e:
-        log(f"[HF request exception] {e}")
-        return {"error": str(e)}
-
-    if r.status_code == 200:
-        try:
-            return r.json()
-        except Exception:
-            return {"raw_text": r.text}
-    else:
-        log(f"[HF Error] {r.status_code} - {r.text}")
-        return {"error": r.text, "status_code": r.status_code}
-
-def detect_language(text):
-    if not text:
-        return "en"
-    vi_chars = "àáảãạăằắẳẵặâầấẩẫậđêềếểễệôồốổỗộơờớởỡợưừứửữự"
-    if any(ch in text for ch in vi_chars):
-        return "vi"
-    return "en"
-
-def generate_tts_audio(text, lang="en"):
-    session_id = str(uuid.uuid4())[:8]
-    filename = f"{AUDIO_DIR}/{session_id}.mp3"
-
-    if HF_TTS_MODEL:
-        try:
-            url = f"https://api-inference.huggingface.co/models/{HF_TTS_MODEL}"
-            headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
-            rr = requests.post(url, headers=headers, json={"inputs": text}, timeout=120)
-            if rr.status_code == 200 and rr.content:
-                with open(filename, "wb") as f:
-                    f.write(rr.content)
-                return filename
-        except Exception as e:
-            log(f"[TTS exception] {e}")
+        return f"⚠️ Exception: {e}"
 
+def gen_tts(text, lang="en"):
     try:
-        from gtts import gTTS
-        lang_code = "vi" if lang == "vi" else "en"
-        tts = gTTS(text=text, lang=lang_code)
-        tts.save(filename)
-        return filename
+        path = os.path.join(AUDIO_DIR, f"tts_{int(datetime.now().timestamp())}.mp3")
+        tts = gTTS(text=text, lang="vi" if lang == "vi" else "en")
+        tts.save(path)
+        return f"/tts/{os.path.basename(path)}"
     except Exception as e:
-        log(f"gTTS fallback failed: {e}")
-    return None
+        log(f"TTS error: {e}")
+        return None
+
 
 # ======================
-# API ROUTES
+# ROUTES
 # ======================
 @app.route("/")
 def home():
-    return "<h2>🤖 RobotAI v8.0 Server Running Successfully!</h2>"
-
-@app.route("/model_check")
-def model_check():
-    info = {"HF_MODEL": HF_MODEL, "token_present": bool(HF_TOKEN)}
-    try:
-        r = hf_query(HF_MODEL, "Hello world", parameters={"max_new_tokens": 10}, timeout=10)
-        info["probe"] = r
-    except Exception as e:
-        info["probe_error"] = str(e)
-    return jsonify(info)
+    return render_template_string("""
+    <html><head><title>RobotAI v8.5</title></head>
+    <body style="font-family:sans-serif;text-align:center;">
+    <h2>🤖 RobotAI v8.5 — All-in-One Server</h2>
+    <p>Server is running successfully! Use /chat API or chat below.</p>
+    <form method="post" action="/chat_ui">
+      <textarea name="text" rows="3" cols="50" placeholder="Nhập tiếng Việt hoặc English..."></textarea><br>
+      <button type="submit">Gửi</button>
+    </form>
+    {% if reply %}
+      <p><b>Robot:</b> {{ reply }}</p>
+      {% if tts_url %}<audio controls autoplay><source src="{{ tts_url }}" type="audio/mpeg"></audio>{% endif %}
+    {% endif %}
+    </body></html>
+    """, reply=None)
+
+@app.route("/chat_ui", methods=["POST"])
+def chat_ui():
+    text = request.form.get("text", "")
+    lang = detect_lang(text)
+    prefix = "Trả lời bằng tiếng Việt:" if lang == "vi" else "Answer in English:"
+    reply = query_huggingface(f"{prefix}\n\n{text}")
+    tts_url = gen_tts(reply, lang)
+    return render_template_string("""
+    <html><head><title>RobotAI v8.5</title></head>
+    <body style="font-family:sans-serif;text-align:center;">
+    <h2>🤖 RobotAI v8.5 — Chat</h2>
+    <form method="post" action="/chat_ui">
+      <textarea name="text" rows="3" cols="50">{{text}}</textarea><br>
+      <button type="submit">Gửi</button>
+    </form>
+    <p><b>Robot:</b> {{ reply }}</p>
+    {% if tts_url %}<audio controls autoplay><source src="{{ tts_url }}" type="audio/mpeg"></audio>{% endif %}
+    </body></html>
+    """, text=text, reply=reply, tts_url=tts_url)
 
 @app.route("/api/chat", methods=["POST"])
 def api_chat():
     data = request.get_json(force=True)
-    text = data.get("text", "").strip()
-    if not text:
-        return jsonify({"error": "missing text"}), 400
-    lang = detect_language(text)
-    prefix = "Hãy trả lời bằng tiếng Việt:" if lang == "vi" else "Answer in English:"
-    prompt = f"{prefix}\n\n{text}"
-    r = hf_query(HF_MODEL, prompt, parameters={"max_new_tokens": 200, "temperature": 0.7}, timeout=60)
-    reply = ""
-    if isinstance(r, list) and r and isinstance(r[0], dict):
-        reply = r[0].get("generated_text") or r[0].get("text") or str(r[0])
-    elif isinstance(r, dict) and "generated_text" in r:
-        reply = r["generated_text"]
-    else:
-        reply = str(r)
-
-    tts_file = generate_tts_audio(reply, lang)
-    return jsonify({"reply": reply, "lang": lang, "tts_url": f"/api/tts/{os.path.basename(tts_file)}" if tts_file else None})
-
-@app.route("/api/tts/<filename>")
-def api_tts_file(filename):
-    path = os.path.join(AUDIO_DIR, filename)
-    if os.path.exists(path):
-        return send_file(path, mimetype="audio/mpeg")
-    return jsonify({"error": "file not found"}), 404
+    text = data.get("text", "")
+    lang = detect_lang(text)
+    reply = query_huggingface(text)
+    tts_url = gen_tts(reply, lang)
+    return jsonify({"reply": reply, "tts": tts_url, "lang": lang})
 
 @app.route("/api/presence", methods=["POST"])
 def api_presence():
     data = request.get_json(force=True)
-    note = data.get("note", "Có người xuất hiện!")
-    send_telegram_message(f"🚶 {note}")
-    greeting = "Xin chào! Mình là RobotAI, có thể giúp gì không?"
-    audio = generate_tts_audio(greeting, lang="vi")
-    return jsonify({"greeting": greeting, "tts_url": f"/api/tts/{os.path.basename(audio)}" if audio else None})
-
-@app.route("/api/control", methods=["POST"])
-def api_control():
-    data = request.get_json(force=True)
-    COMMAND_QUEUE.put(data)
-    log(f"Command added: {data}")
-    return jsonify({"status": "queued"})
-
-@app.route("/api/poll_commands")
-def api_poll_commands():
-    cmds = []
-    while not COMMAND_QUEUE.empty():
-        cmds.append(COMMAND_QUEUE.get())
-    return jsonify({"commands": cmds})
-
-@app.route("/api/config", methods=["GET", "POST"])
-def api_config():
-    if request.method == "POST":
-        data = request.get_json(force=True)
-        save_config(data)
-        return jsonify({"status": "saved"})
-    return jsonify(load_config())
+    note = data.get("note", "🚶 Có người vừa đi qua!")
+    send_telegram(note)
+    reply = "Xin chào! Tôi là RobotAI. Có thể giúp gì cho bạn?"
+    tts_url = gen_tts(reply, "vi")
+    return jsonify({"reply": reply, "tts": tts_url})
+
+@app.route("/tts/<fname>")
+def serve_tts(fname):
+    return app.send_static_file(os.path.join(AUDIO_DIR, fname))
+
+@app.route("/model_check")
+def model_check():
+    return jsonify({"HF_MODEL": HF_MODEL, "Token": bool(HF_TOKEN)})
 
-# ======================
-# Background thread
-# ======================
-def heartbeat():
-    while True:
-        log("Heartbeat: RobotAI active")
-        time.sleep(60)
 
 if __name__ == "__main__":
-    threading.Thread(target=heartbeat, daemon=True).start()
-    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)), debug=True)
+    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)), debug=False)