Spaces:
Running
Running
File size: 3,894 Bytes
cddc5c7 |
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 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import gradio as gr
import openai
import os
import time
import shutil
from gpt_index import GPTSimpleVectorIndex, SimpleDirectoryReader
from threading import Lock
from typing import Optional, Tuple
from azure.ai.formrecognizer import DocumentAnalysisClient
from azure.core.credentials import AzureKeyCredential
os.environ['OPENAI_API_KEY'] = "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"
endpoint = "https://eastus.api.cognitive.microsoft.com/"
credential = AzureKeyCredential("844948341c6d4596b77b770cf12e386b")
form_recognizer_client = DocumentAnalysisClient(endpoint=endpoint, credential=credential)
class ChatWrapper:
def __init__(self):
self.lock = Lock()
def __call__(self, input, history: Optional[Tuple[str, str]]):
self.lock.acquire()
try:
history = history or []
new_index = GPTSimpleVectorIndex.load_from_disk('index.json')
response = new_index.query(input, verbose=True)
history.append((input, str(response)))
except Exception as e:
return gr.HTML(f"Error: {e}")
finally:
self.lock.release()
return history, history
def make_status_box_visible():
return gr.update(visible=True), gr.update(visible=False)
def create_index():
documents = SimpleDirectoryReader('data').load_data()
index = GPTSimpleVectorIndex(documents)
index.save_to_disk('index.json')
def pdf_to_text(file_obj, progress=gr.Progress()):
progress(0.2, desc="Uploading file...")
with open(file_obj.name, "rb") as f:
progress(0.5, desc="Analyzing file...")
poller = form_recognizer_client.begin_analyze_document("prebuilt-document", f)
progress(0.8, desc="Applying OCR...")
result = poller.result()
f.close()
progress(0.9, desc="Azure OpenAI Magic...")
#save the result.content in a text file
with open("data/text.txt", "w") as f:
f.write(str(result.content))
f.close()
create_index()
progress(1.0, desc="Done!")
time.sleep(1.5)
return str(result.content), gr.update(visible=True), gr.update(visible=False)
chat = ChatWrapper()
with gr.Blocks(css="footer {visibility: hidden;}") as demo:
chat_history_state = gr.State()
pdf_content = gr.State()
gr.Markdown("""
<sub><sup>created by [@shamill](https://whoplus.microsoft.com/?_vwp=true&_vwpAlias=SHAMILL)</sup></sub>
# Customized GPT-3 Chatbot
GPT-3.5 is a powerful language model, it can be used to create a chatbot that can have a conversation with you. This demo allows you to customize the context of the conversation, and the chatbot will stick to the confines of the context you provide, avoiding made up answers. The chatbot is powered by Azure's OpenAI GPT-3 API.""")
### this is where they will upload the pdf
with gr.Column(visible=False) as chat_interface:
with gr.Row():
chatbot = gr.Chatbot()
with gr.Row():
message_box = gr.Textbox(lines=2, placeholder="Type a message...", default="Hi there!")
submit_button = gr.Button("Submit").style(full_width=False)
submit_button.click(chat, inputs=[message_box, chat_history_state], outputs=[chatbot, chat_history_state])
with gr.Column(visible=True) as upload_interface:
with gr.Row():
upload = gr.File(fn=pdf_to_text, label="Upload a context pdf file", type="file")
with gr.Row():
button = gr.Button("Upload").style(full_width=False)
with gr.Row():
loadingbox = gr.Textbox("Status", visible=False)
button.click(make_status_box_visible, outputs=[loadingbox, button])
button.click(pdf_to_text, inputs=[upload], outputs=[loadingbox, chat_interface, upload_interface])
demo.queue(concurrency_count=20).launch(share=True) |