Update app.py

app.py

@@ -180,17 +180,16 @@ def predict(text_input, progress=gr.Progress()):
     
     embeddings = create_embeddings(texts_to_embedd)
     print(embeddings)
-    
-    
-    
-    
-    
-    
-    
-    
-    
-    
-    
+    2d_embeddings = mapper.transform(embeddings)
+    records_df[['x','y']] = 2d_embeddings
+
+    basedata_df['color'] = '#ced4d211'
+    records_df['color'] = '#a81a26'
+
+    stacked_df = pd.concat([records_df, basedata_df], axis=0, ignore_index=True)
+    stacked_df = stacked_df.fillna("Unlabelled")
+
+
     
     
     
@@ -203,10 +202,10 @@ def predict(text_input, progress=gr.Progress()):
     progress(0.7, desc="Loading hover data...")
 
     plot = datamapplot.create_interactive_plot(
-        basedata_df[['x','y']].values,
+        stacked_df[['x','y']].values,
         np.array(basedata_df['cluster_1_labels']),
-        hover_text=[str(ix) + ', ' + str(row['parsed_publication']) + str(row['title']) for ix, row in basedata_df.iterrows()],
-        font_family="Roboto Condensed",
+        hover_text=[str(ix) + ', ' + str(row['parsed_publication']) + str(row['title']) for ix, row in stacked_df.iterrows()],
+        font_family="Roboto Condensed",marker_color_array=stacked_df['color']
     )
 
     progress(0.9, desc="Saving plot...")