#Q&A Chatbot
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_response(question):
    llm=OpenAI(openai_api_key=os.getenv("OPENAI_API_KEY"),model_name="gpt-4",temperature=0.6)
    response=llm(question)
    return response

## Initialize 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 Question")

## If ask button is clicked

if submit:
        st.subheader("the response is")
        st.write(response)