Update app.py

app.py

@@ -178,9 +178,10 @@ if search_buttion or prev_query != str(query_input):
     results = zip(range(len(distances)), distances)
     results = sorted(results, key=lambda x: x[1])
 
-    for idx, distance in results[0:closest_n]:
-        # print(sentences[idx].strip(), "(Score: %.4f)" % (distance / 2))
-        st.markdown(
-            f"[%.4f {df.iloc[idx]['title']}]({df.iloc[idx]['page']})"
-            % (distance / 2)
-        )
+    for i, (idx, distance) in enumerate(results[0:closest_n]):
+        md_content = ""
+        page_url = df.iloc[idx]["page"]
+        for img in df.iloc[idx]["imgs"]:
+            md_content += f'<a href="{page_url}" target="_blank" rel="noopener noreferrer"><img src="{img}" width="100"></a>'
+        md_content += f'\n[%.4f {df.iloc[idx]["title"]}]({page_url})' % (distance / 2)
+        st.markdown(md_content, unsafe_allow_html=True)