Update app.py

app.py

@@ -14,7 +14,7 @@ app.secret_key = os.getenv("FLASK_SECRET_KEY")
 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")
-SERPAPI_KEY = os.getenv("SERPAPI_KEY")  # harus punya SerpApi key
+SERPAPI_KEY = os.getenv("SERPAPI_KEY")
 
 # ==== URL ====
 GROQ_URL_CHAT = "https://api.groq.com/openai/v1/chat/completions"
@@ -27,7 +27,7 @@ SYSTEM_PROMPT = (
     "GTE means Generative Text Expert in Vibow AI. "
     "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. "
+   "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 (```). "
@@ -36,7 +36,7 @@ SYSTEM_PROMPT = (
 )
 
 # =========================
-# 🔊 STT
+# 🎤 STT
 # =========================
 def transcribe_audio(file_path: str) -> str:
     try:
@@ -72,18 +72,57 @@ def text_to_speech(text: str) -> bytes:
         return b""
 
 # =========================
-# 💬 Stream Chat
+# 🌐 SerpAPI Search (now with images)
+# =========================
+def serpapi_search(query: str, location: str = None, num_results: int = 3):
+    url = "https://serpapi.com/search.json"
+    params = {
+        "q": query,
+        "location": location or "",
+        "engine": "google",
+        "api_key": SERPAPI_KEY,
+        "num": num_results
+    }
+    try:
+        r = requests.get(url, params=params, timeout=10)
+        r.raise_for_status()
+        data = r.json()
+
+        results = []
+        images = []
+
+        # text results
+        if "organic_results" in data:
+            for item in data["organic_results"][:num_results]:
+                results.append({
+                    "title": item.get("title", ""),
+                    "snippet": item.get("snippet", ""),
+                    "link": item.get("link", "")
+                })
+
+        # image results
+        if "images_results" in data:
+            for img in data["images_results"][:3]:
+                images.append(img.get("thumbnail", ""))
+
+        return {"results": results, "images": images}
+
+    except Exception as e:
+        print(f"[SerpAPI Error] {e}")
+        return {"results": [], "images": []}
+
+# =========================
+# 💬 Chat Stream
 # =========================
 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}."
+    sys_prompt = SYSTEM_PROMPT + f"\nCurrent date and time: {formatted_time}."
 
-    messages = [{"role": "system", "content": dynamic_system_prompt}]
-    if history and isinstance(history, list):
-        messages.extend(history)
+    messages = [{"role": "system", "content": sys_prompt}]
+    if history: messages += history
     messages.append({"role": "user", "content": prompt})
 
     payload = {
@@ -99,116 +138,56 @@ def stream_chat(prompt: str, history=None):
     try:
         with requests.post(GROQ_URL_CHAT, headers=headers, json=payload, stream=True) as r:
             for line in r.iter_lines():
-                if not line:
-                    continue
+                if not line: continue
                 line = line.decode("utf-8")
                 if line.startswith("data: "):
                     data = line[6:]
-                    if data == "[DONE]":
-                        break
+                    if data == "[DONE]": break
                     try:
-                        chunk = json.loads(data)
-                        delta = chunk["choices"][0]["delta"].get("content", "")
-                        if delta:
-                            yield delta
-                    except:
-                        continue
+                        delta = json.loads(data)["choices"][0]["delta"].get("content", "")
+                        if delta: yield delta
+                    except: continue
     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
 # =========================
 @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"
-        audio_file.save(temp_path)
-
-        user_text = transcribe_audio(temp_path)
-        if not user_text:
-            return jsonify({"error": "Failed to transcribe audio"}), 500
-
-        # 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({
-                "mode": "voice",
-                "transcript": user_text,
-                "reply_text": ai_reply,
-                "error": "TTS failed"
-            }), 500
-
-        audio_b64 = base64.b64encode(audio_bytes).decode("utf-8")
-        return jsonify({
-            "mode": "voice",
-            "transcript": user_text,
-            "reply_text": ai_reply,
-            "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."
+    if not prompt: return jsonify({"error": "No input text provided"}), 400
+
+    keywords = ["hotel", "mall", "resort", "tempat wisata", "villa"]
+
+    # add search results if needed
+    if any(k in prompt.lower() for k in keywords):
+        serp = serpapi_search(prompt)
+        text = "\n".join([f"{r['title']} — {r['snippet']} — {r['link']}" for r in serp["results"]])
+        imgs = "\n".join(serp["images"])
+
+        combined = (
+            prompt
+            + "\n\nGoogle Results:\n"
+            + text
+            + "\n\nImages:\n"
+            + imgs
+            + "\n\nExplain & recommend."
+        )
     else:
-        combined_prompt = prompt
+        combined = prompt
 
     def generate():
-        for chunk in stream_chat(combined_prompt, history):
+        for chunk in stream_chat(combined, history):
             yield chunk
 
     return Response(generate(), mimetype="text/plain")
 
 # =========================
-# 🚀 Run server
+# 🚀 Run
 # =========================
 if __name__ == "__main__":
-    print(f"🚀 Vibow AI Chat Server running — {time.strftime('%Y-%m-%d %H:%M:%S')}")
+    print("🚀 Vibow AI Chat Server started")
     app.run(host="0.0.0.0", port=7860, debug=True, threaded=True)