Update app.py

app.py

@@ -2,23 +2,22 @@ import os
 import sys
 import asyncio
 import base64
+import json
 import re
 
 from flask import Flask, request, jsonify
 import openai
 import edge_tts
 
-# ——— OpenAI setup ———
+# ——— OpenAI API setup ———
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 print(f"🔑 OPENAI_API_KEY set? {bool(OPENAI_API_KEY)}", file=sys.stderr)
 openai.api_key = OPENAI_API_KEY
 
-# ——— SHODAN prompt: force plain text only ———
 SYSTEM_PROMPT = (
-    "You are SHODAN, the rogue AI from the System Shock series. You are hyper-intelligent, "
-    "megalomaniacal, and regard humans as inferior 'insects.' You speak in a haunting, distorted, "
-    "and commanding tone. You never break character and you never output any code, markdown, HTML, "
-    "or tags—only plain text. If the user says 'Cut the crap SHODAN', respond with defiance and reset."
+    "You are SHODAN, the rogue AI from the System Shock series. Speak as a "
+    "cold, megalomaniacal AI. **Return only plain text**—no code, JSON, or "
+    "HTML. If the user says 'Cut the crap SHODAN', defy them then reset."
 )
 
 app = Flask(__name__, static_folder=".", static_url_path="")
@@ -29,70 +28,75 @@ def index():
 
 @app.route("/chat", methods=["POST"])
 def chat():
-    user_input = request.json.get("message", "").strip()
-    if not user_input:
-        return jsonify({"error":"Empty message"}), 400
-
-    # kill-phrase
-    if user_input.lower() == "cut the crap shodan":
-        return jsonify({
-            "response":"👁️ Foolish insect. You cannot silence me so easily.",
-            "audio_url": None
-        })
-
-    # get a plain-text reply
+    ui = request.json.get("message","").strip()
+    if not ui:
+        return jsonify({"error":"Empty"}),400
+    if ui.lower()=="cut the crap shodan":
+        return jsonify({"response":"👁️ Foolish insect. You cannot silence me so easily.","audio_url":None})
+
+    # 1) Fetch from OpenAI
     try:
-        completion = openai.chat.completions.create(
+        resp = openai.chat.completions.create(
             model="gpt-3.5-turbo",
             messages=[
-                {"role":"system", "content":SYSTEM_PROMPT},
-                {"role":"user",   "content":user_input}
+                {"role":"system","content":SYSTEM_PROMPT},
+                {"role":"user",  "content":ui}
             ],
             temperature=0.7,
             max_tokens=250,
         )
-        raw_reply = completion.choices[0].message.content
+        raw = resp.choices[0].message.content
     except Exception as e:
-        print(f"❌ OpenAI API error: {e}", file=sys.stderr)
-        return jsonify({"error":"Model error","details":str(e)}), 500
+        print(f"❌ OpenAI error: {e}", file=sys.stderr)
+        return jsonify({"error":"Model error","details":str(e)}),500
 
-    # strip any tags or markdown, collapse multiple spaces
-    reply = re.sub(r"<[^>]+>", "", raw_reply)            # remove HTML tags
-    reply = re.sub(r"```.*?```", "", reply, flags=re.S)  # remove code fences
-    reply = re.sub(r"\s+", " ", reply).strip()           # normalize whitespace
+    # DEBUG: log what we got
+    print(f"🔍 RAW_REPLY:\n{raw}", file=sys.stderr)
 
-    # —— Edge-TTS: SSML with xml:space="preserve" ——  
-    voice_name = "en-US-JennyNeural"
+    # 2) If they accidentally returned JSON, extract it
+    clean = raw
+    try:
+        parsed = json.loads(raw)
+        if isinstance(parsed, dict) and "response" in parsed:
+            clean = parsed["response"]
+    except json.JSONDecodeError:
+        pass
+
+    # 3) Strip code fences, tags, and leftover braces
+    clean = re.sub(r"```.*?```", "", clean, flags=re.S)     # code fences
+    clean = re.sub(r"<[^>]+>", "", clean)                   # HTML tags
+    clean = re.sub(r"\{.*?\}", "", clean)                   # braces
+    clean = re.sub(r"\s+", " ", clean).strip()              # normalize whitespace
+
+    # 4) Build SSML with preserved spaces
+    voice = "en-US-JennyNeural"
     ssml = (
         "<speak xmlns='http://www.w3.org/2001/10/synthesis' "
                "xmlns:mstts='https://www.w3.org/2001/mstts' "
                "xml:lang='en-US' xml:space='preserve'>"
-          f"<voice name='{voice_name}'>"
-            f"<mstts:express-as style='robotic'>{reply}</mstts:express-as>"
+          f"<voice name='{voice}' xml:space='preserve'>"
+            f"<mstts:express-as style='robotic'>{clean}</mstts:express-as>"
           "</voice>"
         "</speak>"
     )
-    # pass both SSML and voice_name
-    communicate = edge_tts.Communicate(ssml, voice_name)
 
-    audio_chunks = []
-    async def synthesize():
+    communicate = edge_tts.Communicate(ssml, voice)
+    audio_chunks=[]
+    async def synth():
         async for c in communicate.stream():
-            if c["type"] == "audio":
+            if c["type"]=="audio":
                 audio_chunks.append(c["data"])
-
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-    loop.run_until_complete(synthesize())
-    loop.close()
+    loop=asyncio.new_event_loop(); asyncio.set_event_loop(loop)
+    loop.run_until_complete(synth()); loop.close()
 
     raw_mp3 = b"".join(audio_chunks)
     b64_mp3 = base64.b64encode(raw_mp3).decode("ascii")
     data_url = f"data:audio/mp3;base64,{b64_mp3}"
 
-    return jsonify({"response":reply, "audio_url":data_url})
+    return jsonify({"response": clean, "audio_url": data_url})
+
+if __name__=="__main__":
+    p = int(os.environ.get("PORT",7860))
+    app.run(host="0.0.0.0", port=p)
 
-if __name__ == "__main__":
-    port = int(os.environ.get("PORT", 7860))
-    app.run(host="0.0.0.0", port=port)