# Chat-Bot
from langchain.llms import OpenAI
from dotenv import load_dotenv
import streamlit as st
import os


load_dotenv()


def GetOpenaiResponse(question,openaikey):
    llm = OpenAI(
        openai_api_key=os.getenv("OPENAI_API_KEY"),
        temperature=0.6,
    )
    response = llm.invoke(question)
    return response


st.set_page_config(page_title="Q&A Demo")
st.header("Open AI GPT Chat:")

input = st.text_input("Input : ", key="input")
apikey = st.text_input("API Key : ", key="input")
response = GetOpenaiResponse(input,apikey)


submit = st.button("Generate Answer")

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