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_reponse(question):
  llm = OpenAI(openai_api_key = os.getenv("OPEN_API_KEY"), model_name="text-davinci-003", temperature=0.5)
  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_reponse(input)

submit = st.button("Ask a question")

# If ask button is clicked
if submit:
  st.subheader("The Response is")
  st.write(response)