Update app.py

app.py

@@ -1,20 +1,18 @@
 import streamlit as st
-from piper import PiperVoice
 import base64, tempfile, requests, os, re
-import whisper   # local Whisper model
+import whisper
+from piper import PiperVoice
+import numpy as np
+import soundfile as sf
 
-# ────────────────────────────────────────────────────
+# ─────────────────────────────────────────────────────────────
 st.set_page_config(page_title="Med11 – AI Emergency Guide", page_icon="🩺")
 
-# Groq key (unchanged)
 GROQ_API_KEY = st.secrets["GROQ_API_KEY"]
-
-# Session flags
 st.session_state.setdefault("show_numbers", False)
 st.session_state.setdefault("upload_id", 0)
 st.session_state.setdefault("chat_history", [])
 
-# ────────────────────────────────────────────────────
 @st.cache_resource(show_spinner="Loading Whisper model…")
 def _load_whisper():
     return whisper.load_model("base")
@@ -26,11 +24,7 @@ def groq_reply(prompt: str, model="llama3-8b-8192"):
         "messages": [
             {
                 "role": "system",
-                "content": (
-                    "You are an emergency-response medic. "
-                    "Classify the emergency, rate severity, and give clear, step-by-step first-aid. "
-                    "Limit to 120 words."
-                ),
+                "content": "You are an emergency-response medic. Classify the emergency, rate severity, and give clear, step-by-step first-aid. Limit to 120 words.",
             },
             *st.session_state.chat_history,
             {"role": "user", "content": prompt},
@@ -40,7 +34,7 @@ def groq_reply(prompt: str, model="llama3-8b-8192"):
     }
     r = requests.post(
         url, json=payload,
-        headers={"Authorization": f"Bearer {GROQ_API_KEY}"},
+        headers={"Authorization": f"Bearer {GROQ_API_KEY.strip()}"},
         timeout=45,
     )
     r.raise_for_status()
@@ -51,40 +45,34 @@ def transcribe_audio(file_path: str) -> str:
     result = model.transcribe(file_path)
     return result["text"]
 
-# ──────────────────────────────────────────
-# Markdown sanitization for TTS
-# ──────────────────────────────────────────
-
 def clean_text_for_tts(text: str) -> str:
-    text = re.sub(r"\*\*|\*|__|_", "", text)              # bold/italic
-    text = re.sub(r"[|~>#]", "", text)                       # misc. MD
-    text = re.sub(r"[\u2600-\u26FF\u2700-\u27BF]+", "", text)  # emojis
-    text = re.sub(r"\s+", " ", text)                        # whitespace
+    text = re.sub(r"\*\*|\*|__|_", "", text)
+    text = re.sub(r"[|~>#]", "", text)
+    text = re.sub(r"[\u2600-\u26FF\u2700-\u27BF]+", "", text)
+    text = re.sub(r"\s+", " ", text)
     return text.strip()
 
 def tts_player(text: str):
     clean_text = clean_text_for_tts(text)
+    voice = PiperVoice.load("models/en_US-amy-low.onnx")
+    samples = voice.synthesize(clean_text)
+    
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as fp:
-        voice = PiperVoice.load("en_US-amy-low.onnx")
-        voice.synthesize(clean_text, fp.name)
+        sf.write(fp.name, samples, voice.sample_rate)
         audio_bytes = open(fp.name, "rb").read()
+    
     b64 = base64.b64encode(audio_bytes).decode()
     st.components.v1.html(
         f"""
-        <audio id=\"ttsAudio\" controls style=\"width:100%;\">
-            <source src=\"data:audio/wav;base64,{b64}\" type=\"audio/wav\">
+        <audio id="ttsAudio" controls style="width:100%;">
+            <source src="data:audio/wav;base64,{b64}" type="audio/wav">
         </audio>
-        <div style=\"margin-top:4px;\">
-            <button onclick=\"document.getElementById('ttsAudio').playbackRate=1\">1×</button>
-            <button onclick=\"document.getElementById('ttsAudio').playbackRate=1.5\">1.5×</button>
-            <button onclick=\"document.getElementById('ttsAudio').playbackRate=2\">2×</button>
-        </div>
         """,
         height=90,
     )
     os.remove(fp.name)
 
-# ──────────────────────────────────────────
+# ─────────────────────────────────────────────────────────────
 st.title("🩺 Med11 – Upload Emergency Voice")
 
 if st.button("📞 Emergency Numbers"):
@@ -92,24 +80,17 @@ if st.button("📞 Emergency Numbers"):
 
 if st.session_state.show_numbers:
     with st.expander("Key Emergency Numbers (Pakistan)", expanded=True):
-        st.markdown(
-            """
+        st.markdown("""
             | Service | Number |
             |---------|--------|
             | 🚑 **Edhi Ambulance** | **115** |
             | 🚒 **Rescue / Fire** | **1122** |
             | 🚓 Police | **15** |
             | 💊 Poison Control (NIH) | **051-9255075** |
-            """
-        )
+        """)
 
-# Upload audio
 uploader_key = f"audio_{st.session_state.upload_id}"
-audio_file = st.file_uploader(
-    "Upload a WAV/MP3 describing the emergency",
-    type=["wav", "mp3", "m4a"],
-    key=uploader_key,
-)
+audio_file = st.file_uploader("Upload a WAV/MP3 describing the emergency", type=["wav", "mp3", "m4a"], key=uploader_key)
 
 if audio_file:
     with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp:
@@ -130,10 +111,7 @@ if audio_file:
     if transcript:
         with st.spinner("Analyzing…"):
             try:
-                result = groq_reply(
-                    f'Analyze this emergency: "{transcript}". '
-                    "Classify type, rate severity, and give brief first-aid steps."
-                )
+                result = groq_reply(f'Analyze this emergency: "{transcript}". Classify type, rate severity, and give brief first-aid steps.')
                 st.session_state.chat_history.append({"role": "user", "content": transcript})
                 st.session_state.chat_history.append({"role": "assistant", "content": result})
             except Exception as e:
@@ -149,17 +127,10 @@ if audio_file:
             st.markdown("---")
             st.markdown("**Need to tell me more? Upload another voice clip below.**")
 
-# ──────────────────────────────────────────
 col1, col2 = st.columns(2)
 with col1:
     if st.button("🏥 Nearest Hospital (Google Maps)"):
-        st.markdown(
-            "[Open Maps →](https://www.google.com/maps/search/hospital/)",
-            unsafe_allow_html=True,
-        )
+        st.markdown("[Open Maps →](https://www.google.com/maps/search/hospital/)", unsafe_allow_html=True)
 with col2:
-    if st.button("🗜️ Nearby Pharmacies"):
-        st.markdown(
-            "[Open Maps →](https://www.google.com/maps/search/pharmacy/)",
-            unsafe_allow_html=True,
-        )
+    if st.button("🗺️ Nearby Pharmacies"):
+        st.markdown("[Open Maps →](https://www.google.com/maps/search/pharmacy/)", unsafe_allow_html=True)