Update app.py
Browse files
app.py
CHANGED
@@ -1,14 +1,19 @@
|
|
1 |
import gradio as gr
|
2 |
import os
|
3 |
-
from llama_index.core import SimpleDirectoryReader, VectorStoreIndex,
|
4 |
from llama_index.llms.openai import OpenAI
|
5 |
|
6 |
-
# Set
|
7 |
os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
|
8 |
|
9 |
-
#
|
|
|
|
|
|
|
10 |
documents = SimpleDirectoryReader("documents").load_data()
|
11 |
-
index = VectorStoreIndex.from_documents(documents
|
|
|
|
|
12 |
query_engine = index.as_query_engine()
|
13 |
|
14 |
def chatbot_response(message):
|
|
|
1 |
import gradio as gr
|
2 |
import os
|
3 |
+
from llama_index.core import SimpleDirectoryReader, VectorStoreIndex, Settings
|
4 |
from llama_index.llms.openai import OpenAI
|
5 |
|
6 |
+
# Set OpenAI API key from environment
|
7 |
os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
|
8 |
|
9 |
+
# Configure the default LLM
|
10 |
+
Settings.llm = OpenAI()
|
11 |
+
|
12 |
+
# Load documents and build index
|
13 |
documents = SimpleDirectoryReader("documents").load_data()
|
14 |
+
index = VectorStoreIndex.from_documents(documents)
|
15 |
+
|
16 |
+
# Create query engine
|
17 |
query_engine = index.as_query_engine()
|
18 |
|
19 |
def chatbot_response(message):
|