import streamlit as st
from langchain_google_genai import ChatGoogleGenerativeAI

def load_answer(question):
    llm=ChatGoogleGenerativeAI(model="gemini-pro", temperature=0)

    answer=llm.invoke(question)
    return answer.content

st.set_page_config(page_title="Skimble", page_icon=":robot:")
st.header("Skimble: Ask Me Anything")

def get_text():
    input_text=st.text_input("You:", key="input")
    return input_text
user_input=get_text()
response=load_answer(user_input)
submit=st.button('Generate')

if submit:
    st.subheader("Answer:")
    st.write(response)