Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,8 @@ import gradio as gr
|
|
4 |
from llama_index import GPTVectorStoreIndex, SimpleDirectoryReader, ServiceContext
|
5 |
from llama_index import OpenAIEmbedding
|
6 |
import os
|
|
|
|
|
7 |
logging.basicConfig(stream=sys.stdout, level=logging.INFO)
|
8 |
logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
|
9 |
os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
|
@@ -16,11 +18,16 @@ def build_index(documents):
|
|
16 |
return GPTVectorStoreIndex.from_documents(documents)
|
17 |
|
18 |
def query_index(index, similarity_top_k=3, streaming=True):
|
19 |
-
|
20 |
-
|
21 |
-
|
|
|
|
|
|
|
|
|
|
|
22 |
)
|
23 |
-
return
|
24 |
|
25 |
data = load_data()
|
26 |
index = build_index(data)
|
@@ -28,7 +35,7 @@ query_engine = query_index(index) # initialize the query engine
|
|
28 |
|
29 |
def get_response(text,history=None):
|
30 |
# Use the initialized query engine to perform the query
|
31 |
-
response = str(query_engine.
|
32 |
return response
|
33 |
|
34 |
t = gr.ChatInterface(get_response, analytics_enabled=True)
|
|
|
4 |
from llama_index import GPTVectorStoreIndex, SimpleDirectoryReader, ServiceContext
|
5 |
from llama_index import OpenAIEmbedding
|
6 |
import os
|
7 |
+
from llama_index.memory import ChatMemoryBuffer
|
8 |
+
|
9 |
logging.basicConfig(stream=sys.stdout, level=logging.INFO)
|
10 |
logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
|
11 |
os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
|
|
|
18 |
return GPTVectorStoreIndex.from_documents(documents)
|
19 |
|
20 |
def query_index(index, similarity_top_k=3, streaming=True):
|
21 |
+
memory = ChatMemoryBuffer.from_defaults(token_limit=1500)
|
22 |
+
chat_engine = index.as_chat_engine(
|
23 |
+
chat_mode="context",
|
24 |
+
memory=memory,
|
25 |
+
system_prompt=(
|
26 |
+
"You are a chatbot, able to have normal interactions, as well as talk"
|
27 |
+
" about Yoga and anything related to it."
|
28 |
+
),
|
29 |
)
|
30 |
+
return chat_engine # return the query engine instance # return the query engine instance
|
31 |
|
32 |
data = load_data()
|
33 |
index = build_index(data)
|
|
|
35 |
|
36 |
def get_response(text,history=None):
|
37 |
# Use the initialized query engine to perform the query
|
38 |
+
response = str(query_engine.chat(text))
|
39 |
return response
|
40 |
|
41 |
t = gr.ChatInterface(get_response, analytics_enabled=True)
|