utils/parser.py

# parser.py
import json
import google.generativeai as genai
import os
from dotenv import load_dotenv

load_dotenv()
genai.configure(api_key=os.getenv("GEMINI_API_KEY"))

def parse_query_with_gemini(query: str):
    model = genai.GenerativeModel("models/gemini-1.5-flash-latest")
    prompt = f"""

You are an intelligent insurance assistant.

Given a natural language query, extract the following fields as JSON. Do not include any explanation or extra text — just valid JSON:



- age (integer)

- gender (male/female/unknown)

- procedure (string)

- location (string)

- policy_duration_months (integer)



Query:

"{query}"

"""
    try:
        response = model.generate_content(prompt)
        response_text = response.text.strip()
        if response_text.startswith("```"):
            response_text = response_text.strip("`").replace("json", "").strip()
        return json.loads(response_text)
    except Exception as e:
        return {
            "error": "Failed to parse Gemini response",
            "raw_response": response.text if 'response' in locals() else str(e)
        }