File size: 775 Bytes
d9814cc
 
 
 
 
 
 
 
 
 
 
9597e6d
d9814cc
 
 
 
 
 
 
 
 
 
 
 
 
 
e2a955f
ab0a935
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from my_functions.save_response import save_response
from llama_index import (
    SimpleDirectoryReader,
    GPTListIndex,
    GPTSimpleVectorIndex,
    LLMPredictor,
    PromptHelper,
    ServiceContext,
)
from llama_index.node_parser import SimpleNodeParser
from langchain import OpenAI
import gradio as gr


def chatbot(input_text):
    index = GPTSimpleVectorIndex.load_from_disk(
        "index.json", service_context=service_context
    )
    response = index.query(input_text, response_mode="default")
    try:
        save_response(input_text, response)
    except Exception as e:
        print("Error saving response:", e)
    return response.response, response.get_formatted_sources()


iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch()