Update app.py

app.py

@@ -1,14 +1,54 @@
 import gradio as gr
-def update(name):
-    return f"Welcome to Gradio, {name}!", "hello_world"
+from transformers import pipeline
+import numpy as np
+import pandas as pd
+from sentence_transformers import SentenceTransformer, util
+
+# Loading in quotes dataset
+df = pd.read_json("krishnamurti_quotes.json")
+
+# Loading back in our sentence similarity and language model
+model = SentenceTransformer("msmarco-roberta-base-v3") # best performing model
+
+krishnamurti_generator = pipeline("text-generation", model="distilgpt2")
+
+############### DEFINING FUNCTIONS ###########################
+
+def ask_krishnamurti(question):
+  answer = krishnamurti_generator(question)[0]['generated_text']
+  list_of_quotes = get_similar_quotes(question)
+  return answer, list_of_quotes
+
+def get_similar_quotes(question):
+  question_embedding = model.encode(question)
+  sims = [util.dot_score(question_embedding, quote_embedding) for quote_embedding in df['Embedding']]
+  ind = np.argpartition(sims, -5)[-5:]
+  similar_sentences = [df['Quotes'][i] for i in ind]
+  top5quotes = pd.DataFrame(data = similar_sentences, columns=["Quotes"], index=range(1,6))
+  return top5quotes
+ 
+ def main(question):
+     return ask_krishnamurti(question), get_similar_quotes(question)
 
 with gr.Blocks() as demo:
-    gr.Markdown("Start typing below and then click **Run** to see the output.")
+    gr.Markdown("""
+    # Ask Krishanmurti
+    """
+    )
     with gr.Row():
-        inp = gr.Textbox(placeholder="What is your name?")
+        inp = gr.Textbox(placeholder="Place your question here...")
         with gr.Column():
-            out1 = gr.Textbox()
-            out2 = gr.Textbox()
+            out1 = gr.Textbox(
+                        lines=3, 
+                        max_lines=10,
+                        label="Answer"
+                    )
+            out2 = gr.DataFrame(
+                        headers=["Quotes"],
+                        max_rows=5,
+                        interactive=False,
+                        wrap=True)]
+                    )
     btn = gr.Button("Run")
     btn.click(fn=update, inputs=inp, outputs=[out1,out2])