Spaces:
Sleeping
Sleeping
File size: 725 Bytes
811152b 5262e51 811152b 5f88a75 811152b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
from langchain_openai import ChatOpenAI
from langchain_core.prompts import PromptTemplate
from langchain_core.output_parsers import StrOutputParser
import os
import streamlit as st
api_key = os.getenv("OPENAI_API_KEY")
print(api_key)
st.title("Query Response APP")
llm = ChatOpenAI(openai_api_key=api_key)
prompt = PromptTemplate(
template= 'Please write your query : {Query}',
input_variables=['Query']
)
parser = StrOutputParser()
chain = prompt | llm | parser
Query = st.text_input("Query: ", key="Query")
submit = st.button("Ask the question")
# Query = 'What is the capital of India'
if submit:
response = chain.invoke({'Query':Query})
st.subheader('the response is')
st.write(response) |