#Q&A chatbot
from langchain.llms import OpenAI
from dotenv import load_dotenv

load_dotenv() # take environment variables from .env.

import streamlit as st
import os

##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="text-davinci-003",temperature=.6)
    response=llm(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.header("The Response is")
    st.write(response)