import streamlit as st
from dotenv import load_dotenv, find_dotenv
import os
from langchain.llms import OpenAI

# load env variables
load_dotenv()


def get_response_from_OPENAI(prompt):
    llm_OPENAI = OpenAI(openai_api_key=os.environ.get('OPENAI_KEY'),
                        temperature=0.6)
    
    response = llm_OPENAI.predict(prompt)

    return response


#streamlit configuration

st.set_page_config(page_title="Q & A Test using OpenAI API",
                   layout='centered',
                   initial_sidebar_state='collapsed')

#header page
st.header("Q & A Test using OpenAI API")


input_text = st.text_input("Enter your question", placeholder="What is the ...")

btn_submit = st.button("Generate")

if btn_submit:
    response = get_response_from_OPENAI(input_text)
    st.write(response)