Update components/palm_summarizer.py

components/palm_summarizer.py

@@ -1,24 +1,48 @@
 import os
 import google.generativeai as genai
 
-# Load API key
+# Load API key - Use consistent environment variable name
 genai.configure(api_key=os.getenv("PALM_API_KEY"))
 
-# Gemini model from MakerSuite
-model = genai.GenerativeModel("gemini-pro")
+# Updated model name and configuration
+model = genai.GenerativeModel("gemini-1.5-flash")  # or "gemini-1.5-pro"
 
 def summarize_with_palm(text):
     try:
+        # Enhanced prompt for medical reports
         prompt = f"""
-You are a medical assistant. Please read the following health/lab report text and:
-1. Summarize it in 3 simple lines
-2. Explain abnormal values in layman terms
-3. Provide bullet-point health concerns
+You are a medical assistant specializing in health report analysis. Please analyze the following medical/lab report and provide:
+
+1. **Summary**: A concise 2-3 line summary of the overall health status
+2. **Key Findings**: List the most important test results and their significance
+3. **Abnormal Values**: Highlight any values outside normal ranges with simple explanations
+4. **Health Recommendations**: Basic health advice based on the results (always recommend consulting a doctor for medical decisions)
+
+Important: Use simple, patient-friendly language. Avoid complex medical jargon.
 
 Report Text:
 {text}
+
+Please format your response clearly with the above sections.
 """
+        
+        # Generate content with the updated API
         response = model.generate_content(prompt)
-        return response.text.strip() if response and response.text else "❌ No summary returned."
+        
+        # Check if response is valid
+        if response and response.text:
+            return response.text.strip()
+        else:
+            return "❌ No summary could be generated from the report."
+            
     except Exception as e:
-        return f"❌ PaLM error: {e}"
+        # Better error handling with specific error types
+        error_msg = str(e)
+        if "API_KEY" in error_msg.upper():
+            return "❌ API Key error: Please check your PALM_API_KEY environment variable."
+        elif "QUOTA" in error_msg.upper():
+            return "❌ API Quota exceeded: Please check your Google AI Studio quota."
+        elif "404" in error_msg:
+            return "❌ Model not found: Please verify the model name and API access."
+        else:
+            return f"❌ Summarization error: {error_msg}"