Spaces:
Sleeping
Sleeping
import os | |
import gradio as gr | |
import pinecone | |
from gpt_index import GPTIndexMemory, GPTPineconeIndex | |
from langchain.agents import Tool | |
from langchain.chains.conversation.memory import ConversationBufferMemory | |
from langchain import OpenAI | |
from langchain.agents import initialize_agent | |
OPENAI_API_KEY=os.environ.get("OPENAI_API_KEY") | |
PINECONE_API_KEY=os.environ.get("PINECONE_API_KEY") | |
PINECONE_ENV=os.environ.get("PINECONE_ENV") | |
pindex=pinecone.Index("sethgodin") | |
pinedex=GPTPineconeIndex([], pinecone_index=pindex) | |
tools = [ | |
Tool( | |
name = "GPT Index", | |
func=lambda q: str(pinedex.query(q)), | |
description="useful for when you want to answer questions about the author. The input to this tool should be a complete english sentence.", | |
return_direct=True | |
), | |
] | |
memory = ConversationBufferMemory(memory_key="chat_history") | |
llm=OpenAI(temperature=0) | |
agent_chain = initialize_agent(tools, llm, agent="conversational-react-description", memory=memory) | |
def run_something(isfa: str): | |
return agent_chain.run(input=isfa) | |
with gr.Blocks() as demo: | |
chatbot = gr.Chatbot() | |
with gr.Row(): | |
txt = gr.Textbox(show_label=False, placeholder="Enter text and press enter").style(container=False) | |
txt.submit(run_something) | |
demo.launch() |