Update pipeline.py

pipeline.py

@@ -25,7 +25,6 @@ def extract_text_from_pdf(pdf_path):
     for page in pages:
         text += pytesseract.image_to_string(page) + "\n"
     return text.strip()
-
 def extract_key_values_with_gemini(raw_text, form_type="pancard_form"):
     prompt = f"""
 You are an intelligent document parser.
@@ -34,13 +33,15 @@ Return strictly as JSON key-value pairs.
 Document text:
 {raw_text}
 """
-    
-    # Use the supported model
     model = genai.GenerativeModel("models/gemini-2.5-flash")
     print("Gemini API called successfully ✅")
     response = model.generate_content(prompt)
+    text = response.text.strip()
+
+    # --- Cleanup ---
+    text = text.replace("```json", "").replace("```", "").strip()
 
     try:
-        return json.loads(response.text)
+        return json.loads(text)
     except Exception:
-        return {"raw_output": response.text}
+        return {"raw_output": text}