# Q & A chatbot
from langchain.llms import OpenAI 
from dotenv import load_dotenv
load_dotenv()
import streamlit as st
import os

# Function to load openai model and get response
def get_open_ai_response(question):
    llm = OpenAI(model_name = "gpt-3.5-turbo-instruct", temperature=0.5)
    response = llm(question)
    return response

# initialise streamlit app
st.set_page_config(page_title="Q&A Demo")
st.header("Langchain Application")

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

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

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