Spaces:

kcrobot102
/

robotAI

Sleeping

App Files Files Community

kcrobot102 commited on Oct 18, 2025

Commit

78a5fbc

verified ·

1 Parent(s): 09db8af

initial commit

Browse files

Files changed (1) hide show

app.py +192 -190

app.py CHANGED Viewed

@@ -1,4 +1,196 @@
 # app.py — KC Robot AI v7.5 FINAL (auto-model-select, bilingual, TTS fallback, Telegram, ESP32 endpoints)
 # Secrets expected (HF Space -> Settings -> Secrets):
 #   HF_TOKEN (required)
@@ -661,193 +853,3 @@ if __name__ == "__main__":
     app.run(host="0.0.0.0", port=PORT)
-# ==========================================================
-#  KC ROBOT AI - APP.PY (V2.0 MAX FINAL)
-#  Cloud AI Robot with Gemini 2.5 Flash + ESP32 + Telegram
-# ==========================================================
-from flask import Flask, request, jsonify, render_template_string
-from google import genai
-import requests
-import os
-import time
-from gtts import gTTS
-from langdetect import detect
-import tempfile
-import base64
-# ==========================================================
-# CONFIGURATION
-# ==========================================================
-# Load environment variables from secrets (Cloud Run or Hugging Face)
-GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
-GEMINI_MODEL = os.getenv("GEMINI_MODEL", "gemini-2.5-flash")
-TELEGRAM_TOKEN = os.getenv("TELEGRAM_TOKEN")
-TELEGRAM_CHAT_ID = os.getenv("TELEGRAM_CHAT_ID")
-# Create Flask app
-app = Flask(__name__)
-# ==========================================================
-# SETUP GEMINI CLIENT
-# ==========================================================
-if not GEMINI_API_KEY:
-    print("❌ ERROR: No Gemini API Key found. Please add GEMINI_API_KEY in Secrets.")
-    client = None
-else:
-    client = genai.Client(api_key=GEMINI_API_KEY)
-# ==========================================================
-# TELEGRAM UTILITIES
-# ==========================================================
-def send_telegram_message(text):
-    if not TELEGRAM_TOKEN or not TELEGRAM_CHAT_ID:
-        print("⚠️ Telegram not configured.")
-        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:
-        print("Telegram Error:", e)
-# ==========================================================
-# GEMINI AI RESPONSE
-# ==========================================================
-def ask_gemini(prompt: str):
-    if not client:
-        return "⚠️ Gemini API key missing. Please configure in Secrets."
-    try:
-        response = client.models.generate_content(
-            model=GEMINI_MODEL,
-            contents=prompt
-        )
-        if hasattr(response, "text"):
-            return response.text.strip()
-        elif "text" in response:
-            return response["text"].strip()
-        else:
-            return "⚠️ No response text from Gemini."
-    except Exception as e:
-        print("Gemini Error:", e)
-        return f"⚠️ Gemini Error: {e}"
-# ==========================================================
-# LANGUAGE DETECTION & TTS
-# ==========================================================
-def text_to_speech(text):
-    try:
-        lang = detect(text)
-        if lang not in ["vi", "en"]:
-            lang = "en"
-        tts = gTTS(text=text, lang=lang)
-        tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
-        tts.save(tmp.name)
-        with open(tmp.name, "rb") as f:
-            audio_b64 = base64.b64encode(f.read()).decode("utf-8")
-        os.unlink(tmp.name)
-        return audio_b64
-    except Exception as e:
-        print("TTS Error:", e)
-        return None
-# ==========================================================
-# SIMPLE HTML INTERFACE (for testing)
-# ==========================================================
-HTML_PAGE = """
-<!DOCTYPE html>
-<html>
-<head>
-<title>KC Robot AI v2.0</title>
-<style>
-body { font-family: Arial; text-align: center; background-color: #101010; color: white; }
-input, button { padding: 10px; font-size: 16px; margin: 5px; }
-#chat { max-width: 700px; margin: auto; text-align: left; background: #202020; padding: 20px; border-radius: 10px; }
-.msg-user { color: #4af; }
-.msg-bot { color: #fa4; margin-left: 20px; }
-audio { margin-top: 10px; }
-</style>
-</head>
-<body>
-<h1>🤖 KC Robot AI v2.0 MAX FINAL</h1>
-<div id="chat"></div>
-<br>
-<input id="user_input" placeholder="Nói gì đó..." style="width:60%">
-<button onclick="sendMessage()">Gửi</button>
-<script>
-async function sendMessage() {
-  const input = document.getElementById("user_input").value;
-  if (!input) return;
-  const chat = document.getElementById("chat");
-  chat.innerHTML += `<div class='msg-user'><b>Bạn:</b> ${input}</div>`;
-  document.getElementById("user_input").value = "";
-  const res = await fetch("/api/chat", {
-    method: "POST",
-    headers: {"Content-Type": "application/json"},
-    body: JSON.stringify({message: input})
-  });
-  const data = await res.json();
-  chat.innerHTML += `<div class='msg-bot'><b>Robot:</b> ${data.reply}</div>`;
-  if (data.audio) {
-    const audio = document.createElement("audio");
-    audio.src = "data:audio/mp3;base64," + data.audio;
-    audio.controls = true;
-    chat.appendChild(audio);
-  }
-  chat.scrollTop = chat.scrollHeight;
-}
-</script>
-</body>
-</html>
-"""
-@app.route("/")
-def home():
-    return render_template_string(HTML_PAGE)
-# ==========================================================
-# API ENDPOINTS
-# ==========================================================
-@app.route("/api/chat", methods=["POST"])
-def api_chat():
-    data = request.get_json()
-    if not data or "message" not in data:
-        return jsonify({"error": "Missing 'message'"}), 400
-    user_message = data["message"]
-    print(f"🧠 User said: {user_message}")
-    send_telegram_message(f"User: {user_message}")
-    ai_reply = ask_gemini(user_message)
-    send_telegram_message(f"Robot: {ai_reply}")
-    audio_b64 = text_to_speech(ai_reply)
-    return jsonify({"reply": ai_reply, "audio": audio_b64})
-# ESP32 sensor endpoint
-@app.route("/api/sensor", methods=["POST"])
-def sensor_data():
-    data = request.get_json()
-    if not data:
-        return jsonify({"error": "No data"}), 400
-    msg = f"👁️ ESP32 Sensor update: {data}"
-    send_telegram_message(msg)
-    return jsonify({"status": "received"})
-# Health check
-@app.route("/ping")
-def ping():
-    return jsonify({"status": "ok", "model": GEMINI_MODEL})
-# ==========================================================
-# MAIN ENTRY POINT
-# ==========================================================
-if __name__ == "__main__":
-    port = int(os.getenv("PORT", 8080))
-    print(f"🚀 KC Robot AI v2.0 running on port {port}")
-    app.run(host="0.0.0.0", port=port)

