custom-chatgpt / app.py
daveckw's picture
Merge branch 'main' of https://huggingface.co/spaces/daveckw/custom-chatgpt
e2a955f
raw
history blame
775 Bytes
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()