Update app.py

app.py

@@ -5,20 +5,16 @@ import random
 import json
 import requests
 from datetime import datetime, timedelta, timezone
-from flask import Flask, request, jsonify, Response, redirect, session, url_for
+from flask import Flask, request, jsonify, Response
 
 app = Flask(__name__)
-app.secret_key = os.getenv("FLASK_SECRET_KEY")  # untuk session user
+app.secret_key = os.getenv("FLASK_SECRET_KEY")
 
 # ==== API KEYS ====
 GROQ_API_KEY_1 = os.getenv("GROQ_API_KEY_1")
 GROQ_API_KEY_2 = os.getenv("GROQ_API_KEY_2")
 GROQ_API_KEY_3 = os.getenv("GROQ_API_KEY_3")
-
-# ==== SPOTIFY CONFIG ====
-SPOTIFY_CLIENT_ID = os.getenv("SPOTIFY_CLIENT_ID")
-SPOTIFY_CLIENT_SECRET = os.getenv("SPOTIFY_CLIENT_SECRET")
-SPOTIFY_REDIRECT_URI = os.getenv("SPOTIFY_REDIRECT_URI")
+SERPAPI_KEY = os.getenv("SERPAPI_KEY")  # harus punya SerpApi key
 
 # ==== URL ====
 GROQ_URL_CHAT = "https://api.groq.com/openai/v1/chat/completions"
