from langchain_openai import OpenAI
from dotenv import load_dotenv

load_dotenv()

import streamlit as st
import os

def get_openai_response(question):
  llm = OpenAI(openai_api_key=os.getenv("OPENAI_API_KEY"),
               model_name="gpt-3.5-turbo-instruct", temperature=0.5)
  response = llm.invoke(question)
  return response

### Streamlit app
st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
st.header("LangChain Demo")

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

sumbit=st.button("Ask Question")
if sumbit:
  st.subheader("The Response is")
  st.write(response)