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

load_dotenv()

import streamlit as st

import os


def get_openai_response(question):
    # llm=OpenAI(openai_api_key=os.getenv("OPEN_API_KEY"),mode_name="gpt-3.5-turbo",temperature=0.5)
    llm=ChatOpenAI(openai_api_key=os.getenv("OPEN_API_KEY"),model_name="gpt-3.5-turbo", temperature=0.5)
    # response=llm(question)

    return llm.predict(question)



 
st.set_page_config(page_title="Q&A Demo")


st.header("Chatbot using Openai key and langchain")

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

submit=st.button("Ask a question")
response=get_openai_response(input)


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