import streamlit as st
import os 
import sys
from langchain_google_genai import ChatGoogleGenerativeAI
google_api_key = st.sidebar.text_input("Provide Your Google Gemini API Key")

if not google_api_key:
    st.write("Please Provide Your API then Try Again")
    sys.exit(0)
    
def llm_response(question):
    llm = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.5, google_api_key=google_api_key)
    response = llm.invoke(question)
    return response.content


st.sidebar.title("LangChain QA Chatbot")
query = st.sidebar.text_area("Ask Anything....", height=250)
generate_button = st.sidebar.button("Generate Answer", use_container_width=True)
response = llm_response(query)

if generate_button:
    if len(query) == 0:
        st.header("Please Ask Your Question.")
    else:
        st.header("Generated Content")
        st.write(response)