langchain-QnA / app.py
kiran475h's picture
Upload 2 files
fca4974 verified
raw
history blame
774 Bytes
# Q&A Chatbot
from langchain.llms import OpenAI
from dotenv import load_dotenv
load_dotenv() # takes the enveronment vareable from .env
import streamlit as st
import os
## function to lad the openai model and to get the response
os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
def get_openai_response(question):
llm = OpenAI(model_name="gpt-3.5-turbo-0125",temperature =0.6)
response = llm(question)
return response
#initialize 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("generate")
if submit:
st.subheader("The response is ")
st.write(response)