Upd logger

vlm.py

@@ -1,5 +1,5 @@
 # vlm.py
-import os, logging
+import os, logging, traceback, json
 from huggingface_hub import InferenceClient
 from translation import translate_query
 
@@ -30,9 +30,22 @@ def process_medical_image(base64_image: str, prompt: str = None, lang: str = "EN
                 ]
             }]
         )
-        result = response.choices[0].message.content.strip()
+        # Validate response
+        if not response or not hasattr(response, "choices") or not response.choices:
+            raise ValueError("Empty or malformed response from MedGEMMA.")
+        # Get choice resp
+        message = response.choices[0].message
+        if not message or not hasattr(message, "content"):
+            raise ValueError("MedGEMMA response missing `.content`.")
+        # Beautify
+        result = message.content.strip()
         logger.info(f"[VLM] MedGemma returned {result}")
         return result
     except Exception as e:
-        logger.error(f"[VLM] ⚠️ Error from image diagnosis model: {e}")
-        return f"[VLM] ⚠️ Error from image diagnosis model: {e}"
+        logger.error(f"[VLM] ❌ Exception: {e}")
+        logger.error(f"[VLM] 🔍 Traceback:\n{traceback.format_exc()}")
+        try:
+            logger.error(f"[VLM] ⚠️ Raw response: {json.dumps(response, default=str, indent=2)}")
+        except:
+            logger.warning("[VLM] ⚠️ Response not serializable.")
+        return f"[VLM] ⚠️ Image diagnosis failed: {str(e)}"