from langchain.llms import OpenAI
from dotenv import load_dotenv

load_dotenv()

import streamlit as st
import os


#for Openai model load 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.6)
    response = llm(question)
    return response
    #streamlit app

st.set_page_config(page_title= "Demo App")
st.header('Yash app')

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


submit = st.button("Ask Question")


if submit:
    st.subheader("Response from GPT")
    st.write(response)