+# ==========================================================
+#  KC ROBOT AI - APP.PY (V2.0 MAX FINAL)
+#  Cloud AI Robot with Gemini 2.5 Flash + ESP32 + Telegram
+# ==========================================================
+from flask import Flask, request, jsonify, render_template_string
+from google import genai
+import requests
+import os
+import time
+from gtts import gTTS
+from langdetect import detect
+import tempfile
+import base64
+# ==========================================================
+# CONFIGURATION
+# ==========================================================
+# Load environment variables from secrets (Cloud Run or Hugging Face)
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+GEMINI_MODEL = os.getenv("GEMINI_MODEL", "gemini-2.5-flash")
+TELEGRAM_TOKEN = os.getenv("TELEGRAM_TOKEN")
+TELEGRAM_CHAT_ID = os.getenv("TELEGRAM_CHAT_ID")
+# Create Flask app
+app = Flask(__name__)
+# ==========================================================
+# SETUP GEMINI CLIENT
+# ==========================================================
+if not GEMINI_API_KEY:
+    print("❌ ERROR: No Gemini API Key found. Please add GEMINI_API_KEY in Secrets.")
+    client = None
+else:
+    client = genai.Client(api_key=GEMINI_API_KEY)
+# ==========================================================
+# TELEGRAM UTILITIES
+# ==========================================================
+def send_telegram_message(text):
+    if not TELEGRAM_TOKEN or not TELEGRAM_CHAT_ID:
+        print("⚠️ Telegram not configured.")
+        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:
+        print("Telegram Error:", e)
+# ==========================================================
+# GEMINI AI RESPONSE
+# ==========================================================
+def ask_gemini(prompt: str):
+    if not client:
+        return "⚠️ Gemini API key missing. Please configure in Secrets."
+    try:
+        response = client.models.generate_content(
+            model=GEMINI_MODEL,
+            contents=prompt
+        )
+        if hasattr(response, "text"):
+            return response.text.strip()
+        elif "text" in response:
+            return response["text"].strip()
+        else:
+            return "⚠️ No response text from Gemini."
+    except Exception as e:
+        print("Gemini Error:", e)
+        return f"⚠️ Gemini Error: {e}"
+# ==========================================================
+# LANGUAGE DETECTION & TTS
+# ==========================================================
+def text_to_speech(text):
+    try:
+        lang = detect(text)
+        if lang not in ["vi", "en"]:
+            lang = "en"
+        tts = gTTS(text=text, lang=lang)
+        tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
+        tts.save(tmp.name)
+        with open(tmp.name, "rb") as f:
+            audio_b64 = base64.b64encode(f.read()).decode("utf-8")
+        os.unlink(tmp.name)
+        return audio_b64
+    except Exception as e:
+        print("TTS Error:", e)
+        return None
+# ==========================================================
+# SIMPLE HTML INTERFACE (for testing)
+# ==========================================================
+HTML_PAGE = """
+<!DOCTYPE html>
+<html>
+<head>
+<title>KC Robot AI v2.0</title>
+<style>
+body { font-family: Arial; text-align: center; background-color: #101010; color: white; }
+input, button { padding: 10px; font-size: 16px; margin: 5px; }
+#chat { max-width: 700px; margin: auto; text-align: left; background: #202020; padding: 20px; border-radius: 10px; }
+.msg-user { color: #4af; }
+.msg-bot { color: #fa4; margin-left: 20px; }
+audio { margin-top: 10px; }
+</style>
+</head>
+<body>
+<h1>🤖 KC Robot AI v2.0 MAX FINAL</h1>
+<div id="chat"></div>
+<br>
+<input id="user_input" placeholder="Nói gì đó..." style="width:60%">
+<button onclick="sendMessage()">Gửi</button>
+<script>
+async function sendMessage() {
+  const input = document.getElementById("user_input").value;
+  if (!input) return;
+  const chat = document.getElementById("chat");
+  chat.innerHTML += `<div class='msg-user'><b>Bạn:</b> ${input}</div>`;
+  document.getElementById("user_input").value = "";
+  const res = await fetch("/api/chat", {
+    method: "POST",
+    headers: {"Content-Type": "application/json"},
+    body: JSON.stringify({message: input})
+  });
+  const data = await res.json();
+  chat.innerHTML += `<div class='msg-bot'><b>Robot:</b> ${data.reply}</div>`;
+  if (data.audio) {
+    const audio = document.createElement("audio");
+    audio.src = "data:audio/mp3;base64," + data.audio;
+    audio.controls = true;
+    chat.appendChild(audio);
+  }
+  chat.scrollTop = chat.scrollHeight;
+}
+</script>
+</body>
+</html>
+"""
+@app.route("/")
+def home():
+    return render_template_string(HTML_PAGE)
+# ==========================================================
+# API ENDPOINTS
+# ==========================================================
+@app.route("/api/chat", methods=["POST"])
+def api_chat():
+    data = request.get_json()
+    if not data or "message" not in data:
+        return jsonify({"error": "Missing 'message'"}), 400
+    user_message = data["message"]
+    print(f"🧠 User said: {user_message}")
+    send_telegram_message(f"User: {user_message}")
+    ai_reply = ask_gemini(user_message)
+    send_telegram_message(f"Robot: {ai_reply}")
+    audio_b64 = text_to_speech(ai_reply)
+    return jsonify({"reply": ai_reply, "audio": audio_b64})
+# ESP32 sensor endpoint
+@app.route("/api/sensor", methods=["POST"])
+def sensor_data():
+    data = request.get_json()
+    if not data:
+        return jsonify({"error": "No data"}), 400
+    msg = f"👁️ ESP32 Sensor update: {data}"
+    send_telegram_message(msg)
+    return jsonify({"status": "received"})
+# Health check
+@app.route("/ping")
+def ping():
+    return jsonify({"status": "ok", "model": GEMINI_MODEL})
+# ==========================================================
+# MAIN ENTRY POINT
+# ==========================================================
+if __name__ == "__main__":
+    port = int(os.getenv("PORT", 8080))
+    print(f"🚀 KC Robot AI v2.0 running on port {port}")
+    app.run(host="0.0.0.0", port=port)
 # app.py — KC Robot AI v7.5 FINAL (auto-model-select, bilingual, TTS fallback, Telegram, ESP32 endpoints)
 # Secrets expected (HF Space -> Settings -> Secrets):
 #   HF_TOKEN (required)
     app.run(host="0.0.0.0", port=PORT)