Anisha Bhatnagar
commited on
Commit
Β·
9a097e7
1
Parent(s):
6aef776
triggering feature span caching on precomputed regions
Browse files- app.py +0 -2
- precompute_caches.py +36 -3
app.py
CHANGED
@@ -428,8 +428,6 @@ def app(share=False):
|
|
428 |
label="Precomputed Regions",
|
429 |
info="Select a region to automatically zoom and analyze"
|
430 |
)
|
431 |
-
# Add a hidden HTML component for JavaScript execution
|
432 |
-
js_trigger = gr.HTML(visible=False, elem_id="js-trigger")
|
433 |
|
434 |
|
435 |
with gr.Row():
|
|
|
428 |
label="Precomputed Regions",
|
429 |
info="Select a region to automatically zoom and analyze"
|
430 |
)
|
|
|
|
|
431 |
|
432 |
|
433 |
with gr.Row():
|
precompute_caches.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import os
|
2 |
import json
|
3 |
import pickle
|
@@ -8,9 +9,8 @@ from datetime import datetime
|
|
8 |
import yaml
|
9 |
|
10 |
# Import your actual modules exactly as app.py does
|
11 |
-
from utils.visualizations import get_instances, load_interp_space,
|
12 |
-
from utils.ui import update_task_display
|
13 |
-
from utils.interp_space_utils import cached_generate_style_embedding, compute_g2v_features, compute_predicted_author
|
14 |
|
15 |
def load_config(path="config/config.yaml"):
|
16 |
with open(path, "r") as f:
|
@@ -132,6 +132,39 @@ def precompute_all_caches(
|
|
132 |
|
133 |
print(f" β Instance {instance_id} with model {model_name} completed successfully")
|
134 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
135 |
except Exception as e:
|
136 |
error_msg = f"Error processing instance {instance_id} with model {model_name}: {str(e)}"
|
137 |
print(f" β {error_msg}")
|
|
|
1 |
+
import ast
|
2 |
import os
|
3 |
import json
|
4 |
import pickle
|
|
|
9 |
import yaml
|
10 |
|
11 |
# Import your actual modules exactly as app.py does
|
12 |
+
from utils.visualizations import get_instances, load_interp_space, trigger_precomputed_region, handle_zoom_with_retries
|
13 |
+
from utils.ui import update_task_display
|
|
|
14 |
|
15 |
def load_config(path="config/config.yaml"):
|
16 |
with open(path, "r") as f:
|
|
|
132 |
|
133 |
print(f" β Instance {instance_id} with model {model_name} completed successfully")
|
134 |
|
135 |
+
|
136 |
+
print(" β Testing region zoom simulation...")
|
137 |
+
if precomputed_regions_state:
|
138 |
+
regions_dict = ast.literal_eval(precomputed_regions_state)
|
139 |
+
test_regions = list(regions_dict.keys())
|
140 |
+
|
141 |
+
for region_name in test_regions:
|
142 |
+
try:
|
143 |
+
print(f" β Testing region: {region_name}")
|
144 |
+
|
145 |
+
# Step 3a: Simulate region selection (trigger_precomputed_region)
|
146 |
+
zoom_payload = trigger_precomputed_region(region_name, regions_dict)
|
147 |
+
|
148 |
+
if zoom_payload: # Only proceed if we got a valid zoom payload
|
149 |
+
# Step 3b: Simulate axis_ranges.change() (handle_zoom_with_retries)
|
150 |
+
zoom_results = handle_zoom_with_retries(
|
151 |
+
event_json=zoom_payload,
|
152 |
+
bg_proj=bg_proj,
|
153 |
+
bg_lbls=bg_ids,
|
154 |
+
clustered_authors_df=background_authors_embeddings_df,
|
155 |
+
task_authors_df=task_authors_embeddings_df
|
156 |
+
)
|
157 |
+
|
158 |
+
# Extract results like app.py does
|
159 |
+
(features_rb_update, gram2vec_rb_update, llm_style_feats_analysis,
|
160 |
+
feature_list_state, visible_zoomed_authors) = zoom_results
|
161 |
+
|
162 |
+
print(f" β LLM features cached for region: {region_name}")
|
163 |
+
|
164 |
+
except Exception as e:
|
165 |
+
print(f" β Failed to cache features for region {region_name}: {e}")
|
166 |
+
# Continue with other regions even if one fails
|
167 |
+
continue
|
168 |
except Exception as e:
|
169 |
error_msg = f"Error processing instance {instance_id} with model {model_name}: {str(e)}"
|
170 |
print(f" β {error_msg}")
|