update the web demo

Sid2697 · Sid2697 · commit 80c4fe819e93 · 2022-10-20T11:22:27.000+05:30
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,4 @@
 dataset/
 **__pycache__**
 .vscode
+*.DS_Store
diff --git a/kg_query/main.py b/kg_query/main.py
@@ -7,6 +7,7 @@
 import argparse
 import numpy as np
 from tqdm import tqdm
+from pprint import pprint
 
 from .query_embds import QueryEmbds
 from img_models.generate_features import ImgFeatures
@@ -153,7 +154,24 @@ def generate_img_results(self, image_path, num_img=10):
         default='/scratch/sid/dataset_embeddings',
         help='Path to the directory containing the embeddings',
     )
+    parser.add_argument(
+        '-n',
+        default=10,
+        type=int,
+        help='Number of results to show',
+    )
+    parser.add_argument(
+        '-p',
+        required=True,
+        help='Path to the image for quering the DBpedia articles',
+    )
     args = parser.parse_args()
     ranked_list = RankedList()
-    images_ranked_list, uris_ranked_list = ranked_list.generate_results(0)
-    print(images_ranked_list[:10])
+    images_ranked_list, uris_ranked_list = ranked_list.generate_img_results(
+        image_path=args.p,
+        num_img=args.n,
+    )
+    print('List of DBpedia articles...')
+    print(f'Rank\tURI')
+    for i, uri in enumerate(uris_ranked_list):
+        print(f'{i+1}\t{uri}')
diff --git a/website_demo/app.py b/website_demo/app.py
@@ -10,7 +10,7 @@
 ranked_list = RankedList()
 
 
-@app.route('/upload')
+@app.route('/')
 def upload_file_():
     return render_template('upload.html')
 
diff --git a/website_demo/templates/results.html b/website_demo/templates/results.html
@@ -2,11 +2,11 @@
 <html lang="en">
 <head>
     <meta charset="UTF-8">
-    <title>Image-based DBpedia Search</title>
+    <title>Image-based Knowledge Graph Query Framework</title>
 </head>
 <body>
 <form action="/" method="get">
-<h1 align="center">Image-based DBpedia Search</h1>
+<h1 align="center">Image-based Knowledge Graph Query Framework</h1>
 
 <div style="display:flex">
     <div style="flex:1;padding-right:5px;padding-left:5px">
