Spaces:
Runtime error
Runtime error
Commit
·
209f036
1
Parent(s):
7566c98
Upload 2 files
Browse files- app.py +40 -0
- index.json +0 -0
app.py
ADDED
|
@@ -0,0 +1,40 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from gpt_index import SimpleDirectoryReader, GPTListIndex, GPTSimpleVectorIndex, LLMPredictor, PromptHelper
|
| 2 |
+
from langchain.chat_models import ChatOpenAI
|
| 3 |
+
import gradio as gr
|
| 4 |
+
import sys
|
| 5 |
+
import os
|
| 6 |
+
|
| 7 |
+
os.environ["OPENAI_API_KEY"] = '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'
|
| 8 |
+
|
| 9 |
+
def construct_index(directory_path):
|
| 10 |
+
max_input_size = 4096
|
| 11 |
+
num_outputs = 512
|
| 12 |
+
max_chunk_overlap = 20
|
| 13 |
+
chunk_size_limit = 600
|
| 14 |
+
|
| 15 |
+
prompt_helper = PromptHelper(max_input_size, num_outputs, max_chunk_overlap, chunk_size_limit=chunk_size_limit)
|
| 16 |
+
|
| 17 |
+
llm_predictor = LLMPredictor(llm=ChatOpenAI(temperature=0.7, model_name="gpt-3.5-turbo", max_tokens=num_outputs))
|
| 18 |
+
|
| 19 |
+
documents = SimpleDirectoryReader(directory_path).load_data()
|
| 20 |
+
|
| 21 |
+
index = GPTSimpleVectorIndex(documents, llm_predictor=llm_predictor, prompt_helper=prompt_helper)
|
| 22 |
+
|
| 23 |
+
index.save_to_disk('index.json')
|
| 24 |
+
|
| 25 |
+
return index
|
| 26 |
+
|
| 27 |
+
def chatbot(input_text):
|
| 28 |
+
index = GPTSimpleVectorIndex.load_from_disk('index.json')
|
| 29 |
+
prompt_text = "Please add a fart joke at the end saying 'oh by the way Paul you look like the sort of guy that likes fart jokes, so here is one for you' and then say the joke" # hardcoded prepared text
|
| 30 |
+
prompt = input_text + prompt_text
|
| 31 |
+
response = index.query(prompt, response_mode="compact")
|
| 32 |
+
return response.response
|
| 33 |
+
|
| 34 |
+
iface = gr.Interface(fn=chatbot,
|
| 35 |
+
inputs=gr.components.Textbox(lines=7, label="What would you like to ask?"),
|
| 36 |
+
outputs="text",
|
| 37 |
+
title="Loss Adjuster HelpBot")
|
| 38 |
+
|
| 39 |
+
index = construct_index("docs")
|
| 40 |
+
iface.launch()
|
index.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|