stephenmccartney1234 commited on
Commit
209f036
·
1 Parent(s): 7566c98

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +40 -0
  2. 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