from langchain.llms import OpenAI
from dotenv import load_dotenv
import streamlit as st
import os

load_dotenv()

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

st.set_page_config(page_title="Q&A Demo", page_icon="🔗", layout="wide")
st.header("LangChain Application")

input = st.text_input("INPUT: ", key="input")
submit_button = st.button("Ask a question")

response = get_llm_response(input)
if submit_button:
    st.subheader("The response is")
    st.write(response)