fixed bug when plotting tfidf

app.py

@@ -175,15 +175,18 @@ class ArxivSearch:
                 marker=dict(size=3.5, color='orange', opacity=0.75),
                 name='Results'
             )
-            query_trace = go.Scatter3d(
-                x=query_point[:, 0],
-                y=query_point[:, 1],
-                z=query_point[:, 2],
-                mode='markers',
-                marker=dict(size=5, color='red', opacity=0.8),
-                name='Query'
-            )
-            fig = go.Figure(data=[trace, results_trace, query_trace], layout=layout)
+            if self.query_encoding is not None and self.query_encoding.shape[0] > 0:
+                query_trace = go.Scatter3d(
+                    x=query_point[:, 0],
+                    y=query_point[:, 1],
+                    z=query_point[:, 2],
+                    mode='markers',
+                    marker=dict(size=5, color='red', opacity=0.8),
+                    name='Query'
+                )
+                fig = go.Figure(data=[trace, results_trace, query_trace], layout=layout)
+            else:
+                fig = go.Figure(data=[trace, results_trace], layout=layout)
         else:
             fig = go.Figure(data=[trace], layout=layout)
         return fig