normalize

handler.py

@@ -96,6 +96,13 @@ image_processor = None
 context_len = None
 args = None
 
+# Configuration for consistent responses
+PROMPT_NORMALIZATION = True  # Set to False to disable prompt normalization
+DEFAULT_ECG_PROMPT = "What are the main features and diagnosis in this ECG image? Provide a comprehensive clinical analysis"
+
+# Note: When PROMPT_NORMALIZATION is True, all ECG diagnosis requests will be
+# standardized to ensure consistent response lengths and content across different clients.
+
 def get_conv_log_filename():
     t = datetime.datetime.now()
     name = os.path.join(LOGDIR, f"{t.year}-{t.month:02d}-{t.day:02d}-user_conv.json")
@@ -557,6 +564,19 @@ def query(payload):
                        payload.get("prompt") or 
                        payload.get("istem") or "")
         
+        # Normalize prompt to ensure consistent responses
+        # This helps maintain consistency across different clients
+        if PROMPT_NORMALIZATION and "ecg" in message_text.lower() and "diagnosis" in message_text.lower():
+            # Standardize ECG analysis prompts for consistency
+            if "comprehensive" in message_text.lower():
+                message_text = DEFAULT_ECG_PROMPT
+            elif "concise" in message_text.lower():
+                message_text = "What are the main features and diagnosis in this ECG image? Provide a concise, clinical answer."
+            else:
+                # Default to comprehensive analysis for consistency
+                message_text = DEFAULT_ECG_PROMPT
+            print(f"[DEBUG] Normalized prompt to: {message_text}")
+        
         # Extract image with multiple possible keys
         image_input = (payload.get("image") or 
                       payload.get("image_url") or