import gradio as gr
import pickle
import os
from google import genai

# Load Vector DB
with open('benefit_spa_vector_db.pkl', 'rb') as f:
    vector_db = pickle.load(f)

# Set API key from Secrets
genai.configure(api_key=os.environ['GOOGLE_API_KEY'])

def respond(message, history):
    return generate_response(vector_db, message)

gr.ChatInterface(respond).launch()