Update app.py

app.py

@@ -1,4 +1,6 @@
 import os
+import time
+print(f"Starting up: {time.strftime('%Y-%m-%d %H:%M:%S')}")
 #os.system("pip uninstall -y gradio")
 #os.system("pip install --upgrade gradio")
 #os.system("pip install datamapplot==0.3.0")
@@ -65,6 +67,7 @@ import umap
 
 
 
+print(f"Imports are done: {time.strftime('%Y-%m-%d %H:%M:%S')}")
 
 
 
@@ -171,6 +174,8 @@ def get_pub(x):
 
 
 
+print(f"Setting up language model: {time.strftime('%Y-%m-%d %H:%M:%S')}")
+
 #device = torch.device("mps" if torch.backends.mps.is_available() else "cuda")
 #print(f"Using device: {device}")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -230,6 +235,9 @@ def create_embeddings(texts_to_embedd):
 
 
 
+print(f"Language model is set up: {time.strftime('%Y-%m-%d %H:%M:%S')}")
+
+
 
 def predict(text_input, sample_size_slider, reduce_sample_checkbox, progress=gr.Progress()):
 
@@ -243,13 +251,14 @@ def predict(text_input, sample_size_slider, reduce_sample_checkbox, progress=gr.
     records = []
     total_pages = (query_length + 199) // 200  # Calculate total number of pages
     per_page = 0.3 / total_pages
-    
+    progress_counter = 0
     for i, record in enumerate(chain(*query.paginate(per_page=200))):
         records.append(record)
         
         
         # Update progress bar
-        progress(per_page * i, desc="Getting queried data...")
+        progress(progress_counter * per_page, desc="Getting queried data...")
+        progress_counter += 1
 
 
     
@@ -434,7 +443,8 @@ with gr.Blocks() as block:
 
 def setup_basemap_data():
     # get data.
-    print("getting basemap data...")
+    print(f"Getting basemap data: {time.strftime('%Y-%m-%d %H:%M:%S')}")
+
     #basedata_df = load("100k_filtered_OA_sample_cluster_and_positions.bz")
     basedata_df =pickle.load(open('100k_filtered_OA_sample_cluster_and_positions.pkl', 'rb'))
     print(basedata_df)
@@ -443,7 +453,7 @@ def setup_basemap_data():
 
     
 def setup_mapper():
-    print("getting mapper...")
+    print(f"Getting Mapper: {time.strftime('%Y-%m-%d %H:%M:%S')}")
 
     params_new  = pickle.load(open('umap_mapper_300k_random_OA_specter_2_params.pkl', 'rb'))
     print("setting up mapper...")
@@ -467,6 +477,8 @@ def setup_mapper():
 
 basedata_df = setup_basemap_data()
 mapper = setup_mapper()
+print(f"Setup done, starting up app: {time.strftime('%Y-%m-%d %H:%M:%S')}")
+
 
 
 # mount Gradio app to FastAPI app