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 and get response

def get_openai_response(question):
    llm=OpenAI(openai_api_key=os.environ["OPENAI_API_KEY"],
               model_name="gpt-3.5-turbo-instruct",
               temperature=0.6)
    response = llm(question)
    return response

# Initialize the Streamlit APP #
    
st.set_page_config(page_title="Q&A Demo")
st.header("Langchain Application")

input = st.text_input(label="Input: ", key="input")

response = get_openai_response(input)

submit = st.button(label="Ask the question")

# If the button is clicked 

if submit:
    st.subheader("The response is")
    st.write(response)