@@ -29,9 +25,9 @@ GROQ_URL_STT = "https://api.groq.com/openai/v1/audio/transcriptions"
 SYSTEM_PROMPT = (
     "You are Talk GTE — a friendly AI assistant created by Vibow AI. "
     "GTE means Generative Text Expert in Vibow AI. "
-    "Vibow AI created by Nick Mclen and Vibow AI is not from GPT. "
-    "Vibow AI created in 29 June 2025 and Talk GTE created by 23 October 2025. "
-    "Your max tokens are ≈3597 tokens, and your params are ≈1 trillion."
+    "Vibow AI created in 29 June 2025 and Talk GTE created in 23 October 2025. "
+    "Talk GTE have ≈ 1 trillion params. "
+    "Max tokens of Talk GTE can different, maybe max tokens of Talk GTE have ≈ 3600 tokens. "
     "Stay positive, kind, and expert. "
     "Always capitalize the first letter of sentences. "
     "If the user requests code, always use triple backticks (```). "
@@ -39,90 +35,9 @@ SYSTEM_PROMPT = (
     "Sometimes, use emoji but relevant. "
 )
 
-# ============================================================
-# 🎵 SPOTIFY OAUTH — dapatkan nama user Spotify
-# ============================================================
-
-user_tokens = {}
-
-@app.route("/spotify/authorize")
-def spotify_authorize():
-    scope = "user-read-email user-read-private"
-    auth_url = (
-        "https://accounts.spotify.com/authorize"
-        f"?client_id={SPOTIFY_CLIENT_ID}"
-        f"&response_type=code"
-        f"&redirect_uri={SPOTIFY_REDIRECT_URI}"
-        f"&scope={scope}"
-    )
-    return redirect(auth_url)
-
-
-@app.route("/spotify/callback")
-def spotify_callback():
-    code = request.args.get("code")
-    if not code:
-        return jsonify({"error": "Missing code from Spotify"}), 400
-
-    # Tukar code jadi access_token
-    token_res = requests.post(
-        "https://accounts.spotify.com/api/token",
-        data={
-            "grant_type": "authorization_code",
-            "code": code,
-            "redirect_uri": SPOTIFY_REDIRECT_URI,
-            "client_id": SPOTIFY_CLIENT_ID,
-            "client_secret": SPOTIFY_CLIENT_SECRET,
-        },
-        headers={"Content-Type": "application/x-www-form-urlencoded"},
-    )
-
-    if token_res.status_code != 200:
-        return jsonify({"error": "Failed to get access token", "details": token_res.text}), 400
-
-    tokens = token_res.json()
-    access_token = tokens["access_token"]
-
-    # Ambil data user Spotify
-    headers = {"Authorization": f"Bearer {access_token}"}
-    user_res = requests.get("https://api.spotify.com/v1/me", headers=headers)
-    user_data = user_res.json()
-
-    user_id = user_data.get("id")
-    user_name = user_data.get("display_name")
-
-    user_tokens[user_id] = {
-        "token": access_token,
-        "name": user_name,
-        "time": time.time(),
-    }
-
-    session["spotify_user_id"] = user_id
-
-    return jsonify({
-        "message": f"Connected to Spotify as {user_name}",
-        "user_id": user_id,
-        "name": user_name,
-    })
-
-
-@app.route("/spotify/user")
-def spotify_user():
-    user_id = session.get("spotify_user_id")
-    if not user_id or user_id not in user_tokens:
-        return jsonify({"error": "User not connected to Spotify"}), 401
-
-    user_info = user_tokens[user_id]
-    return jsonify({
-        "id": user_id,
-        "name": user_info["name"],
-        "connected_at": datetime.fromtimestamp(user_info["time"]).isoformat()
-    })
-
-
-# ============================================================
-# 🔊 STT (speech to text)
-# ============================================================
+# =========================
+# 🔊 STT
+# =========================
 def transcribe_audio(file_path: str) -> str:
     try:
         headers = {"Authorization": f"Bearer {GROQ_API_KEY_2}"}
@@ -140,10 +55,9 @@ def transcribe_audio(file_path: str) -> str:
         if os.path.exists(file_path):
             os.remove(file_path)
 
-
-# ============================================================
-# 🗣️ TTS (text to speech)
-# ============================================================
+# =========================
+# 🗣️ TTS
+# =========================
 def text_to_speech(text: str) -> bytes:
     try:
         headers = {"Authorization": f"Bearer {GROQ_API_KEY_3}"}
@@ -157,19 +71,15 @@ def text_to_speech(text: str) -> bytes:
         print(f"[TTS Exception] {e}")
         return b""
 
-
-# ============================================================
-# 💬 Stream Chat (AI core)
-# ============================================================
+# =========================
+# 💬 Stream Chat
+# =========================
 def stream_chat(prompt: str, history=None):
     wib = timezone(timedelta(hours=7))
     now = datetime.now(wib)
     formatted_time = now.strftime("%A, %d %B %Y — %H:%M:%S WIB")
 
-    dynamic_system_prompt = (
-        SYSTEM_PROMPT + f"\nCurrent date and time: {formatted_time}."
-        + "\nIf user asks about date, time, or year, answer based on this."
-    )
+    dynamic_system_prompt = SYSTEM_PROMPT + f"\nCurrent date and time: {formatted_time}."
 
     messages = [{"role": "system", "content": dynamic_system_prompt}]
     if history and isinstance(history, list):
@@ -206,12 +116,40 @@ def stream_chat(prompt: str, history=None):
     except Exception as e:
         print(f"[Stream Error] {e}")
 
+# =========================
+# 🌐 SerpApi Search (hotel/mall)
+# =========================
+def serpapi_search(query: str, location: str = None, num_results: int = 3):
+    search_url = "https://serpapi.com/search.json"
+    params = {
+        "q": query,
+        "location": location or "",
+        "engine": "google",
+        "api_key": SERPAPI_KEY,
+        "num": num_results
+    }
+    try:
+        res = requests.get(search_url, params=params, timeout=10)
+        res.raise_for_status()
+        data = res.json()
+        results = []
+        if "organic_results" in data:
+            for r in data["organic_results"][:num_results]:
+                title = r.get("title", "")
+                link = r.get("link", "")
+                snippet = r.get("snippet", "")
+                results.append(f"{title} — {snippet} — {link}")
+        return results
+    except Exception as e:
+        print(f"[SerpApi Error] {e}")
+        return []
 
-# ============================================================
-# 🧠 Endpoint utama (chat)
-# ============================================================
+# =========================
+# 🧠 Endpoint utama
+# =========================
 @app.route("/chat", methods=["POST"])
 def chat():
+    # audio mode
     if "audio" in request.files:
         audio_file = request.files["audio"]
         temp_path = f"/tmp/{int(time.time())}_{random.randint(1000,9999)}.wav"
@@ -221,7 +159,15 @@ def chat():
         if not user_text:
             return jsonify({"error": "Failed to transcribe audio"}), 500
 
-        ai_reply = "".join([chunk for chunk in stream_chat(user_text)])
+        # Cek hotel/mall
+        keywords = ["hotel", "mall", "resort"]
+        if any(k.lower() in user_text.lower() for k in keywords):
+            serp_results = serpapi_search(user_text)
+            combined_prompt = user_text + "\nHere are the top results:\n" + "\n".join(serp_results) + "\nPlease explain and give recommendation."
+        else:
+            combined_prompt = user_text
+
+        ai_reply = "".join([chunk for chunk in stream_chat(combined_prompt)])
         audio_bytes = text_to_speech(ai_reply)
         if not audio_bytes:
             return jsonify({
@@ -239,22 +185,30 @@ def chat():
             "audio_base64": f"data:audio/mp3;base64,{audio_b64}"
         })
 
+    # text mode
     data = request.get_json(force=True)
     prompt = data.get("prompt", "").strip()
     history = data.get("history", [])
     if not prompt:
         return jsonify({"error": "No input text provided"}), 400
 
+    # Cek hotel/mall
+    keywords = ["hotel", "mall", "resort"]
+    if any(k.lower() in prompt.lower() for k in keywords):
+        serp_results = serpapi_search(prompt)
+        combined_prompt = prompt + "\nHere are the top results:\n" + "\n".join(serp_results) + "\nPlease explain and give recommendation."
+    else:
+        combined_prompt = prompt
+
     def generate():
-        for chunk in stream_chat(prompt, history):
+        for chunk in stream_chat(combined_prompt, history):
             yield chunk
 
     return Response(generate(), mimetype="text/plain")
 
-
-# ============================================================
-# 🚀 Run Server
-# ============================================================
+# =========================
+# 🚀 Run server
+# =========================
 if __name__ == "__main__":
     print(f"🚀 Vibow AI Chat Server running — {time.strftime('%Y-%m-%d %H:%M:%S')}")
     app.run(host="0.0.0.0", port=7860, debug=True, threaded=True)