Update api.py

api.py

@@ -232,7 +232,7 @@ def voice_chat():
             audio.save(tmp.name)
             audio_path = tmp.name
 
-        # Step 1️⃣: Transcribe via OpenAI Whisper
+        # Step 1️⃣: Transcribe user speech to text using Whisper
         transcription = client.audio.transcriptions.create(
             model="whisper-1",
             file=open(audio_path, "rb"),
@@ -240,37 +240,65 @@ def voice_chat():
         text = transcription.text.strip()
         print(f"🎤 Transcribed: {text}")
 
-        # Step 2️⃣: Generate answer via GPT
-        response = client.chat.completions.create(
-            model="gpt-4o",
-            messages=[
-                {
-                    "role": "system",
-                    "content": (
-                        "You are CoachingBot, a professional mentor helping users improve their mindset, "
-                        "motivation, and business success. Be clear, empathetic, and practical."
-                    ),
-                },
-                {"role": "user", "content": text},
-            ],
-            max_tokens=700,
-        )
-        answer_text = response.choices[0].message.content.strip()
-
-        # Step 3️⃣: Optional TTS response
-        speech_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
-        with client.audio.speech.with_streaming_response.create(
-            model="gpt-4o-tts",
-            voice="alloy",
-            input=answer_text,
-        ) as speech:
-            speech.stream_to_file(speech_file.name)
-
+        if not text:
+            return jsonify({"error": "Transcription failed or empty"}), 400
+
+        # Step 2️⃣: Retrieve Pinecone context (book knowledge)
+        context, results = "", []
+        try:
+            raw_results = retriever.retrieve(text)
+            MIN_SCORE = 0.02  # slightly lower for better recall
+            results = [r for r in raw_results if r.get("score", 0) >= MIN_SCORE]
+            if results:
+                context = "\n\n---\n\n".join(
+                    [f"(Seite {r['page']}) {r['context']}" for r in results]
+                )
+        except Exception as e:
+            print("⚠️ Retriever error:", e)
+
+        # Step 3️⃣: Choose prompt
+        if context:
+            sys_prompt = system_prompt_book_only()
+            user_prompt = f"Question: {text}\n\nBook context:\n{context}"
+        else:
+            sys_prompt = system_prompt_fallback()
+            user_prompt = text
+
+        # Step 4️⃣: Generate answer with GPT (based on book context)
+        try:
+            response = client.chat.completions.create(
+                model="gpt-4o",
+                messages=[
+                    {"role": "system", "content": sys_prompt},
+                    {"role": "user", "content": user_prompt},
+                ],
+                max_tokens=700,
+            )
+            answer_text = response.choices[0].message.content.strip()
+        except Exception as e:
+            traceback.print_exc()
+            return jsonify({"error": f"GPT generation failed: {e}"}), 500
+
+        # Step 5️⃣: Generate voice reply with GPT TTS
+        try:
+            speech_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
+            with client.audio.speech.with_streaming_response.create(
+                model="gpt-4o-tts",
+                voice="alloy",
+                input=answer_text,
+            ) as speech:
+                speech.stream_to_file(speech_file.name)
+        except Exception as e:
+            traceback.print_exc()
+            return jsonify({"error": f"TTS failed: {e}"}), 500
+
+        # Step 6️⃣: Return transcript + answer + audio
         return jsonify(
             {
                 "transcript": text,
                 "answer": answer_text,
                 "audio_url": f"https://mahmous-chatbot3.hf.space/audio/{os.path.basename(speech_file.name)}",
+                "source": [r.get("page") for r in results if r.get("page")],
             }
         )
 
@@ -279,6 +307,7 @@ def voice_chat():
         return jsonify({"error": str(e)}), 500
 
 
+
 @app.route("/audio/<filename>")
 def serve_audio(filename):
     return send_file(