Update app.py

app.py

@@ -26,12 +26,14 @@ def search_bi_encoder(query,top_k=15):
   resultlist = []
 
   hlist = []
+  rank=1  
   for r in results.index:
       if results.title[r] not in hlist:
           resultlist.append(
           {
             "name":results.title[r],
             "bi_encoder_score": results.bi_similarity[r][0],
+            "retrieval_rank":rank,
             "year": results.year[r],
             "language": results.language[r],
             "cast":results.cast[r],
@@ -39,6 +41,8 @@ def search_bi_encoder(query,top_k=15):
             "link":results.link[r]
           })
           hlist.append(results.title[r])
+          rank = rank+1
+    
   return resultlist
 
 
@@ -46,7 +50,9 @@ def search_cross_encoder(query,candidates):
   cross_inp = [[query, candidate['plot']] for candidate in candidates]
   cross_scores = cross_encoder.predict(cross_inp)
   for idx in range(len(cross_scores)):
-    candidates[idx]['cross-score'] = cross_scores[idx]
+      candidates[idx]['cross-score'] = cross_scores[idx]
+      candidates[idx]['re-rank'] = idx+1
+
 
   sortedResult = sorted(candidates, key=lambda x: x['cross-score'], reverse=True)
   return sortedResult
@@ -61,12 +67,12 @@ def search(query,top_k=15):
 
 def displayResults(results, container):
     for result in results:
-        container.header(result.name)
-        container.caption(f"Language:{result.language}, Released in:{result.year}")
-        cast = ",".join(result.cast)
+        container.header(result['name'])
+        container.caption(f"Language:{result['language']}, Released in:{result['year']}")
+        cast = ",".join(result['cast'])
         container.caption(f"Cast:{cast}")
         container.text(result.plot)
-        container.markdown(f"![{result.name}](https://en.wikipedia.org/{result.link}")
+        container.markdown(f"![{result['name']}](https://en.wikipedia.org/{result['link']}")
         with container.expander("Movie result internals"):
             container.json(result)