Upload folder using huggingface_hub

main.py

@@ -270,15 +270,10 @@ async def get_sim_map(query_id: str, idx: int, token: str):
 
 async def update_full_image_cache(docid: str, query_id: str, idx: int, image_data: str):
     result = result_cache.get(query_id)
-    if result is None:
-        await asyncio.sleep(0.5)
-        return
-    search_results = get_results_children(result)
-    # Check if idx exists in list of children
-    if idx >= len(search_results):
-        await asyncio.sleep(0.5)
-        return
-    search_results[idx]["fields"]["full_image"] = image_data
+    try:
+        result["root"]["children"][idx]["fields"]["full_image"] = image_data
+    except KeyError as err:
+        print(f"Error updating full image cache: {err}")
     result_cache.set(query_id, result)
     return