import streamlit as st
from langchain_google_genai import ChatGoogleGenerativeAI

# Set the app to wide mode
st.set_page_config(layout="wide")

st.title("Gemini ChatBOT")

def llm_response(user_input):
  llm = ChatGoogleGenerativeAI(model="gemini-pro")
  result = llm.invoke(user_input)
  return st.write(result.content)

user_input = st.text_area("Enter your query here...")

if st.button("Get Response") and user_input:
    with st.spinner("Generating Response..."):
        llm_response(user_input)