langchain-QA / app.py
atifali-pm's picture
Upload app.py
7021825 verified
raw
history blame contribute delete
708 Bytes
from langchain.llms import OpenAI
from dotenv import load_dotenv
import os
load_dotenv()
import streamlit as st
## function to load openai and get responses
def get_openai_response(quesetion):
llm=OpenAI(openai_api_key=os.getenv("OPEN_API_KEY"), model_name="text-davinci-003", temperature=0.5)
response = llm(quesetion)
return response
## initialize streamlit and set input field and button
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 the question")
## if button is submitted
if submit:
st.subheader("The response is")
st.write(response)