from langchain.llms import OpenAI
# from dotenv import load_dotenv
# load_dotenv() # takes environment variables
import streamlit as st
import os

def get_openai_response(question):
    llm=OpenAI(openai_api_key=os.getenv('OPENAI_API_KEY'),temperature=0.6)
    response = llm(question)
    return response

st.set_page_config(page_title='Simple Q&A bot')
st.header('Langchain Application')

input = st.text_input('Input: ', key='input')
response = get_openai_response(input)
submit= st.button('Ask')

if submit:
    st.subheader('Response:')
    st.write(response)