app added

README.md

@@ -1,6 +1,6 @@
 ---
-title: The Gpt Who Lived
-emoji: 😻
+title: 🧙‍♂️ The GPT Who Lived 🤖
+emoji: 🪄
 colorFrom: purple
 colorTo: purple
 sdk: gradio

app.py

@@ -0,0 +1,56 @@
+import os
+from pathlib import Path
+
+import gradio as gr
+import hpqa
+from langchain import VectorDBQA
+from langchain.llms import OpenAI
+
+os.environ["OPENAI_API_KEY"] = ""
+index_path = Path("data/hpqa_faiss_index_500")
+
+
+examples = [
+    "How would you sneak into Hogwarts without being detected?",
+    "Why did Snape kill Dumbledore?",
+    "Who is the most badass wizard in the world?",
+    "Who would win a fight between Dumbledore and a grizzly bear?",
+    "How many siblings does Hermione have?",
+    "Why are the Dursleys so mean to Harry?",
+]
+
+
+def api(question, temperature, api_key=None):
+    if api_key is None or len(api_key) == 0:
+        return "You must provide an OpenAI API key to use this demo 👇"
+    if len(question) == 0:
+        return ""
+    document_store = hpqa.load_document_store(index_path, openai_api_key=api_key)
+    chain = VectorDBQA.from_chain_type(
+        llm=OpenAI(temperature=temperature, openai_api_key=api_key),
+        chain_type="stuff",
+        vectorstore=document_store,
+        return_source_documents=True,
+    )
+    response = chain(question)
+    return response["result"].strip()
+
+
+demo = gr.Blocks()
+
+with demo:
+    gr.Markdown("# 🪄 Harry Potter Question-Answering with GPT 🤖")
+    with gr.Row():
+        with gr.Column():
+            question = gr.Textbox(lines=4, label="Question")
+            temperature = gr.Slider(0.0, 2.0, 0.7, step=0.1, label="🍺 Butterbeer Consumed")
+            with gr.Row():
+                clear = gr.Button("Clear")
+                btn = gr.Button("Submit", variant="primary")
+        with gr.Column():
+            answer = gr.Textbox(lines=4, label="Answer")
+            openai_api_key = gr.Textbox(type="password", label="OpenAI API key")
+    btn.click(api, [question, temperature, openai_api_key], answer)
+    clear.click(lambda _: "", question, question)
+    gr.Examples(examples, question)
+demo.launch()

data/hpqa_faiss_index_500/index.faiss

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fc48f66967f6bf7474dc98242ff4ac8660d88d397140f47912c54b2df2a4862d
+size 20195373

data/hpqa_faiss_index_500/index.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dc6dcdacdf6a35b8a71e6b779cde261339c073830048e784935fe4b88b6b0ca8
+size 7093398

requirements.txt

@@ -0,0 +1 @@
+git+https://github.com/johnnygreco/hpqa.git