Spaces:
Sleeping
Sleeping
File size: 1,660 Bytes
061b38b |
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 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
# Q&A Chatbot
from langchain_openai import OpenAI
from dotenv import load_dotenv
load_dotenv() # take environment variables from .env.
import streamlit as st
import os
from langchain.prompts import PromptTemplate
from langchain.chains import LLMChain
from langchain.chains import SequentialChain
## Function to load OpenAI model and get respones
def get_openai_response(country_name):
llm=OpenAI(openai_api_key=os.environ["OPEN_API_KEY"],temperature=0.5)
capital_template=PromptTemplate(input_variables=['country'],
template="Please tell me the capital of the {country}")
capital_chain=LLMChain(llm=llm,prompt=capital_template,output_key="capital")
famous_template=PromptTemplate(input_variables=['capital'],
template="Suggest me some amazing places to visit in {capital}")
famous_chain=LLMChain(llm=llm,prompt=famous_template,output_key="places")
eats_template=PromptTemplate(input_variables=['capital'],
template="Suggest the top 5 famous dishes to eat in {capital}")
eats_chain=LLMChain(llm=llm,prompt=eats_template,output_key="dishes")
chain=SequentialChain(chains=[capital_chain,famous_chain,eats_chain],
input_variables=['country'],
output_variables=['capital',"places","dishes"])
response=chain.invoke({'country':country_name})
return response
##initialize our streamlit app
st.set_page_config(page_title="Q&A Chatbot")
st.header("January Capital Guide")
input=st.text_input("Enter Country Name: ",key="input")
response=get_openai_response(input)
submit=st.button("Answer")
## If ask button is clicked
if submit:
st.subheader("The Answer is")
st.write(response)
|