Update app.py

app.py

@@ -40,6 +40,9 @@ def search_embeddings(query_text, output_option):
     elapsed_time_meilisearch = time.time() - start_time_meilisearch
     hits = response["hits"]
 
+    sources_md = [f"[\"{hit['source_page_title']}\"]({hit['source_page_url']})" for hit in hits]
+    sources_md = ",".join(sources_md)
+
     # step3: present the results in markdown
     if output_option == "human-friendly":
         md = f"Stats:\n\nembedding time: {elapsed_time_embedding:.2f}s\n\nmeilisearch time: {elapsed_time_meilisearch:.2f}s\n\n---\n\n"
@@ -47,20 +50,20 @@ def search_embeddings(query_text, output_option):
             text, source_page_url, source_page_title = hit["text"], hit["source_page_url"], hit["source_page_title"]
             source = f"src: [\"{source_page_title}\"]({source_page_url})"
             md += text + f"\n\n{source}\n\n---\n\n"
-        return md
+        return md, sources_md
     elif output_option == "RAG-friendly":
         hit_texts = [hit["text"] for hit in hits]
         hit_text_str = "\n------------\n".join(hit_texts)
-        return hit_text_str
+        return hit_text_str, sources_md
 
 
 demo = gr.Interface(
     fn=search_embeddings,
     inputs=[gr.Textbox(label="enter your query", placeholder="Type Markdown here...", lines=10), gr.Radio(label="Select an output option", choices=output_options, value="RAG-friendly")],
-    outputs=gr.Markdown(),
+    outputs=[gr.Markdown(), gr.Markdown()],
     title="HF Docs Emebddings Explorer",
     allow_flagging="never"
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()