include answers

app.py

@@ -46,8 +46,6 @@ def main():
 
     # Filters
     st.sidebar.markdown("**Filters**")
-
-    filter_scores = st.sidebar.slider("Citations", 0, 250, 0)
     num_results = st.sidebar.slider("Number of search results", 1, 50, 1)
 
     vector = query([user_input])
@@ -55,30 +53,23 @@ def main():
     if user_input:
         # Get paper IDs
         _, I = faiss_index.search(np.array(vector).astype("float32"), k=num_results)
-        #D, I = vector_search([user_input],tok, model, faiss_index, num_results)
-        
-        # Slice data on year
-        #frame = data[
-        #    (data.scores >= filter_scores)
-        #]
         
+        answers=row['answers']['text']
+        answers_URLs = row['answers_urls']['url']
+        for k in range(len(answers_URLs)):
+            answers = [answer.replace(f'_URL_{k}_',answers_URLs[k]) 
+            for answer in answers]
+
         frame = data
-        st.write(user_input)
         # Get individual results
         for id_ in I.flatten().tolist():
             f = frame[id_]
-            #if id_ in set(frame.id):
-            #    f = frame[(frame.id == id_)]
-            #else:
-            #    continue
-
             st.write(
-                f"""**title**: {f['title']}"""
+                f"""**Title**: {f['title']}
+                **Top Answer**: {answers[0]}
+                """
             )
-            if f['selftext']!='':
-               st.write(
-                f"""**text**: {f['selftext']}"""
-            ) 
+            st.write("-"*20)