# create a simple chatbot using streamlit

from langchain.llms import OpenAI
#from langchain_openai import OpenAI
from dotenv import load_dotenv

load_dotenv()

import os
import streamlit as st

## create a function to load OpenAI model and get response

def get_openai_response(question):

    llm = OpenAI(openai_api_key = os.getenv('OPEN_API_KEY'),model_name = 'gpt-3.5-turbo-instruct',
    temperature = 0.5)
    response = llm.invoke(question)
    return response

## initialize our streamlit app

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

input = st.text_input('Input : ',key="input")
response = get_openai_response(input)

submit = st.button('Ask the question')

## if ask button is clicked
if submit:
    st.subheader('the response is')
    st.write(response)