Update app.py

app.py

@@ -1,11 +1,14 @@
 import gradio as gr
 import csv
 import random
+import pickle
 import uuid
 from langchain.vectorstores import FAISS
 from langchain.embeddings import HuggingFaceEmbeddings
 
 USER_ID = uuid.uuid4()
+with open("article_list.pkl","rb") as articles:
+    article_list = pickle.load(articles)
 INDEXES = ["miread_large", "miread_contrastive", "scibert_contrastive"]
 MODELS = [
     "biodatlab/MIReAD-Neuro-Large",
@@ -33,9 +36,8 @@ def get_comp(prompt):
     right_output = inference(prompt,right)
     return left_output, right_output
 
-def get_article(db_name="miread_contrastive"):
-    db = vecdbs[INDEXES.index(db_name)]
-    return db.docstore._dict[db.index_to_docstore_id[0]]
+def get_article():
+    return random.choice(article_list)
 
 
 def send_result(l_output, r_output, prompt, pick):
@@ -136,7 +138,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     l_btn.click(fn=lambda x,y,z: send_result(x,y,z,'left'),
                 inputs=[l_output,r_output,prompt],
                 api_name="feedleft")
-    l_btn.click(fn=lambda x,y,z: send_result(x,y,z,'right'),
+    r_btn.click(fn=lambda x,y,z: send_result(x,y,z,'right'),
                 inputs=[l_output,r_output,prompt],
                 api_name="feedright")