Update app.py

app.py

@@ -95,17 +95,36 @@ def extract_text_from_image(img_path):
     raw_text = pytesseract.image_to_string(image)
     return extract_tablet_names(raw_text)
 
-# ✅ Tablet Claim Checker
-def check_tablets(img):
-    tablets = extract_text_from_image(img)
-    if not tablets:
-        return "❌ No tablets found in receipt."
-    result = ""
-    for med in tablets:
-        question = f"Is the medicine {med} covered under the healthcare policy?"
-        answer = qa_chain.run(question)
-        result += f"🧾 **{med}** → {answer}\n\n"
-    return result
+def check_tablets(img_path):
+    try:
+        # Step 1: Confirm image path is received
+        if not img_path or not os.path.exists(img_path):
+            return "❌ Error: Image path is invalid or file not found."
+
+        # Step 2: Run OCR
+        image = Image.open(img_path)
+        raw_text = pytesseract.image_to_string(image)
+
+        # Step 3: Extract medicine names
+        tablets = extract_tablet_names(raw_text)
+        if not tablets:
+            return "❌ No tablets found in the receipt text."
+
+        # Step 4: Use RAG to check each tablet
+        result = ""
+        for med in tablets:
+            question = f"Is the medicine {med} covered under the healthcare policy?"
+            try:
+                answer = qa_chain.run(question)
+            except Exception as e:
+                answer = f"RAG error: {str(e)}"
+            result += f"💊 {med} → {answer}\n\n"
+
+        return result.strip()
+
+    except Exception as e:
+        return f"❌ Critical error during tablet check: {str(e)}"
+
 
 # ✅ Gradio UI
 with gr.Blocks(title="Healthcare Chatbot") as app: