Spaces:
Runtime error
Runtime error
from langchain.llms import OpenAI | |
import streamlit as st | |
import os | |
from openaiapikey import openai_key | |
from langchain import PromptTemplate | |
from langchain.chains import LLMChain | |
from langchain.chains import SequentialChain | |
os.environ['OPENAI_API_KEY'] = openai_key | |
st.title('Book Summary') | |
input_text = st.text_input("Search the book you want") | |
#Prompt Template | |
first_input_prompt = PromptTemplate(input_variables = ['name'], | |
template="Provide me a summary of the book {name}" | |
) | |
#Open AI LLMS | |
llm = OpenAI(temperature=0.8) | |
#LLM Chain | |
chain1 = LLMChain(llm=llm, prompt = first_input_prompt, verbose=True, output_key = 'summaryofbook') | |
#Prompt Template | |
second_input_prompt = PromptTemplate(input_variables = ['summaryofbook'], | |
template="when was the {summaryofbook} published" | |
) | |
#LLM Chain | |
chain2 = LLMChain(llm=llm, prompt = second_input_prompt, verbose=True, output_key = 'bookpublishdate') | |
#Prompt Template | |
third_input_prompt = PromptTemplate(input_variables = ['summaryofbook'], | |
template="Please tell me about the authors of the {summaryofbook}" | |
) | |
#LLM Chain | |
chain3 = LLMChain(llm=llm, prompt = third_input_prompt, verbose=True, output_key = 'authorsofthebook') | |
parent_chain = SequentialChain(chains = [chain1, chain2, chain3], input_variables = ['name'], output_variables = ['summaryofbook', 'bookpublishdate','authorsofthebook'], verbose = True) | |
if input_text: | |
st.write(parent_chain({'name':input_text})) |