feat: AI patient onboarding backend and documentation

2026-08-27 12:15:09 +05:30
parent b9440772d1
commit 59c1372fe3
49 changed files with 4012 additions and 0 deletions
@@ -0,0 +1,84 @@
+import base64
+import httpx2
+from app.config import SARVAM_API_KEY
+
+
+class SarvamClient:
+
+    _client = None
+
+    def __init__(self):
+        self.api_key = SARVAM_API_KEY
+        self.base_url = "https://api.sarvam.ai"
+
+    def _get_client(self) -> httpx2.AsyncClient:
+        if SarvamClient._client is None:
+            SarvamClient._client = httpx2.AsyncClient(
+                timeout=12.0,
+                limits=httpx2.Limits(max_keepalive_connections=5, max_connections=10),
+            )
+        return SarvamClient._client
+
+    async def speech_to_text(
+        self,
+        audio_bytes: bytes,
+        filename: str = "audio.webm",
+        language_code: str = "ta-IN",
+    ) -> str:
+        if not self.api_key or self.api_key == "your_sarvam_key":
+            return ""
+
+        client = self._get_client()
+        files = {"file": (filename, audio_bytes, "audio/webm")}
+        data = {"model": "saaras:v2", "language_code": language_code}
+        headers = {"api-subscription-key": self.api_key}
+
+        try:
+            res = await client.post(
+                f"{self.base_url}/speech-to-text",
+                files=files,
+                data=data,
+                headers=headers,
+            )
+            if res.status_code == 200:
+                return res.json().get("transcript", "").strip()
+        except Exception as e:
+            print(f"[Sarvam STT Error]: {e}")
+        return ""
+
+    async def text_to_speech(
+        self,
+        text: str,
+        language_code: str = "ta-IN",
+    ) -> bytes | None:
+        if not self.api_key or self.api_key == "your_sarvam_key":
+            return None
+
+        lang_prefix = language_code.lower()[:2]
+        speaker = "meera" if lang_prefix == "ta" else "arvind"
+
+        client = self._get_client()
+        headers = {
+            "api-subscription-key": self.api_key,
+            "Content-Type": "application/json",
+        }
+        payload = {
+            "inputs": [text],
+            "target_language_code": language_code,
+            "speaker": speaker,
+            "model": "bulbul:v1",
+        }
+        try:
+            res = await client.post(
+                f"{self.base_url}/text-to-speech",
+                json=payload,
+                headers=headers,
+            )
+            if res.status_code == 200:
+                audios = res.json().get("audios", [])
+                if audios:
+                    return base64.b64decode(audios[0])
+        except Exception as e:
+            print(f"[Sarvam TTS Error]: {e}")
+        return None
+