from dotenv import load_dotenv
import os
import streamlit as st
import google.generativeai as genai


load_dotenv()

genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))


model = genai.GenerativeModel("gemini-pro")
def get_gemini_response(question):
    response = model.generate_content(question)
    return response.text

st.set_page_config(page_title="Q&A Chatbot")

st.header("Ask your questions to Gemini")

input = st.text_input("Question: ", key="input")
submit = st.button("Ask to Gemini")


if submit:
    response = get_gemini_response(input)
    st.subheader("Answer is")
    st.write(response)