Spaces:
Runtime error
Runtime error
File size: 3,048 Bytes
b73e690 20cd498 b73e690 20cd498 b73e690 20cd498 b73e690 20cd498 |
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 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
import time
import pandas as pd
import gradio as gr
from llama_index import GPTSimpleVectorIndex
title = "Confidential forensics tool with ChatGPT"
examples = ["Who is Phillip Allen?", "What is the project in Austin is about?", "Give me more details about the real estate project"]
index = GPTSimpleVectorIndex.load_from_disk('email.json')
docs_arr = []
for doc in index.docstore.docs:
docs_arr.append(doc)
dat_fr = pd.DataFrame({"Documents loaded": docs_arr})
def respond_upload(btn_upload, message, chat_history):
time.sleep(2)
message = "***File uploaded***"
bot_message = "Your document has been uploaded and will be accounted for your queries."
chat_history.append((message, bot_message))
return btn_upload, "", chat_history
def respond2(message, chat_history, box, btn):
if len(message.strip()) < 1:
message = "***Empty***"
bot_message = "Oops, it looks like your query was not valid. Please make sure you typed something in your text box and then try again."
else:
try:
bot_message = str(index.query(message)).strip()
except:
bot_message = "An error occured when handling your query, please try again."
chat_history.append((message, bot_message))
return message, chat_history, box
def respond(message, chat_history):
if len(message.strip()) < 1:
message = "***Empty***"
bot_message = "Oops, it looks like your query was not valid. Please make sure you typed something in your text box and then try again."
else:
try:
bot_message = str(index.query(message)).strip()
except:
bot_message = "An error occured when handling your query, please try again."
chat_history.append((message, bot_message))
return "", chat_history
with gr.Blocks(title=title) as demo:
gr.Markdown(
"""
# """ + title + """
...
""")
dat = gr.Dataframe(
value=dat_fr
)
gr.Markdown(
"""
## Chatbot
""")
chatbot = gr.Chatbot().style(height=400)
with gr.Row():
with gr.Column(scale=0.70):
msg = gr.Textbox(
show_label=False,
placeholder="Enter text and press enter, or click on Send.",
).style(container=False)
with gr.Column(scale=0.15, min_width=0):
btn_send = gr.Button("Send your query")
with gr.Column(scale=0.15, min_width=0):
btn_upload = gr.UploadButton("Upload a new document...", file_types=["text"])
with gr.Row():
gr.Markdown(
"""
Example of queries
""")
i = 0
for ex in examples:
btn = gr.Button(examples[i])
btn.click(respond2, [btn, chatbot, msg], [btn, chatbot, msg])
i += 1
msg.submit(respond, [msg, chatbot], [msg, chatbot])
btn_send.click(respond, [msg, chatbot], [msg, chatbot])
btn_upload.upload(respond_upload, [btn_upload, msg, chatbot], [btn_upload, msg, chatbot])
if __name__ == "__main__":
demo.launch() |