from langchain_community.llms import OpenAI
#from langchain.llms import OpenAI
from langchain_community.chat_models import ChatOpenAI
from dotenv import load_dotenv

load_dotenv()

import streamlit as st
import os

#Function to load openai model and get response

def get_openai_response(question):
    llm=OpenAI(model_name='gpt-3.5-turbo',temperature=0.5)
    response=llm(question)
    return response

# Inilize out 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.subheader("The response is")
    st.write(response)