updated with rollback logic

services/geminiService.ts

@@ -1,7 +1,7 @@
 import { GoogleGenAI, Type } from "@google/genai";
 import { PatientProfile, ClinicalVitals, AppMode, RiskAnalysisResult, ChatMessage, ExtractionResult, HealthInsights } from "../types";
 
-// --- API KEY & CLIENT INITIALIZATION ---
+// --- API KEY & CONFIG ---
 const getApiKey = () => {
   try {
     // @ts-ignore
@@ -17,20 +17,34 @@ const getApiKey = () => {
   return '';
 };
 
+// --- BACKEND CONFIGURATION ---
+// Primary Backend (Text Logic - Phi-3)
+const PRIMARY_API_BASE = 'https://arshenoy-somai-backend.hf.space';
+
+// Secondary Backend (Media - Moondream/Whisper)
+// If you create a new space, put its URL here (e.g. via VITE_MEDIA_API_URL env var), 
+// otherwise it defaults to the primary one.
+const getMediaApiBase = () => {
+  try {
+    // @ts-ignore
+    if (typeof import.meta !== 'undefined' && import.meta.env && import.meta.env.VITE_MEDIA_API_URL) {
+      // @ts-ignore
+      return import.meta.env.VITE_MEDIA_API_URL;
+    }
+  } catch (e) {}
+  return PRIMARY_API_BASE;
+};
+
+const MEDIA_API_BASE = getMediaApiBase();
+
 const API_KEY = getApiKey();
 const ai = new GoogleGenAI({ apiKey: API_KEY });
 
 // --- TIERED MODEL STRATEGY ---
-// Tier 1: 1,000 RPD (Primary - Fast & High Quota)
 const MODEL_TIER_1 = 'gemini-2.5-flash-lite'; 
-// Tier 2: 250 RPD (Backup - Very Reliable)
 const MODEL_TIER_2 = 'gemini-2.5-flash'; 
-// TTS Model
 const MODEL_TTS = 'gemini-2.5-flash-preview-tts';
 
-const FALLBACK_API_BASE = 'https://arshenoy-somai-backend.hf.space';
-
-// Cleaning for final blocks
 const cleanText = (text: string) => {
   if (!text) return "";
   return text.replace(/\*\*/g, '').replace(/###/g, '').replace(/\*/g, '-').trim();
@@ -44,12 +58,10 @@ const compressImage = async (base64Str: string, maxWidth = 800): Promise<string>
       const canvas = document.createElement('canvas');
       let width = img.width;
       let height = img.height;
-      
       if (width > maxWidth) {
         height = (height * maxWidth) / width;
         width = maxWidth;
       }
-      
       canvas.width = width;
       canvas.height = height;
       const ctx = canvas.getContext('2d');
@@ -62,19 +74,26 @@ const compressImage = async (base64Str: string, maxWidth = 800): Promise<string>
 
 export const wakeUpBackend = async () => {
   try {
-    await fetch(`${FALLBACK_API_BASE}/`, { method: 'GET', mode: 'cors' });
+    // Ping both potential backends
+    fetch(`${PRIMARY_API_BASE}/`, { method: 'GET', mode: 'cors' }).catch(()=>{});
+    if (PRIMARY_API_BASE !== MEDIA_API_BASE) {
+      fetch(`${MEDIA_API_BASE}/`, { method: 'GET', mode: 'cors' }).catch(()=>{});
+    }
   } catch (e) {}
 };
 
-const callFallbackAPI = async (endpoint: string, payload: any): Promise<string> => {
-  console.info(`[SomAI System] Switching to Fallback API (Tier 3): ${FALLBACK_API_BASE}${endpoint}`);
+// Generic Fallback Caller
+const callBackend = async (baseUrl: string, endpoint: string, payload: any, onStatus?: (msg: string) => void): Promise<string> => {
+  const url = `${baseUrl}${endpoint}`;
+  console.info(`[SomAI] Calling Backend: ${url}`);
+  if (onStatus) onStatus("🐢 Switching to local backup...");
   
   const makeRequest = async (retries = 2) => {
     const controller = new AbortController();
-    const timeoutId = setTimeout(() => controller.abort(), 90000); 
+    const timeoutId = setTimeout(() => controller.abort(), 90000); // 90s timeout for CPU
     
     try {
-      const response = await fetch(`${FALLBACK_API_BASE}${endpoint}`, {
+      const response = await fetch(url, {
         method: 'POST',
         mode: 'cors', 
         credentials: 'omit',
@@ -85,6 +104,7 @@ const callFallbackAPI = async (endpoint: string, payload: any): Promise<string>
       clearTimeout(timeoutId);
 
       if (!response.ok && (response.status === 503 || response.status === 504) && retries > 0) {
+        if (onStatus) onStatus(`💤 Backend waking up... (${retries} retries left)`);
         await new Promise(r => setTimeout(r, 5000));
         return makeRequest(retries - 1);
       }
@@ -106,6 +126,7 @@ const callFallbackAPI = async (endpoint: string, payload: any): Promise<string>
     } catch (error: any) {
       clearTimeout(timeoutId);
       if (retries > 0 && (error.name === 'AbortError' || error.message.includes('Failed'))) {
+         if (onStatus) onStatus("📡 Connection unstable, retrying...");
          await new Promise(r => setTimeout(r, 5000));
          return makeRequest(retries - 1);
       }
@@ -152,16 +173,13 @@ const parseRiskResponse = (text: string, calculatedScore: number): RiskAnalysisR
   }
 };
 
-// --- UPDATED: VISION EXTRACTION (TIER 1 -> TIER 2 -> FALLBACK) ---
-export const extractClinicalData = async (imageBase64: string): Promise<ExtractionResult> => {
+// --- VISION EXTRACTION ---
+export const extractClinicalData = async (imageBase64: string, onStatus?: (msg: string) => void): Promise<ExtractionResult> => {
   const base64Data = imageBase64.includes('base64,') ? imageBase64.split('base64,')[1] : imageBase64;
-  const prompt = `Analyze this medical document. 
-  CRITICAL: Look for the Patient's Name at the top, headers, or labeled 'Patient', 'Name', 'Mr/Mrs'. 
-  Extract JSON: { name, age, condition, history, allergies, systolicBp, glucose, heartRate, weight, temperature, spo2, clinicalNote }. 
-  If name is missing, use "Guest". Return JSON only.`;
+  const prompt = `Analyze this medical document. CRITICAL: Look for Patient Name. Extract JSON: { name, age, condition, history, allergies, systolicBp, glucose, heartRate, weight, temperature, spo2, clinicalNote }. Return JSON only.`;
   
   const callGeminiVision = async (modelName: string) => {
-    console.log(`[Vision] Trying ${modelName}...`);
+    if (onStatus) onStatus(`⚡ Scanning with ${modelName}...`);
     const response = await ai.models.generateContent({
       model: modelName,
       contents: [{ role: 'user', parts: [{ text: prompt }, { inlineData: { mimeType: 'image/jpeg', data: base64Data } }] }],
@@ -180,18 +198,21 @@ export const extractClinicalData = async (imageBase64: string): Promise<Extracti
     if (!API_KEY) throw new Error("API Key missing");
     return await callGeminiVision(MODEL_TIER_1);
   } catch (e: any) { 
-    console.warn(`[Vision] Tier 1 Failed: ${e.message}. Trying Tier 2...`);
-    try {
-        return await callGeminiVision(MODEL_TIER_2);
-    } catch (e2) {
-        console.warn(`[Vision] Tier 2 Failed. Trying Tier 3 (Backend)...`);
+    if (e.toString().includes('429') || e.toString().includes('Quota')) {
+        try {
+            return await callGeminiVision(MODEL_TIER_2);
+        } catch (e2) {}
     }
 
-    // 3. Fallback (Moondream)
+    // Fallback: Moondream on Media Backend
     try {
+        if (onStatus) onStatus("🐢 Compressing for Moondream...");
         const compressedBase64 = await compressImage(imageBase64);
         const cleanBase64 = compressedBase64.includes('base64,') ? compressedBase64.split('base64,')[1] : compressedBase64;
-        const resText = await callFallbackAPI('/vision', { image: cleanBase64, prompt: "Extract patient name and vitals from this document in JSON format." });
+        
+        if (onStatus) onStatus("🐢 Using Local Vision Node...");
+        const resText = await callBackend(MEDIA_API_BASE, '/vision', { image: cleanBase64, prompt: "Extract patient name and vitals from this document." }, onStatus);
+        
         return {
             profile: {}, 
             vitals: { clinicalNote: `[Auto-Scanned]: ${resText}` },
@@ -203,7 +224,6 @@ export const extractClinicalData = async (imageBase64: string): Promise<Extracti
   }
 };
 
-// NEW: Gemini TTS Service
 export const generateSpeech = async (text: string): Promise<string | null> => {
   if (!API_KEY) return null;
   try {
@@ -217,7 +237,6 @@ export const generateSpeech = async (text: string): Promise<string | null> => {
     });
     return response.candidates?.[0]?.content?.parts?.[0]?.inlineData?.data || null;
   } catch (e) {
-    console.warn("TTS Failed", e);
     return null;
   }
 };
@@ -228,7 +247,8 @@ export const transcribeAudio = async (audioBlob: Blob): Promise<string> => {
         reader.onloadend = async () => {
             const base64 = (reader.result as string).split(',')[1];
             try {
-                const text = await callFallbackAPI('/transcribe', { audio: base64 });
+                // Whisper calls go to Media Backend
+                const text = await callBackend(MEDIA_API_BASE, '/transcribe', { audio: base64 });
                 resolve(text);
             } catch (e) { reject("Voice transcription failed."); }
         };
@@ -236,21 +256,25 @@ export const transcribeAudio = async (audioBlob: Blob): Promise<string> => {
     });
 };
 
-// --- UPDATED: RISK ANALYSIS (TIER 1 -> TIER 2 -> FALLBACK) ---
-export const analyzeRisk = async (profile: PatientProfile, vitals: ClinicalVitals, calculatedScore: number): Promise<RiskAnalysisResult> => {
+// --- RISK ANALYSIS ---
+export const analyzeRisk = async (
+  profile: PatientProfile, 
+  vitals: ClinicalVitals, 
+  calculatedScore: number,
+  onStatus?: (msg: string) => void
+): Promise<RiskAnalysisResult> => {
   const prompt = `
     Act as a Senior Clinical Risk Assessor.
     Patient: ${profile.name} (${profile.age}, ${profile.gender}). Condition: ${profile.condition}.
-    History: ${profile.history}. Surgeries: ${profile.surgeries}. Family History: ${profile.familyHistory}.
-    Lifestyle: Diet-${profile.diet}, Exercise-${profile.exerciseFrequency}, Smoke-${profile.smokingStatus}, Alcohol-${profile.alcoholConsumption}.
-    Vitals: BP Morning ${vitals.systolicBpMorning} / Evening ${vitals.systolicBpEvening}. Glucose ${vitals.glucose}. HR ${vitals.heartRate}. SpO2 ${vitals.spo2}%. Temp ${vitals.temperature}F. Weight ${vitals.weight}kg.
+    History: ${profile.history}. 
+    Vitals: BP ${vitals.systolicBp}, Glucose ${vitals.glucose}, SpO2 ${vitals.spo2}%.
     Note: ${vitals.clinicalNote}.
-    Task: 1. Summary (Risk level). 2. 3 Action Items. 3. ICD-10 Pipeline (Condition, History, Symptoms). 4. Insurance Note.
+    Task: 1. Summary. 2. 3 Action Items. 3. ICD-10 Pipeline (Condition, History). 4. Insurance Note.
     Return JSON.
   `;
 
   const callGeminiRisk = async (modelName: string) => {
-    console.log(`[Risk] Trying ${modelName}...`);
+    if (onStatus) onStatus(`⚡ Analyzing with ${modelName}...`);
     const response = await ai.models.generateContent({
         model: modelName,
         contents: prompt,
@@ -277,16 +301,14 @@ export const analyzeRisk = async (profile: PatientProfile, vitals: ClinicalVital
     if (!API_KEY) throw new Error("API Key missing");
     return await callGeminiRisk(MODEL_TIER_1);
   } catch (err: any) {
-    console.warn(`[Risk] Tier 1 Failed: ${err.message}. Trying Tier 2...`);
-    try {
-        return await callGeminiRisk(MODEL_TIER_2);
-    } catch (e2) {
-        console.warn(`[Risk] Tier 2 Failed. Trying Tier 3 (Backend)...`);
+    if (err.toString().includes('429') || err.toString().includes('Quota')) {
+        try { return await callGeminiRisk(MODEL_TIER_2); } catch (e2) {}
     }
 
     try {
       const payload = { ...profile, ...vitals, riskScore: calculatedScore, prompt };
-      const fallback = await callFallbackAPI('/analyze', payload);
+      // Fallback goes to Primary Backend (Text Node)
+      const fallback = await callBackend(PRIMARY_API_BASE, '/analyze', payload, onStatus);
       return { 
         ...parseRiskResponse(fallback, calculatedScore),
         source: 'Phi-3 Mini (Fallback)' 
@@ -298,7 +320,7 @@ export const analyzeRisk = async (profile: PatientProfile, vitals: ClinicalVital
 };
 
 export const generateHealthInsights = async (profile: PatientProfile, vitals: ClinicalVitals): Promise<HealthInsights> => {
-  const prompt = `Based on Patient: ${profile.name}, ${profile.age}y, ${profile.condition}. Vitals: BP ${vitals.systolicBp}, SpO2 ${vitals.spo2}%. Generate JSON: { weeklySummary, progress, tips: [] }.`;
+  const prompt = `Based on Patient: ${profile.name}, ${profile.age}y, ${profile.condition}. Vitals: BP ${vitals.systolicBp}. Generate JSON: { weeklySummary, progress, tips: [] }.`;
   
   const callGeminiInsights = async (model: string) => {
     const response = await ai.models.generateContent({
@@ -313,7 +335,9 @@ export const generateHealthInsights = async (profile: PatientProfile, vitals: Cl
     if (!API_KEY) throw new Error("No Key");
     return await callGeminiInsights(MODEL_TIER_1);
   } catch (err: any) {
-    try { return await callGeminiInsights(MODEL_TIER_2); } catch (e) {}
+    if (err.toString().includes('429')) {
+        try { return await callGeminiInsights(MODEL_TIER_2); } catch (e) {}
+    }
     return { weeklySummary: "Keep tracking your vitals.", progress: "Data accumulated.", tips: ["Maintain a balanced diet.", "Stay hydrated."] };
   }
 };
@@ -325,38 +349,32 @@ export const generateSessionName = async (userText: string, aiText: string): Pro
     const response = await ai.models.generateContent({ model: MODEL_TIER_1, contents: prompt, config: { maxOutputTokens: 20 } });
     return cleanText(response.text || "New Consultation").replace(/^["']|["']$/g, '');
   } catch (e) {
-    try {
-        const fallbackRes = await callFallbackAPI('/generate', { prompt: prompt });
-        return cleanText(fallbackRes).replace(/^["']|["']$/g, '');
-    } catch { return "New Consultation"; }
+    return "New Consultation";
   }
 };
 
-// --- UPDATED: CHAT (TIER 1 -> TIER 2 -> FALLBACK) ---
+// --- CHAT ---
 export const generateChatResponse = async (
   history: ChatMessage[], 
   currentMessage: string, 
   image: string | undefined, 
   profile: PatientProfile, 
   mode: AppMode,
-  onSource: (source: string) => void
+  onSource: (source: string) => void,
+  onStatus?: (msg: string) => void
 ): Promise<string> => {
   const context = `
     Patient: ${profile.name} (${profile.age}y).
     Condition: ${profile.condition}. History: ${profile.history}.
-    Surgeries: ${profile.surgeries}. Family Hx: ${profile.familyHistory}.
-    Lifestyle: ${profile.diet}, ${profile.exerciseFrequency}, Smoke: ${profile.smokingStatus}.
-    Emergency Contact: ${profile.emergencyContactName} (${profile.emergencyContactPhone}).
-    Tone: ${mode === AppMode.THERAPY ? 'Empathetic, calm, therapeutic (CBT).' : 'Professional, educational, clear.'}
+    Tone: ${mode === AppMode.THERAPY ? 'Empathetic CBT' : 'Medical Guide'}.
     Format: Plain text. No markdown.
   `;
   
   const contents = history.map(msg => ({ role: msg.role === 'user' ? 'user' : 'model', parts: [{ text: msg.text }, ...(msg.image ? [{ inlineData: { mimeType: 'image/jpeg', data: msg.image.split('base64,')[1] } }] : [])] }));
   contents.push({ role: 'user', parts: [{ text: context + "\nUser: " + currentMessage }, ...(image ? [{ inlineData: { mimeType: 'image/jpeg', data: image.split('base64,')[1] } }] : [])] });
 
-  // Helper for Chat
   const callGeminiChat = async (modelName: string) => {
-    console.log(`[Chat] Trying ${modelName}...`);
+    if (onStatus) onStatus(`Generating with ${modelName}...`);
     onSource(modelName === MODEL_TIER_1 ? 'Gemini 2.5 Flash-Lite' : 'Gemini 2.5 Flash'); 
     const response = await ai.models.generateContent({
         model: modelName,
@@ -368,22 +386,20 @@ export const generateChatResponse = async (
 
   try {
     if (!API_KEY) throw new Error("No Key");
-    // 1. Try Tier 1
     return await callGeminiChat(MODEL_TIER_1);
   } catch (e: any) {
-    console.warn(`[Chat] Tier 1 Failed: ${e.message}. Switching to Tier 2...`);
-    // 2. Try Tier 2 on ANY failure
-    try {
-        return await callGeminiChat(MODEL_TIER_2);
-    } catch (e2) {
-        console.warn(`[Chat] Tier 2 Failed. Switching to Tier 3 (Backend)...`);
+    if (e.toString().includes('429') || e.toString().includes('Quota')) {
+        try {
+            return await callGeminiChat(MODEL_TIER_2);
+        } catch (e2) {}
     }
 
-    // 3. Fallback
     try {
+      if (onStatus) onStatus("Falling back to Local Phi-3...");
       onSource('Phi-3 Mini (Fallback)'); 
       const fallbackPrompt = `${context}\n\nChat History:\n${history.slice(-3).map(m => m.text).join('\n')}\nUser: ${currentMessage}`;
-      const responseText = await callFallbackAPI('/generate', { prompt: fallbackPrompt });
+      // Fallback goes to Primary Backend (Text Node)
+      const responseText = await callBackend(PRIMARY_API_BASE, '/generate', { prompt: fallbackPrompt }, onStatus);
       return cleanText(responseText);
     } catch { 
         return "I'm having trouble connecting. Please check your internet.";
@@ -394,7 +410,7 @@ export const generateChatResponse = async (
 export const generateQuickReplies = async (history: ChatMessage[]) => {
   if (!API_KEY || history.length === 0) return [];
   const recentContext = history.slice(-3).map(m => `${m.role}: ${m.text}`).join('\n');
-  const prompt = `Based on this conversation:\n${recentContext}\n\nSuggest 3 short, relevant follow-up questions the USER might want to ask next. Return ONLY a JSON array of strings.`;
+  const prompt = `Based on: ${recentContext}. Suggest 3 short follow-up questions. JSON array.`;
   try {
     const res = await ai.models.generateContent({ model: MODEL_TIER_1, contents: prompt, config: { responseMimeType: "application/json" } });
     return JSON.parse(res.text || "[]");
@@ -404,7 +420,7 @@ export const generateQuickReplies = async (history: ChatMessage[]) => {
 export const summarizeConversation = async (history: ChatMessage[]) => {
   if (!API_KEY) return "Summary unavailable.";
   try {
-    const res = await ai.models.generateContent({ model: MODEL_TIER_1, contents: `Summarize clinical conversation:\n${history.map(m=>m.text).join('\n')}` });
+    const res = await ai.models.generateContent({ model: MODEL_TIER_1, contents: `Summarize:\n${history.map(m=>m.text).join('\n')}` });
     return cleanText(res.text || "");
   } catch { return "Could not summarize."; }
 };