import keyfile
import streamlit as st
import google.generativeai as genai

genai.configure(api_key=keyfile.GEMINI_API_KEY)

if "history" not in st.session_state:
    st.session_state["history"] = []

def get_response(question):
    model = genai.GenerativeModel("gemini-1.5-flash")
    response = model.generate_content(question)
    return response.text

st.set_page_config(page_title="Chat Bot", page_icon=":robot:")
st.header("Chat Application")

user_input = st.text_input("Enter your question:")

if st.button("Ask!"):
    if user_input:
        response = get_response(user_input)

        st.session_state["history"].append({"user": user_input, "bot": response})

        st.session_state["user_input"] = ""

for chat in st.session_state["history"]:
    st.write(f"You: {chat['user']}")
    st.write(f"Answer: {chat['bot']}")